class TestMyGatewayQuery(unittest.TestCase): def setUp(self): self.query = MyGatewayQuery(parent=Iq(from_='from', to='to', type_='get'), host=hostEmul(desc='desc', prompt='prompt') ) def test_getHandler(self): res = self.query.getHandler() host = self.query.host self.assertTrue(isinstance(res.parent, Iq)) self.assertEqual(res.desc, host.desc) self.assertEqual(res.prompt, host.prompt) self.assertEqual(res.from_, self.query.to) self.assertEqual(res.to, self.query.from_) def test_setHandler(self): self.query.iq.type_ = 'set' self.query.prompt = 'jid' res = self.query.setHandler() jid = MyJID.escaped(self.query.prompt, self.query.iq.to) self.assertTrue(isinstance(res.parent, Iq)) self.assertEqual(res.jid, jid) self.assertEqual(res.from_, self.query.to) self.assertEqual(res.to, self.query.from_)
def setUp(self): self.query = MyGatewayQuery(parent=Iq(from_='from', to='to', type_='get'), host=hostEmul(desc='desc', prompt='prompt') )