Mercurial > libervia-backend
annotate src/test/test_plugin_xep_0033.py @ 853:c2f6ada7858f
core (sqlite): automatic database update:
- new Updater class check database consistency (by calculating a hash on the .schema), and updates base if necessary
- database now has a version (1 for current, 0 will be for 0.3's database), for each change this version will be increased
- creation statements and update statements are in the form of dict of dict with tuples. There is a help text at the top of the module to explain how it works
- if we are on a development version, the updater try to update the database automaticaly (without deleting table or columns). The Updater.generateUpdateData method can be used to ease the creation of update data (i.e. the dictionary at the top, see the one for the key 1 for an example).
- if there is an inconsistency, an exception is raised, and a message indicate the SQL statements that should fix the situation.
- well... this is rather complicated, a KISS method would maybe have been better. The future will say if we need to simplify it :-/
- new DatabaseError exception
| author | Goffi <goffi@goffi.org> |
|---|---|
| date | Sun, 23 Feb 2014 23:30:32 +0100 |
| parents | 1fe00f0c9a91 |
| children | 6f96ee4d8cc0 |
| rev | line source |
|---|---|
| 787 | 1 #!/usr/bin/python |
| 2 # -*- coding: utf-8 -*- | |
| 3 | |
| 4 # SAT: a jabber client | |
| 811 | 5 # Copyright (C) 2009, 2010, 2011, 2012, 2013, 2014 Jérôme Poisson (goffi@goffi.org) |
| 6 # Copyright (C) 2013, 2014 Adrien Cossa (souliane@mailoo.org) | |
| 787 | 7 |
| 8 # This program is free software: you can redistribute it and/or modify | |
| 9 # it under the terms of the GNU Affero General Public License as published by | |
| 10 # the Free Software Foundation, either version 3 of the License, or | |
| 11 # (at your option) any later version. | |
| 12 | |
| 13 # This program is distributed in the hope that it will be useful, | |
| 14 # but WITHOUT ANY WARRANTY; without even the implied warranty of | |
| 15 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the | |
| 16 # GNU Affero General Public License for more details. | |
| 17 | |
| 18 # You should have received a copy of the GNU Affero General Public License | |
| 19 # along with this program. If not, see <http://www.gnu.org/licenses/>. | |
| 20 | |
| 21 """ Plugin extended addressing stanzas """ | |
| 22 | |
| 23 from constants import Const | |
| 24 from sat.test import helpers | |
| 25 from sat.plugins import plugin_xep_0033 as plugin | |
| 26 from sat.memory.memory import NO_SECURITY_LIMIT | |
| 27 from sat.core.sat_main import AbortSendMessage, MessageSentAndStored | |
| 28 from copy import deepcopy | |
| 29 from twisted.internet import defer | |
| 30 from wokkel.generic import parseXml | |
| 31 from twisted.words.protocols.jabber.jid import JID | |
|
793
cb2db0d85029
test: silent info/warning that were polluting the output
souliane <souliane@mailoo.org>
parents:
792
diff
changeset
|
32 import logging |
| 787 | 33 |
| 34 | |
| 35 class XEP_0033Test(helpers.SatTestCase): | |
| 36 | |
| 37 def setUp(self): | |
| 38 self.host = helpers.FakeSAT() | |
| 39 self.plugin = plugin.XEP_0033(self.host) | |
| 40 | |
| 41 def test_messageReceived(self): | |
| 42 self.host.memory.init() | |
| 43 xml = u""" | |
| 44 <message type="chat" from="%s" to="%s" id="test_1"> | |
| 45 <body>test</body> | |
| 46 <addresses xmlns='http://jabber.org/protocol/address'> | |
| 47 <address type='to' jid='%s'/> | |
| 48 <address type='cc' jid='%s'/> | |
| 49 <address type='bcc' jid='%s'/> | |
| 50 </addresses> | |
| 51 </message> | |
|
792
2136be5a44a8
test: define the constants JIDs and profiles as lists
souliane <souliane@mailoo.org>
parents:
789
diff
changeset
|
52 """ % (Const.JID_STR[1], self.host.getClientHostJid(Const.PROFILE[0]), |
|
2136be5a44a8
test: define the constants JIDs and profiles as lists
souliane <souliane@mailoo.org>
parents:
789
diff
changeset
|
53 Const.JID_STR[0], Const.JID_STR[1], Const.JID_STR[2]) |
| 787 | 54 stanza = parseXml(xml.encode("utf-8")) |
| 55 treatments = defer.Deferred() | |
|
792
2136be5a44a8
test: define the constants JIDs and profiles as lists
souliane <souliane@mailoo.org>
parents:
789
diff
changeset
|
56 self.plugin.messageReceivedTrigger(stanza, treatments, Const.PROFILE[0]) |
| 787 | 57 data = {'extra': {}} |
| 58 | |
|
789
0cb423500fbb
test: use the SatTestCase methods instead of builtin "assert" in tests for memory, plugin xep-0033
souliane <souliane@mailoo.org>
parents:
787
diff
changeset
|
59 def cb(data): |
|
792
2136be5a44a8
test: define the constants JIDs and profiles as lists
souliane <souliane@mailoo.org>
parents:
789
diff
changeset
|
60 expected = ('to', Const.JID_STR[0], 'cc', Const.JID_STR[1], 'bcc', Const.JID_STR[2]) |
|
789
0cb423500fbb
test: use the SatTestCase methods instead of builtin "assert" in tests for memory, plugin xep-0033
souliane <souliane@mailoo.org>
parents:
787
diff
changeset
|
61 msg = 'Expected: %s\nGot: %s' % (expected, data['extra']['addresses']) |
|
0cb423500fbb
test: use the SatTestCase methods instead of builtin "assert" in tests for memory, plugin xep-0033
souliane <souliane@mailoo.org>
parents:
787
diff
changeset
|
62 self.assertEqual(data['extra']['addresses'], '%s:%s\n%s:%s\n%s:%s\n' % expected, msg) |
| 787 | 63 |
|
789
0cb423500fbb
test: use the SatTestCase methods instead of builtin "assert" in tests for memory, plugin xep-0033
souliane <souliane@mailoo.org>
parents:
787
diff
changeset
|
64 treatments.addCallback(cb) |
| 787 | 65 treatments.callback(data) |
| 66 | |
| 67 def test_sendMessageTrigger(self): | |
|
792
2136be5a44a8
test: define the constants JIDs and profiles as lists
souliane <souliane@mailoo.org>
parents:
789
diff
changeset
|
68 mess_data = {"to": self.host.getClientHostJid(Const.PROFILE[0]), |
| 787 | 69 "type": "chat", |
| 70 "message": "content", | |
| 71 "extra": {} | |
| 72 } | |
|
792
2136be5a44a8
test: define the constants JIDs and profiles as lists
souliane <souliane@mailoo.org>
parents:
789
diff
changeset
|
73 addresses = ('to', Const.JID_STR[0], 'cc', Const.JID_STR[1], 'bcc', Const.JID_STR[2]) |
| 787 | 74 mess_data["extra"]["address"] = '%s:%s\n%s:%s\n%s:%s\n' % addresses |
| 75 original_stanza = u""" | |
| 76 <message type="chat" from="%s" to="%s" id="test_1"> | |
| 77 <body>content</body> | |
| 78 </message> | |
|
792
2136be5a44a8
test: define the constants JIDs and profiles as lists
souliane <souliane@mailoo.org>
parents:
789
diff
changeset
|
79 """ % (Const.JID_STR[1], self.host.getClientHostJid(Const.PROFILE[0])) |
| 787 | 80 mess_data['xml'] = parseXml(original_stanza.encode("utf-8")) |
| 81 expected = deepcopy(mess_data['xml']) | |
| 82 addresses_extra = """ | |
| 83 <addresses xmlns='http://jabber.org/protocol/address'> | |
| 84 <address type='%s' jid='%s'/> | |
| 85 <address type='%s' jid='%s'/> | |
| 86 <address type='%s' jid='%s'/> | |
| 87 </addresses>""" % addresses | |
| 88 addresses_element = parseXml(addresses_extra.encode('utf-8')) | |
| 89 expected.addChild(addresses_element) | |
| 90 | |
|
789
0cb423500fbb
test: use the SatTestCase methods instead of builtin "assert" in tests for memory, plugin xep-0033
souliane <souliane@mailoo.org>
parents:
787
diff
changeset
|
91 def assertAddresses(mess_data): |
|
0cb423500fbb
test: use the SatTestCase methods instead of builtin "assert" in tests for memory, plugin xep-0033
souliane <souliane@mailoo.org>
parents:
787
diff
changeset
|
92 """The mess_data that we got here has been modified by self.plugin.sendMessageTrigger, |
| 787 | 93 check that the addresses element has been added to the stanza.""" |
| 94 self.assertEqualXML(mess_data['xml'].toXml().encode("utf-8"), expected.toXml().encode("utf-8")) | |
| 95 | |
|
789
0cb423500fbb
test: use the SatTestCase methods instead of builtin "assert" in tests for memory, plugin xep-0033
souliane <souliane@mailoo.org>
parents:
787
diff
changeset
|
96 def sendMessageErrback(failure, exception_class): |
| 787 | 97 """If the failure does encapsulate the expected exception, it will be silently |
| 98 trapped, otherwise it will be re-raised and will make the test fail""" | |
| 99 if exception_class == MessageSentAndStored: | |
|
789
0cb423500fbb
test: use the SatTestCase methods instead of builtin "assert" in tests for memory, plugin xep-0033
souliane <souliane@mailoo.org>
parents:
787
diff
changeset
|
100 assertAddresses(failure.value.mess_data) |
| 787 | 101 failure.trap(exception_class) |
| 102 | |
| 103 def checkSentAndStored(): | |
| 104 """Check that all the recipients got their messages and that the history has been filled. | |
| 105 /!\ see the comments in XEP_0033.sendAndStoreMessage""" | |
| 106 sent = [] | |
| 107 stored = [] | |
| 108 cache = set() | |
| 109 for to_s in [addresses[1], addresses[3], addresses[5]]: | |
| 110 to_jid = JID(to_s) | |
| 111 host = JID(to_jid.host) | |
|
793
cb2db0d85029
test: silent info/warning that were polluting the output
souliane <souliane@mailoo.org>
parents:
792
diff
changeset
|
112 logger = logging.getLogger() |
|
cb2db0d85029
test: silent info/warning that were polluting the output
souliane <souliane@mailoo.org>
parents:
792
diff
changeset
|
113 level = logger.getEffectiveLevel() |
|
cb2db0d85029
test: silent info/warning that were polluting the output
souliane <souliane@mailoo.org>
parents:
792
diff
changeset
|
114 logger.setLevel(logging.ERROR) # remove warning pollution |
|
792
2136be5a44a8
test: define the constants JIDs and profiles as lists
souliane <souliane@mailoo.org>
parents:
789
diff
changeset
|
115 if self.host.memory.hasServerFeature(plugin.NS_ADDRESS, host, Const.PROFILE[0]): |
| 787 | 116 if host not in cache: |
| 117 sent.append(host) | |
| 118 stored.append(host) | |
| 119 cache.add(host) | |
| 120 stored.append(to_jid) | |
| 121 else: | |
| 122 sent.append(to_jid) | |
| 123 stored.append(to_jid) | |
|
793
cb2db0d85029
test: silent info/warning that were polluting the output
souliane <souliane@mailoo.org>
parents:
792
diff
changeset
|
124 logger.setLevel(level) |
|
789
0cb423500fbb
test: use the SatTestCase methods instead of builtin "assert" in tests for memory, plugin xep-0033
souliane <souliane@mailoo.org>
parents:
787
diff
changeset
|
125 msg = "/!\ see the comments in XEP_0033.sendAndStoreMessage" |
|
0cb423500fbb
test: use the SatTestCase methods instead of builtin "assert" in tests for memory, plugin xep-0033
souliane <souliane@mailoo.org>
parents:
787
diff
changeset
|
126 self.assertEqualUnsortedList(self.host.sent_messages, sent, msg) |
|
0cb423500fbb
test: use the SatTestCase methods instead of builtin "assert" in tests for memory, plugin xep-0033
souliane <souliane@mailoo.org>
parents:
787
diff
changeset
|
127 self.assertEqualUnsortedList(self.host.stored_messages, stored, msg) |
| 787 | 128 |
|
793
cb2db0d85029
test: silent info/warning that were polluting the output
souliane <souliane@mailoo.org>
parents:
792
diff
changeset
|
129 def trigger(treatments, data, *args): |
|
cb2db0d85029
test: silent info/warning that were polluting the output
souliane <souliane@mailoo.org>
parents:
792
diff
changeset
|
130 logger = logging.getLogger() |
|
cb2db0d85029
test: silent info/warning that were polluting the output
souliane <souliane@mailoo.org>
parents:
792
diff
changeset
|
131 level = logger.getEffectiveLevel() |
|
cb2db0d85029
test: silent info/warning that were polluting the output
souliane <souliane@mailoo.org>
parents:
792
diff
changeset
|
132 logger.setLevel(logging.ERROR) # remove warning pollution |
|
cb2db0d85029
test: silent info/warning that were polluting the output
souliane <souliane@mailoo.org>
parents:
792
diff
changeset
|
133 self.plugin.sendMessageTrigger(*args) |
|
cb2db0d85029
test: silent info/warning that were polluting the output
souliane <souliane@mailoo.org>
parents:
792
diff
changeset
|
134 treatments.callback(data) |
|
cb2db0d85029
test: silent info/warning that were polluting the output
souliane <souliane@mailoo.org>
parents:
792
diff
changeset
|
135 logger.setLevel(level) |
|
cb2db0d85029
test: silent info/warning that were polluting the output
souliane <souliane@mailoo.org>
parents:
792
diff
changeset
|
136 |
| 787 | 137 # feature is not supported, abort the message |
| 138 self.host.memory.init() | |
| 139 treatments = defer.Deferred() | |
| 140 data = deepcopy(mess_data) | |
|
793
cb2db0d85029
test: silent info/warning that were polluting the output
souliane <souliane@mailoo.org>
parents:
792
diff
changeset
|
141 trigger(treatments, data, data, treatments, Const.PROFILE[0]) |
|
789
0cb423500fbb
test: use the SatTestCase methods instead of builtin "assert" in tests for memory, plugin xep-0033
souliane <souliane@mailoo.org>
parents:
787
diff
changeset
|
142 treatments.addCallbacks(assertAddresses, lambda failure: sendMessageErrback(failure, AbortSendMessage)) |
| 787 | 143 |
| 144 # feature is supported | |
| 145 self.host.init() | |
| 146 self.host.memory.init() | |
|
792
2136be5a44a8
test: define the constants JIDs and profiles as lists
souliane <souliane@mailoo.org>
parents:
789
diff
changeset
|
147 self.host.memory.addServerFeature(plugin.NS_ADDRESS, self.host.getClientHostJid(Const.PROFILE[0]), Const.PROFILE[0]) |
| 787 | 148 treatments = defer.Deferred() |
| 149 data = deepcopy(mess_data) | |
|
793
cb2db0d85029
test: silent info/warning that were polluting the output
souliane <souliane@mailoo.org>
parents:
792
diff
changeset
|
150 trigger(treatments, data, data, treatments, Const.PROFILE[0]) |
|
789
0cb423500fbb
test: use the SatTestCase methods instead of builtin "assert" in tests for memory, plugin xep-0033
souliane <souliane@mailoo.org>
parents:
787
diff
changeset
|
151 treatments.addCallbacks(assertAddresses, lambda failure: sendMessageErrback(failure, MessageSentAndStored)) |
| 787 | 152 checkSentAndStored() |
| 153 | |
| 154 # check that a wrong recipient entity is fixed by the backend | |
| 155 self.host.init() | |
| 156 self.host.memory.init() | |
|
792
2136be5a44a8
test: define the constants JIDs and profiles as lists
souliane <souliane@mailoo.org>
parents:
789
diff
changeset
|
157 self.host.memory.addServerFeature(plugin.NS_ADDRESS, self.host.getClientHostJid(Const.PROFILE[0]), Const.PROFILE[0]) |
| 787 | 158 treatments = defer.Deferred() |
| 159 data = deepcopy(mess_data) | |
|
792
2136be5a44a8
test: define the constants JIDs and profiles as lists
souliane <souliane@mailoo.org>
parents:
789
diff
changeset
|
160 data["to"] = Const.JID[0] |
|
793
cb2db0d85029
test: silent info/warning that were polluting the output
souliane <souliane@mailoo.org>
parents:
792
diff
changeset
|
161 trigger(treatments, mess_data, data, treatments, Const.PROFILE[0]) |
|
789
0cb423500fbb
test: use the SatTestCase methods instead of builtin "assert" in tests for memory, plugin xep-0033
souliane <souliane@mailoo.org>
parents:
787
diff
changeset
|
162 treatments.addCallbacks(assertAddresses, lambda failure: sendMessageErrback(failure, MessageSentAndStored)) |
| 787 | 163 checkSentAndStored() |