1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
|
import unittest
from slixmpp import ComponentXMPP, Iq, Message
from slixmpp.roster import RosterItem
from slixmpp.test import SlixTest
class TestPermissions(SlixTest):
def setUp(self):
self.stream_start(
mode="component",
plugins=["xep_0356"],
jid="pubsub.capulet.lit",
server="capulet.net",
)
def testPluginEnd(self):
exc = False
try:
self.xmpp.plugin.disable("xep_0356")
except Exception as e:
exc = True
self.assertFalse(exc)
def testGrantedPrivileges(self):
# https://xmpp.org/extensions/xep-0356.html#example-4
results = {"event": False}
self.xmpp.add_event_handler(
"privileges_advertised", lambda msg: results.__setitem__("event", True)
)
self.recv(
"""
<message from='capulet.net' to='pubub.capulet.lit' id='54321'>
<privilege xmlns='urn:xmpp:privilege:1'>
<perm access='roster' type='both'/>
<perm access='message' type='outgoing'/>
</privilege>
</message>
"""
)
self.assertEqual(self.xmpp["xep_0356"].granted_privileges["roster"], "both")
self.assertEqual(
self.xmpp["xep_0356"].granted_privileges["message"], "outgoing"
)
self.assertEqual(self.xmpp["xep_0356"].granted_privileges["presence"], "none")
self.assertTrue(results["event"])
def testGetRosterIq(self):
iq = self.xmpp["xep_0356"]._make_get_roster("juliet@example.com")
xmlstring = """
<iq xmlns="jabber:component:accept"
id='1'
from='pubsub.capulet.lit'
to='juliet@example.com'
type='get'>
<query xmlns='jabber:iq:roster'/>
</iq>
"""
self.check(iq, xmlstring, use_values=False)
def testSetRosterIq(self):
jid = "juliet@example.com"
items = {
"friend1@example.com": {
"name": "Friend 1",
"subscription": "both",
"groups": ["group1", "group2"],
},
"friend2@example.com": {
"name": "Friend 2",
"subscription": "from",
"groups": ["group3"],
},
}
iq = self.xmpp["xep_0356"]._make_set_roster(jid, items)
xmlstring = f"""
<iq xmlns="jabber:component:accept"
id='1'
from='pubsub.capulet.lit'
to='{jid}'
type='set'>
<query xmlns='jabber:iq:roster'>
<item name='Friend 1' jid='friend1@example.com' subscription='both'>
<group>group1</group>
<group>group2</group>
</item>
<item name='Friend 2' jid='friend2@example.com' subscription='from'>
<group>group3</group>
</item>
</query>
</iq>
"""
self.check(iq, xmlstring, use_values=False)
def testMakeOutgoingMessage(self):
xmlstring = """
<message xmlns="jabber:component:accept" from='pubsub.capulet.lit' to='capulet.net'>
<privilege xmlns='urn:xmpp:privilege:1'>
<forwarded xmlns='urn:xmpp:forward:0'>
<message from="juliet@capulet.lit" to="romeo@montague.lit" xmlns="jabber:client">
<body>I do not hate you</body>
</message>
</forwarded>
</privilege>
</message>
"""
msg = Message()
msg["from"] = "juliet@capulet.lit"
msg["to"] = "romeo@montague.lit"
msg["body"] = "I do not hate you"
priv_msg = self.xmpp["xep_0356"]._make_privileged_message(msg)
self.check(priv_msg, xmlstring, use_values=False)
suite = unittest.TestLoader().loadTestsFromTestCase(TestPermissions)
|
