Activate ad-hoc commands support

darcs-hash:20070711061314-86b55-fb79a117909a5eb12fa81b251bd94d6f3dc78ed3.gz
This commit is contained in:
David Rousselie
2007-07-11 08:13:14 +02:00
parent fcf3a1bdbf
commit 721a33d5d5
2 changed files with 7 additions and 4 deletions

View File

@@ -20,13 +20,13 @@
## Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
##
from jcl.jabber.disco import RootDiscoGetInfoHandler
from jcl.jabber.command import CommandRootDiscoGetInfoHandler
class MailRootDiscoGetInfoHandler(RootDiscoGetInfoHandler):
class MailRootDiscoGetInfoHandler(CommandRootDiscoGetInfoHandler):
def handle(self, stanza, lang_class, node, disco_obj, data):
"""Add jabber:iq:gateway support"""
disco_infos = RootDiscoGetInfoHandler.handle(self, stanza, lang_class,
node, disco_obj, data)
disco_infos = CommandRootDiscoGetInfoHandler.handle(self, stanza, lang_class,
node, disco_obj, data)
disco_infos[0].add_feature("jabber:iq:gateway")
disco_infos[0].add_identity(self.component.name, "headline", "newmail")
return disco_infos

View File

@@ -50,6 +50,9 @@ class MailRootDiscoGetInfoHandler_TestCase(unittest.TestCase):
# stanza, lang_class, node, disco_obj, data
disco_infos = handler.handle(None, None, None, None, None)
self.assertTrue(disco_infos[0].has_feature("jabber:iq:gateway"))
self.assertTrue(disco_infos[0].has_feature("http://jabber.org/protocol/disco#info"))
self.assertTrue(disco_infos[0].has_feature("http://jabber.org/protocol/disco#items"))
self.assertTrue(disco_infos[0].has_feature("http://jabber.org/protocol/commands"))
self.assertEquals(len(disco_infos[0].get_identities()), 2)
self.assertTrue(disco_infos[0].identity_is("gateway", "smtp"))
self.assertTrue(disco_infos[0].identity_is("headline", "newmail"))