jshiffer/slixmpp
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Add registration to components

Browse Source
This commit is contained in:
Nicoco K
2021-02-18 20:05:25 +01:00
committed by mathieui
parent 5d7918b122
commit cef7b5cf05
2 changed files with 264 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

112
tests/test_stream_xep_0077.py Normal file
View File

@@ -0,0 +1,112 @@
"""
This only covers the component registration side of the XEP-0077 plugin
"""
import unittest
from slixmpp import ComponentXMPP, Iq
from slixmpp.xmlstream import register_stanza_plugin
from slixmpp.test import SlixTest
from slixmpp.plugins.xep_0077 import Register
class TestRegistration(SlixTest):
def setUp(self):
self.stream_start(
mode="component", plugins=["xep_0077"], jid="shakespeare.lit", server="lit"
)
def testRegistrationForm(self):
self.stream_start(
mode="component", plugins=["xep_0077"], jid="shakespeare.lit", server="lit"
)
self.recv(
"""
<iq type='get' id='reg1' to='shakespeare.lit' from='bill@server/resource'>
<query xmlns='jabber:iq:register'/>
</iq>
""",
)
self.send(
f"""
<iq type='result' id='reg1' from='shakespeare.lit' to='bill@server/resource'>
<query xmlns='jabber:iq:register'>
<instructions>{self.xmpp["xep_0077"].form_instructions}</instructions>
<username/>
<password/>
</query>
</iq>
""",
use_values=False # Fails inconsistently without this
)
def testRegistrationSuccessAndModif(self):
self.recv(
"""
<iq type='set' id='reg2' to='shakespeare.lit' from="bill@server/resource">
<query xmlns='jabber:iq:register'>
<username>bill</username>
<password>Calliope</password>
</query>
</iq>
"""
)
self.send("<iq type='result' id='reg2' from='shakespeare.lit' to='bill@server/resource'/>")
user_store = self.xmpp["xep_0077"]._user_store
self.assertEqual(user_store["bill@server"]["username"], "bill")
self.assertEqual(user_store["bill@server"]["password"], "Calliope")
self.recv(
"""
<iq type='get' id='reg1' to='shakespeare.lit' from="bill@server/resource">
<query xmlns='jabber:iq:register'/>
</iq>
""",
)
self.send(
f"""
<iq type='result' id='reg1' to="bill@server/resource" from='shakespeare.lit'>
<query xmlns='jabber:iq:register'>
<instructions>{self.xmpp["xep_0077"].form_instructions}</instructions>
<username>bill</username>
<password>Calliope</password>
<registered />
</query>
</iq>
""",
use_values=False # Fails inconsistently without this
)
def testRegistrationAndRemove(self):
self.recv(
"""
<iq type='set' id='reg2' to='shakespeare.lit' from="bill@shakespeare.lit/globe">
<query xmlns='jabber:iq:register'>
<username>bill</username>
<password>Calliope</password>
</query>
</iq>
"""
)
self.send("<iq type='result' id='reg2' from='shakespeare.lit' to='bill@shakespeare.lit/globe'/>")
pseudo_iq = self.xmpp.Iq()
pseudo_iq["from"] = "bill@shakespeare.lit/globe"
user = self.xmpp["xep_0077"].api["user_get"](None, None, None, pseudo_iq)
self.assertEqual(user["username"], "bill")
self.assertEqual(user["password"], "Calliope")
self.recv(
"""
<iq type='set' from='bill@shakespeare.lit/globe' id='unreg1'>
<query xmlns='jabber:iq:register'>
<remove/>
</query>
</iq>
"""
)
self.send("<iq type='result' to='bill@shakespeare.lit/globe' id='unreg1'/>")
user_store = self.xmpp["xep_0077"]._user_store
self.assertIs(user_store.get("bill@shakespeare.lit"), None)
suite = unittest.TestLoader().loadTestsFromTestCase(TestRegistration)