Mercurial > libervia-backend
annotate sat/test/test_plugin_xep_0297.py @ 3133:87b8808ac49d
plugin XEP-0198: don't crash when xmlstream has been closed
| author | Goffi <goffi@goffi.org> |
|---|---|
| date | Tue, 28 Jan 2020 22:26:12 +0100 |
| parents | ab2696e34d29 |
| children | 9d0df638c8b4 |
| rev | line source |
|---|---|
| 3028 | 1 #!/usr/bin/env python3 |
| 1274 | 2 # -*- coding: utf-8 -*- |
| 3 | |
| 4 # SAT: a jabber client | |
| 2771 | 5 # Copyright (C) 2009-2019 Jérôme Poisson (goffi@goffi.org) |
| 1766 | 6 # Copyright (C) 2013-2016 Adrien Cossa (souliane@mailoo.org) |
| 1274 | 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 XEP-0297 """ | |
| 22 | |
| 3028 | 23 from .constants import Const as C |
| 1274 | 24 from sat.test import helpers |
| 25 from sat.plugins.plugin_xep_0203 import XEP_0203 | |
| 26 from sat.plugins.plugin_xep_0297 import XEP_0297 | |
| 27 from twisted.words.protocols.jabber.jid import JID | |
| 28 from dateutil.tz import tzutc | |
| 29 import datetime | |
| 30 from wokkel.generic import parseXml | |
| 31 | |
| 32 | |
|
2624
56f94936df1e
code style reformatting using black
Goffi <goffi@goffi.org>
parents:
2562
diff
changeset
|
33 NS_PUBSUB = "http://jabber.org/protocol/pubsub" |
| 1274 | 34 |
| 35 | |
| 36 class XEP_0297Test(helpers.SatTestCase): | |
| 37 def setUp(self): | |
| 38 self.host = helpers.FakeSAT() | |
| 39 self.plugin = XEP_0297(self.host) | |
|
2624
56f94936df1e
code style reformatting using black
Goffi <goffi@goffi.org>
parents:
2562
diff
changeset
|
40 self.host.plugins["XEP-0203"] = XEP_0203(self.host) |
| 1274 | 41 |
| 42 def test_delay(self): | |
|
2624
56f94936df1e
code style reformatting using black
Goffi <goffi@goffi.org>
parents:
2562
diff
changeset
|
43 stanza = parseXml( |
|
56f94936df1e
code style reformatting using black
Goffi <goffi@goffi.org>
parents:
2562
diff
changeset
|
44 """ |
| 1274 | 45 <message from='juliet@capulet.lit/orchard' |
| 46 id='0202197' | |
| 47 to='romeo@montague.lit' | |
| 48 type='chat'> | |
| 49 <body>Yet I should kill thee with much cherishing.</body> | |
| 50 <mood xmlns='http://jabber.org/protocol/mood'> | |
| 51 <amorous/> | |
| 52 </mood> | |
| 53 </message> | |
|
2624
56f94936df1e
code style reformatting using black
Goffi <goffi@goffi.org>
parents:
2562
diff
changeset
|
54 """.encode( |
|
56f94936df1e
code style reformatting using black
Goffi <goffi@goffi.org>
parents:
2562
diff
changeset
|
55 "utf-8" |
|
56f94936df1e
code style reformatting using black
Goffi <goffi@goffi.org>
parents:
2562
diff
changeset
|
56 ) |
|
56f94936df1e
code style reformatting using black
Goffi <goffi@goffi.org>
parents:
2562
diff
changeset
|
57 ) |
| 1274 | 58 output = """ |
| 59 <message to='mercutio@verona.lit' type='chat'> | |
| 60 <body>A most courteous exposition!</body> | |
| 61 <forwarded xmlns='urn:xmpp:forward:0'> | |
| 62 <delay xmlns='urn:xmpp:delay' stamp='2010-07-10T23:08:25Z'/> | |
| 63 <message from='juliet@capulet.lit/orchard' | |
| 64 id='0202197' | |
| 65 to='romeo@montague.lit' | |
| 66 type='chat' | |
| 67 xmlns='jabber:client'> | |
| 68 <body>Yet I should kill thee with much cherishing.</body> | |
| 69 <mood xmlns='http://jabber.org/protocol/mood'> | |
| 70 <amorous/> | |
| 71 </mood> | |
| 72 </message> | |
| 73 </forwarded> | |
| 74 </message> | |
| 75 """ | |
| 76 stamp = datetime.datetime(2010, 7, 10, 23, 8, 25, tzinfo=tzutc()) | |
|
2624
56f94936df1e
code style reformatting using black
Goffi <goffi@goffi.org>
parents:
2562
diff
changeset
|
77 d = self.plugin.forward( |
|
56f94936df1e
code style reformatting using black
Goffi <goffi@goffi.org>
parents:
2562
diff
changeset
|
78 stanza, |
|
56f94936df1e
code style reformatting using black
Goffi <goffi@goffi.org>
parents:
2562
diff
changeset
|
79 JID("mercutio@verona.lit"), |
|
56f94936df1e
code style reformatting using black
Goffi <goffi@goffi.org>
parents:
2562
diff
changeset
|
80 stamp, |
|
56f94936df1e
code style reformatting using black
Goffi <goffi@goffi.org>
parents:
2562
diff
changeset
|
81 body="A most courteous exposition!", |
|
56f94936df1e
code style reformatting using black
Goffi <goffi@goffi.org>
parents:
2562
diff
changeset
|
82 profile_key=C.PROFILE[0], |
|
56f94936df1e
code style reformatting using black
Goffi <goffi@goffi.org>
parents:
2562
diff
changeset
|
83 ) |
|
56f94936df1e
code style reformatting using black
Goffi <goffi@goffi.org>
parents:
2562
diff
changeset
|
84 d.addCallback( |
|
2765
378188abe941
misc: replaced all "dummy" by the more conventional and readable "__" ("_" being used for gettext)
Goffi <goffi@goffi.org>
parents:
2624
diff
changeset
|
85 lambda __: self.assertEqualXML( |
|
2624
56f94936df1e
code style reformatting using black
Goffi <goffi@goffi.org>
parents:
2562
diff
changeset
|
86 self.host.getSentMessageXml(0), output, True |
|
56f94936df1e
code style reformatting using black
Goffi <goffi@goffi.org>
parents:
2562
diff
changeset
|
87 ) |
|
56f94936df1e
code style reformatting using black
Goffi <goffi@goffi.org>
parents:
2562
diff
changeset
|
88 ) |
| 1274 | 89 return d |