def test_getDiscoItems_badNodeID(self):
csm = CollabSystemManager(self.testJid)
csm.menu.getChildItems = Mock(side_effect=KeyError)
actual = csm.getDiscoItems(jid.JID('*****@*****.**'), self.testJid, 'the node')
self.assertEquals(actual, [])
csm.menu.getChildItems.assert_called_with('the node')
def test_getDiscoItems_withNodeID(self):
csm = CollabSystemManager(self.testJid)
items = ['item1', 'item2']
csm.menu.getChildItems = Mock(return_value=items)
actual = csm.getDiscoItems(jid.JID('*****@*****.**'), self.testJid, 'the node')
self.assertEquals(actual, items)
csm.menu.getChildItems.assert_called_with('the node')
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')
def test_getDiscoItems_badTarget(self):
csm = CollabSystemManager(self.testJid)
items = csm.getDiscoItems(jid.JID('*****@*****.**'), jid.JID('*****@*****.**'))
self.assertEquals(items, [])
def test_getDiscoInfo_badTarget(self):
csm = CollabSystemManager(self.testJid)
info = csm.getDiscoInfo(jid.JID('*****@*****.**'), jid.JID('*****@*****.**'))
self.assertTrue(info is None)
