From 26b7103afdb796d88646d13075a576a42c7230c3 Mon Sep 17 00:00:00 2001 From: David Rousselie Date: Thu, 2 Nov 2006 19:33:54 +0100 Subject: [PATCH] Better send probe assertions in authentication test darcs-hash:20061102183354-86b55-5967563165e992e0a96ffe1f597028d5f97b25f2.gz --- tests/jcl/jabber/test_component.py | 43 +++++++++++++++++++----------- 1 file changed, 27 insertions(+), 16 deletions(-) diff --git a/tests/jcl/jabber/test_component.py b/tests/jcl/jabber/test_component.py index ee66da4..d263581 100644 --- a/tests/jcl/jabber/test_component.py +++ b/tests/jcl/jabber/test_component.py @@ -271,23 +271,34 @@ class JCLComponent_TestCase(unittest.TestCase): name = "account2", \ jid = "account2@jcl.test.com") del account.hub.threadConnection - self.comp.stream = stream = MockStream() + self.comp.stream = MockStream() self.comp.authenticated() - - presence_sent = stream.sent - self.assertEqual(len(presence_sent), 5) - self.assertEqual(len([presence \ - for presence in presence_sent \ - if presence.get_from_jid() == "jcl.test.com"]), \ - 2) - self.assertEqual(len([presence \ - for presence in presence_sent \ - if presence.get_to_jid() == "test1@test.com"]), \ - 3) - self.assertEqual(len([presence \ - for presence in presence_sent \ - if presence.get_to_jid() == "test2@test.com"]), \ - 2) + self.assertEqual(len(self.comp.stream.sent), 5) + presence = self.comp.stream.sent[0] + self.assertTrue(isinstance(presence, Presence)) + self.assertEquals(presence.get_from(), "jcl.test.com") + self.assertEquals(presence.get_to(), "test1@test.com") + self.assertEquals(presence.get_node().prop("type"), "probe") + presence = self.comp.stream.sent[1] + self.assertTrue(isinstance(presence, Presence)) + self.assertEquals(presence.get_from(), "account11@jcl.test.com") + self.assertEquals(presence.get_to(), "test1@test.com") + self.assertEquals(presence.get_node().prop("type"), "probe") + presence = self.comp.stream.sent[2] + self.assertTrue(isinstance(presence, Presence)) + self.assertEquals(presence.get_from(), "account12@jcl.test.com") + self.assertEquals(presence.get_to(), "test1@test.com") + self.assertEquals(presence.get_node().prop("type"), "probe") + presence = self.comp.stream.sent[3] + self.assertTrue(isinstance(presence, Presence)) + self.assertEquals(presence.get_from(), "jcl.test.com") + self.assertEquals(presence.get_to(), "test2@test.com") + self.assertEquals(presence.get_node().prop("type"), "probe") + presence = self.comp.stream.sent[4] + self.assertTrue(isinstance(presence, Presence)) + self.assertEquals(presence.get_from(), "account2@jcl.test.com") + self.assertEquals(presence.get_to(), "test2@test.com") + self.assertEquals(presence.get_node().prop("type"), "probe") def test_signal_handler(self): self.comp.running = True