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_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')
