Mercurial > libervia-backend
annotate sat/test/test_core_xmpp.py @ 3652:6e34307319c0
plugin XEP-0353: fix jingle initiation on disco "Service Unavailable" error:
When requesting disco info on a bare jid which is not in our roster, server may return
"Service Unavailable" (to avoid leaking valid JIDs). In this case, the initiation was
failing, this is now fixed by using empty categories in this case.
| author | Goffi <goffi@goffi.org> |
|---|---|
| date | Wed, 08 Sep 2021 11:16:52 +0200 |
| parents | be6d91572633 |
| children | 524856bd7b19 |
| rev | line source |
|---|---|
| 3028 | 1 #!/usr/bin/env python3 |
| 3137 | 2 |
| 335 | 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 |
| 3479 | 5 # Copyright (C) 2009-2021 Jérôme Poisson (goffi@goffi.org) |
| 335 | 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 | 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 | 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 | 19 |
| 20 from sat.test import helpers | |
| 3028 | 21 from .constants import Const |
| 335 | 22 from twisted.trial import unittest |
| 23 from sat.core import xmpp | |
| 24 from twisted.words.protocols.jabber.jid import JID | |
| 25 from wokkel.generic import parseXml | |
| 26 from wokkel.xmppim import RosterItem | |
| 27 | |
| 28 | |
| 29 class SatXMPPClientTest(unittest.TestCase): | |
| 30 | |
| 31 def setUp(self): | |
| 32 self.host = helpers.FakeSAT() | |
|
792
2136be5a44a8
test: define the constants JIDs and profiles as lists
souliane <souliane@mailoo.org>
parents:
780
diff
changeset
|
33 self.client = xmpp.SatXMPPClient(self.host, Const.PROFILE[0], JID("test@example.org"), "test") |
|
587
952322b1d490
Remove trailing whitespaces.
Emmanuel Gil Peyrot <linkmauve@linkmauve.fr>
parents:
572
diff
changeset
|
34 |
| 335 | 35 def test_init(self): |
| 36 """Check that init values are correctly initialised""" | |
|
792
2136be5a44a8
test: define the constants JIDs and profiles as lists
souliane <souliane@mailoo.org>
parents:
780
diff
changeset
|
37 self.assertEqual(self.client.profile, Const.PROFILE[0]) |
|
3398
467d6c709f1d
test: fixed use of `client.host` in legacy test:
Goffi <goffi@goffi.org>
parents:
3137
diff
changeset
|
38 print(self.client.jid.host) |
| 335 | 39 self.assertEqual(self.client.host_app, self.host) |
| 40 | |
|
591
65821b3fa7ab
Fix pep8 support in src/test.
Emmanuel Gil Peyrot <linkmauve@linkmauve.fr>
parents:
587
diff
changeset
|
41 |
| 335 | 42 class SatMessageProtocolTest(unittest.TestCase): |
| 43 | |
| 44 def setUp(self): | |
| 45 self.host = helpers.FakeSAT() | |
| 46 self.message = xmpp.SatMessageProtocol(self.host) | |
| 649 | 47 self.message.parent = helpers.FakeClient(self.host) |
| 335 | 48 |
| 49 def test_onMessage(self): | |
| 50 xml = """ | |
| 51 <message type="chat" from="sender@example.net/house" to="test@example.org/SàT" id="test_1"> | |
| 52 <body>test</body> | |
| 53 </message> | |
| 54 """ | |
| 55 stanza = parseXml(xml) | |
| 3028 | 56 self.host.bridge.expectCall("messageNew", "sender@example.net/house", "test", "chat", "test@example.org/SàT", {}, profile=Const.PROFILE[0]) |
| 335 | 57 self.message.onMessage(stanza) |
|
587
952322b1d490
Remove trailing whitespaces.
Emmanuel Gil Peyrot <linkmauve@linkmauve.fr>
parents:
572
diff
changeset
|
58 |
|
591
65821b3fa7ab
Fix pep8 support in src/test.
Emmanuel Gil Peyrot <linkmauve@linkmauve.fr>
parents:
587
diff
changeset
|
59 |
| 335 | 60 class SatRosterProtocolTest(unittest.TestCase): |
| 61 | |
| 62 def setUp(self): | |
| 63 self.host = helpers.FakeSAT() | |
| 64 self.roster = xmpp.SatRosterProtocol(self.host) | |
| 649 | 65 self.roster.parent = helpers.FakeClient(self.host) |
| 335 | 66 |
| 1412 | 67 def test__registerItem(self): |
|
792
2136be5a44a8
test: define the constants JIDs and profiles as lists
souliane <souliane@mailoo.org>
parents:
780
diff
changeset
|
68 roster_item = RosterItem(Const.JID[0]) |
| 3028 | 69 roster_item.name = "Test Man" |
| 335 | 70 roster_item.subscriptionTo = True |
| 71 roster_item.subscriptionFrom = True | |
| 72 roster_item.ask = False | |
| 3028 | 73 roster_item.groups = set(["Test Group 1", "Test Group 2", "Test Group 3"]) |
| 74 self.host.bridge.expectCall("newContact", Const.JID_STR[0], {'to': 'True', 'from': 'True', 'ask': 'False', 'name': 'Test Man'}, set(["Test Group 1", "Test Group 2", "Test Group 3"]), Const.PROFILE[0]) | |
| 1412 | 75 self.roster._registerItem(roster_item) |
| 335 | 76 |
|
591
65821b3fa7ab
Fix pep8 support in src/test.
Emmanuel Gil Peyrot <linkmauve@linkmauve.fr>
parents:
587
diff
changeset
|
77 |
| 335 | 78 class SatPresenceProtocolTest(unittest.TestCase): |
| 79 | |
| 80 def setUp(self): | |
| 81 self.host = helpers.FakeSAT() | |
| 82 self.presence = xmpp.SatPresenceProtocol(self.host) | |
| 649 | 83 self.presence.parent = helpers.FakeClient(self.host) |
| 335 | 84 |
| 85 def test_availableReceived(self): | |
|
792
2136be5a44a8
test: define the constants JIDs and profiles as lists
souliane <souliane@mailoo.org>
parents:
780
diff
changeset
|
86 self.host.bridge.expectCall("presenceUpdate", Const.JID_STR[0], "xa", 15, {'default': "test status", 'fr': 'statut de test'}, Const.PROFILE[0]) |
|
2136be5a44a8
test: define the constants JIDs and profiles as lists
souliane <souliane@mailoo.org>
parents:
780
diff
changeset
|
87 self.presence.availableReceived(Const.JID[0], 'xa', {None: "test status", 'fr': 'statut de test'}, 15) |
| 335 | 88 |
| 89 def test_availableReceived_empty_statuses(self): | |
|
792
2136be5a44a8
test: define the constants JIDs and profiles as lists
souliane <souliane@mailoo.org>
parents:
780
diff
changeset
|
90 self.host.bridge.expectCall("presenceUpdate", Const.JID_STR[0], "xa", 15, {}, Const.PROFILE[0]) |
|
2136be5a44a8
test: define the constants JIDs and profiles as lists
souliane <souliane@mailoo.org>
parents:
780
diff
changeset
|
91 self.presence.availableReceived(Const.JID[0], 'xa', None, 15) |
| 335 | 92 |
| 93 def test_unavailableReceived(self): | |
|
792
2136be5a44a8
test: define the constants JIDs and profiles as lists
souliane <souliane@mailoo.org>
parents:
780
diff
changeset
|
94 self.host.bridge.expectCall("presenceUpdate", Const.JID_STR[0], "unavailable", 0, {}, Const.PROFILE[0]) |
|
2136be5a44a8
test: define the constants JIDs and profiles as lists
souliane <souliane@mailoo.org>
parents:
780
diff
changeset
|
95 self.presence.unavailableReceived(Const.JID[0], None) |
| 335 | 96 |
| 97 def test_subscribedReceived(self): | |
|
792
2136be5a44a8
test: define the constants JIDs and profiles as lists
souliane <souliane@mailoo.org>
parents:
780
diff
changeset
|
98 self.host.bridge.expectCall("subscribe", "subscribed", Const.JID[0].userhost(), Const.PROFILE[0]) |
|
2136be5a44a8
test: define the constants JIDs and profiles as lists
souliane <souliane@mailoo.org>
parents:
780
diff
changeset
|
99 self.presence.subscribedReceived(Const.JID[0]) |
|
587
952322b1d490
Remove trailing whitespaces.
Emmanuel Gil Peyrot <linkmauve@linkmauve.fr>
parents:
572
diff
changeset
|
100 |
| 335 | 101 def test_unsubscribedReceived(self): |
|
792
2136be5a44a8
test: define the constants JIDs and profiles as lists
souliane <souliane@mailoo.org>
parents:
780
diff
changeset
|
102 self.host.bridge.expectCall("subscribe", "unsubscribed", Const.JID[0].userhost(), Const.PROFILE[0]) |
|
2136be5a44a8
test: define the constants JIDs and profiles as lists
souliane <souliane@mailoo.org>
parents:
780
diff
changeset
|
103 self.presence.unsubscribedReceived(Const.JID[0]) |
|
587
952322b1d490
Remove trailing whitespaces.
Emmanuel Gil Peyrot <linkmauve@linkmauve.fr>
parents:
572
diff
changeset
|
104 |
| 335 | 105 def test_subscribeReceived(self): |
|
792
2136be5a44a8
test: define the constants JIDs and profiles as lists
souliane <souliane@mailoo.org>
parents:
780
diff
changeset
|
106 self.host.bridge.expectCall("subscribe", "subscribe", Const.JID[0].userhost(), Const.PROFILE[0]) |
|
2136be5a44a8
test: define the constants JIDs and profiles as lists
souliane <souliane@mailoo.org>
parents:
780
diff
changeset
|
107 self.presence.subscribeReceived(Const.JID[0]) |
|
587
952322b1d490
Remove trailing whitespaces.
Emmanuel Gil Peyrot <linkmauve@linkmauve.fr>
parents:
572
diff
changeset
|
108 |
| 335 | 109 def test_unsubscribeReceived(self): |
|
792
2136be5a44a8
test: define the constants JIDs and profiles as lists
souliane <souliane@mailoo.org>
parents:
780
diff
changeset
|
110 self.host.bridge.expectCall("subscribe", "unsubscribe", Const.JID[0].userhost(), Const.PROFILE[0]) |
|
2136be5a44a8
test: define the constants JIDs and profiles as lists
souliane <souliane@mailoo.org>
parents:
780
diff
changeset
|
111 self.presence.unsubscribeReceived(Const.JID[0]) |