annotate src/test/test_core_xmpp.py @ 663:8004c7d4aba7

core: Deferred in onMessage. onMessage now use a deferred which is passed to MessageReceived trigger through the post_treat parameter. This can be used by plugins to add deferred in the callback chain.
author Goffi <goffi@goffi.org>
date Thu, 31 Oct 2013 17:18:04 +0100
parents e20c823f23e2
children 9810f22ba733
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
rev   line source
335
Goffi <goffi@goffi.org>
parents:
diff changeset
1 #!/usr/bin/python
Goffi <goffi@goffi.org>
parents:
diff changeset
2 # -*- coding: utf-8 -*-
Goffi <goffi@goffi.org>
parents:
diff changeset
3
609
84a6e83157c2 fixed licences in docstrings (they are now in comments)
Goffi <goffi@goffi.org>
parents: 591
diff changeset
4 # SAT: a jabber client
84a6e83157c2 fixed licences in docstrings (they are now in comments)
Goffi <goffi@goffi.org>
parents: 591
diff changeset
5 # Copyright (C) 2009, 2010, 2011, 2012, 2013 Jérôme Poisson (goffi@goffi.org)
335
Goffi <goffi@goffi.org>
parents:
diff changeset
6
609
84a6e83157c2 fixed licences in docstrings (they are now in comments)
Goffi <goffi@goffi.org>
parents: 591
diff changeset
7 # This program is free software: you can redistribute it and/or modify
84a6e83157c2 fixed licences in docstrings (they are now in comments)
Goffi <goffi@goffi.org>
parents: 591
diff changeset
8 # it under the terms of the GNU Affero General Public License as published by
84a6e83157c2 fixed licences in docstrings (they are now in comments)
Goffi <goffi@goffi.org>
parents: 591
diff changeset
9 # the Free Software Foundation, either version 3 of the License, or
84a6e83157c2 fixed licences in docstrings (they are now in comments)
Goffi <goffi@goffi.org>
parents: 591
diff changeset
10 # (at your option) any later version.
335
Goffi <goffi@goffi.org>
parents:
diff changeset
11
609
84a6e83157c2 fixed licences in docstrings (they are now in comments)
Goffi <goffi@goffi.org>
parents: 591
diff changeset
12 # This program is distributed in the hope that it will be useful,
84a6e83157c2 fixed licences in docstrings (they are now in comments)
Goffi <goffi@goffi.org>
parents: 591
diff changeset
13 # but WITHOUT ANY WARRANTY; without even the implied warranty of
84a6e83157c2 fixed licences in docstrings (they are now in comments)
Goffi <goffi@goffi.org>
parents: 591
diff changeset
14 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
84a6e83157c2 fixed licences in docstrings (they are now in comments)
Goffi <goffi@goffi.org>
parents: 591
diff changeset
15 # GNU Affero General Public License for more details.
335
Goffi <goffi@goffi.org>
parents:
diff changeset
16
609
84a6e83157c2 fixed licences in docstrings (they are now in comments)
Goffi <goffi@goffi.org>
parents: 591
diff changeset
17 # You should have received a copy of the GNU Affero General Public License
84a6e83157c2 fixed licences in docstrings (they are now in comments)
Goffi <goffi@goffi.org>
parents: 591
diff changeset
18 # along with this program. If not, see <http://www.gnu.org/licenses/>.
335
Goffi <goffi@goffi.org>
parents:
diff changeset
19
Goffi <goffi@goffi.org>
parents:
diff changeset
20 from sat.test import helpers
Goffi <goffi@goffi.org>
parents:
diff changeset
21 from twisted.trial import unittest
Goffi <goffi@goffi.org>
parents:
diff changeset
22 from sat.core.sat_main import SAT
Goffi <goffi@goffi.org>
parents:
diff changeset
23 from sat.core import xmpp
Goffi <goffi@goffi.org>
parents:
diff changeset
24 from twisted.internet import defer
Goffi <goffi@goffi.org>
parents:
diff changeset
25 from twisted.words.protocols.jabber.jid import JID
Goffi <goffi@goffi.org>
parents:
diff changeset
26 from wokkel.generic import parseXml
Goffi <goffi@goffi.org>
parents:
diff changeset
27 from wokkel.xmppim import RosterItem
Goffi <goffi@goffi.org>
parents:
diff changeset
28
Goffi <goffi@goffi.org>
parents:
diff changeset
29
Goffi <goffi@goffi.org>
parents:
diff changeset
30 class SatXMPPClientTest(unittest.TestCase):
Goffi <goffi@goffi.org>
parents:
diff changeset
31
Goffi <goffi@goffi.org>
parents:
diff changeset
32 def setUp(self):
Goffi <goffi@goffi.org>
parents:
diff changeset
33 self.host = helpers.FakeSAT()
587
952322b1d490 Remove trailing whitespaces.
Emmanuel Gil Peyrot <linkmauve@linkmauve.fr>
parents: 572
diff changeset
34 self.client = xmpp.SatXMPPClient(self.host, "test_profile", JID("test@example.org"), "test")
952322b1d490 Remove trailing whitespaces.
Emmanuel Gil Peyrot <linkmauve@linkmauve.fr>
parents: 572
diff changeset
35
335
Goffi <goffi@goffi.org>
parents:
diff changeset
36 def test_init(self):
Goffi <goffi@goffi.org>
parents:
diff changeset
37 """Check that init values are correctly initialised"""
Goffi <goffi@goffi.org>
parents:
diff changeset
38 self.assertEqual(self.client.profile, "test_profile")
Goffi <goffi@goffi.org>
parents:
diff changeset
39 print self.client.host
Goffi <goffi@goffi.org>
parents:
diff changeset
40 self.assertEqual(self.client.host_app, self.host)
Goffi <goffi@goffi.org>
parents:
diff changeset
41 self.assertTrue(isinstance(self.client.client_initialized, defer.Deferred))
Goffi <goffi@goffi.org>
parents:
diff changeset
42
591
65821b3fa7ab Fix pep8 support in src/test.
Emmanuel Gil Peyrot <linkmauve@linkmauve.fr>
parents: 587
diff changeset
43
335
Goffi <goffi@goffi.org>
parents:
diff changeset
44 class SatMessageProtocolTest(unittest.TestCase):
Goffi <goffi@goffi.org>
parents:
diff changeset
45
Goffi <goffi@goffi.org>
parents:
diff changeset
46 def setUp(self):
Goffi <goffi@goffi.org>
parents:
diff changeset
47 self.host = helpers.FakeSAT()
Goffi <goffi@goffi.org>
parents:
diff changeset
48 self.message = xmpp.SatMessageProtocol(self.host)
649
e20c823f23e2 tests: helpers improvments:
Goffi <goffi@goffi.org>
parents: 648
diff changeset
49 self.message.parent = helpers.FakeClient(self.host)
335
Goffi <goffi@goffi.org>
parents:
diff changeset
50
Goffi <goffi@goffi.org>
parents:
diff changeset
51 def test_onMessage(self):
Goffi <goffi@goffi.org>
parents:
diff changeset
52 xml = """
Goffi <goffi@goffi.org>
parents:
diff changeset
53 <message type="chat" from="sender@example.net/house" to="test@example.org/SàT" id="test_1">
Goffi <goffi@goffi.org>
parents:
diff changeset
54 <body>test</body>
Goffi <goffi@goffi.org>
parents:
diff changeset
55 </message>
Goffi <goffi@goffi.org>
parents:
diff changeset
56 """
Goffi <goffi@goffi.org>
parents:
diff changeset
57 stanza = parseXml(xml)
648
29cea30f20f5 tests: fix SatMessageProtocolTest.test_onMessage
Goffi <goffi@goffi.org>
parents: 609
diff changeset
58 self.host.bridge.expectCall("newMessage", u"sender@example.net/house", u"test", u"chat", u"test@example.org/SàT", {}, profile="test_profile")
335
Goffi <goffi@goffi.org>
parents:
diff changeset
59 self.message.onMessage(stanza)
587
952322b1d490 Remove trailing whitespaces.
Emmanuel Gil Peyrot <linkmauve@linkmauve.fr>
parents: 572
diff changeset
60
591
65821b3fa7ab Fix pep8 support in src/test.
Emmanuel Gil Peyrot <linkmauve@linkmauve.fr>
parents: 587
diff changeset
61
335
Goffi <goffi@goffi.org>
parents:
diff changeset
62 class SatRosterProtocolTest(unittest.TestCase):
Goffi <goffi@goffi.org>
parents:
diff changeset
63
Goffi <goffi@goffi.org>
parents:
diff changeset
64 def setUp(self):
Goffi <goffi@goffi.org>
parents:
diff changeset
65 self.host = helpers.FakeSAT()
Goffi <goffi@goffi.org>
parents:
diff changeset
66 self.roster = xmpp.SatRosterProtocol(self.host)
649
e20c823f23e2 tests: helpers improvments:
Goffi <goffi@goffi.org>
parents: 648
diff changeset
67 self.roster.parent = helpers.FakeClient(self.host)
335
Goffi <goffi@goffi.org>
parents:
diff changeset
68
Goffi <goffi@goffi.org>
parents:
diff changeset
69 def test_onRosterSet(self):
Goffi <goffi@goffi.org>
parents:
diff changeset
70 roster_item = RosterItem(helpers.TEST_JID)
Goffi <goffi@goffi.org>
parents:
diff changeset
71 roster_item.name = u"Test Man"
Goffi <goffi@goffi.org>
parents:
diff changeset
72 roster_item.subscriptionTo = True
Goffi <goffi@goffi.org>
parents:
diff changeset
73 roster_item.subscriptionFrom = True
Goffi <goffi@goffi.org>
parents:
diff changeset
74 roster_item.ask = False
Goffi <goffi@goffi.org>
parents:
diff changeset
75 roster_item.groups = set([u"Test Group 1", u"Test Group 2", u"Test Group 3"])
591
65821b3fa7ab Fix pep8 support in src/test.
Emmanuel Gil Peyrot <linkmauve@linkmauve.fr>
parents: 587
diff changeset
76 self.host.bridge.expectCall("newContact", helpers.TEST_JID_STR, {'to': 'True', 'from': 'True', 'ask': 'False', 'name': u'Test Man'}, set([u"Test Group 1", u"Test Group 2", u"Test Group 3"]), "test_profile")
335
Goffi <goffi@goffi.org>
parents:
diff changeset
77 self.roster.onRosterSet(roster_item)
Goffi <goffi@goffi.org>
parents:
diff changeset
78
591
65821b3fa7ab Fix pep8 support in src/test.
Emmanuel Gil Peyrot <linkmauve@linkmauve.fr>
parents: 587
diff changeset
79
335
Goffi <goffi@goffi.org>
parents:
diff changeset
80 class SatPresenceProtocolTest(unittest.TestCase):
Goffi <goffi@goffi.org>
parents:
diff changeset
81
Goffi <goffi@goffi.org>
parents:
diff changeset
82 def setUp(self):
Goffi <goffi@goffi.org>
parents:
diff changeset
83 self.host = helpers.FakeSAT()
Goffi <goffi@goffi.org>
parents:
diff changeset
84 self.presence = xmpp.SatPresenceProtocol(self.host)
649
e20c823f23e2 tests: helpers improvments:
Goffi <goffi@goffi.org>
parents: 648
diff changeset
85 self.presence.parent = helpers.FakeClient(self.host)
335
Goffi <goffi@goffi.org>
parents:
diff changeset
86
Goffi <goffi@goffi.org>
parents:
diff changeset
87 def test_availableReceived(self):
591
65821b3fa7ab Fix pep8 support in src/test.
Emmanuel Gil Peyrot <linkmauve@linkmauve.fr>
parents: 587
diff changeset
88 self.host.bridge.expectCall("presenceUpdate", helpers.TEST_JID_STR, "xa", 15, {'default': "test status", 'fr': 'statut de test'}, helpers.TEST_PROFILE)
65821b3fa7ab Fix pep8 support in src/test.
Emmanuel Gil Peyrot <linkmauve@linkmauve.fr>
parents: 587
diff changeset
89 self.presence.availableReceived(helpers.TEST_JID, 'xa', {None: "test status", 'fr': 'statut de test'}, 15)
335
Goffi <goffi@goffi.org>
parents:
diff changeset
90
Goffi <goffi@goffi.org>
parents:
diff changeset
91 def test_availableReceived_empty_statuses(self):
Goffi <goffi@goffi.org>
parents:
diff changeset
92 self.host.bridge.expectCall("presenceUpdate", helpers.TEST_JID_STR, "xa", 15, {}, helpers.TEST_PROFILE)
Goffi <goffi@goffi.org>
parents:
diff changeset
93 self.presence.availableReceived(helpers.TEST_JID, 'xa', None, 15)
Goffi <goffi@goffi.org>
parents:
diff changeset
94
Goffi <goffi@goffi.org>
parents:
diff changeset
95 def test_unavailableReceived(self):
Goffi <goffi@goffi.org>
parents:
diff changeset
96 self.host.bridge.expectCall("presenceUpdate", helpers.TEST_JID_STR, "unavailable", 0, {}, helpers.TEST_PROFILE)
Goffi <goffi@goffi.org>
parents:
diff changeset
97 self.presence.unavailableReceived(helpers.TEST_JID, None)
Goffi <goffi@goffi.org>
parents:
diff changeset
98
Goffi <goffi@goffi.org>
parents:
diff changeset
99 def test_subscribedReceived(self):
Goffi <goffi@goffi.org>
parents:
diff changeset
100 self.host.bridge.expectCall("subscribe", "subscribed", helpers.TEST_JID.userhost(), helpers.TEST_PROFILE)
Goffi <goffi@goffi.org>
parents:
diff changeset
101 self.presence.subscribedReceived(helpers.TEST_JID)
587
952322b1d490 Remove trailing whitespaces.
Emmanuel Gil Peyrot <linkmauve@linkmauve.fr>
parents: 572
diff changeset
102
335
Goffi <goffi@goffi.org>
parents:
diff changeset
103 def test_unsubscribedReceived(self):
Goffi <goffi@goffi.org>
parents:
diff changeset
104 self.host.bridge.expectCall("subscribe", "unsubscribed", helpers.TEST_JID.userhost(), helpers.TEST_PROFILE)
Goffi <goffi@goffi.org>
parents:
diff changeset
105 self.presence.unsubscribedReceived(helpers.TEST_JID)
587
952322b1d490 Remove trailing whitespaces.
Emmanuel Gil Peyrot <linkmauve@linkmauve.fr>
parents: 572
diff changeset
106
335
Goffi <goffi@goffi.org>
parents:
diff changeset
107 def test_subscribeReceived(self):
Goffi <goffi@goffi.org>
parents:
diff changeset
108 self.host.bridge.expectCall("subscribe", "subscribe", helpers.TEST_JID.userhost(), helpers.TEST_PROFILE)
Goffi <goffi@goffi.org>
parents:
diff changeset
109 self.presence.subscribeReceived(helpers.TEST_JID)
587
952322b1d490 Remove trailing whitespaces.
Emmanuel Gil Peyrot <linkmauve@linkmauve.fr>
parents: 572
diff changeset
110
335
Goffi <goffi@goffi.org>
parents:
diff changeset
111 def test_unsubscribeReceived(self):
Goffi <goffi@goffi.org>
parents:
diff changeset
112 self.host.bridge.expectCall("subscribe", "unsubscribe", helpers.TEST_JID.userhost(), helpers.TEST_PROFILE)
Goffi <goffi@goffi.org>
parents:
diff changeset
113 self.presence.unsubscribeReceived(helpers.TEST_JID)