def test_getDiscoInfo_withNodeID(self): csm = CollabSystemManager(self.testJid) csm.menu = MagicMock() csm.menu.__getitem__.return_value = 'node info', None, None info = csm.getDiscoInfo(jid.JID('*****@*****.**'), self.testJid, 'the node') csm.menu.__getitem__.assert_called_with('the node') self.assertEquals(info, 'node info')
def test_getDiscoInfo_withUnknownNodeID(self): csm = CollabSystemManager(self.testJid) csm.menu = MagicMock() csm.menu.__getitem__.side_effect = KeyError info = csm.getDiscoInfo(jid.JID('*****@*****.**'), self.testJid, 'unknown node') csm.menu.__getitem__.assert_called_with('unknown node') self.assertTrue(info is None)
def test_getDiscoInfo_badTarget(self): csm = CollabSystemManager(self.testJid) info = csm.getDiscoInfo(jid.JID('*****@*****.**'), jid.JID('*****@*****.**')) self.assertTrue(info is None)