def test_sub_command(self): client = SentinelClient(io_loop=self.io_loop) result = {"message_count": 0} conn = SentinelClient(io_loop=self.io_loop) self.connect(self.stop, client) self.connect(self.stop, conn) response = yield gen.Task(client.subscribe, "foobar") if response[0] == "subscribe": result["sub"] = response yield gen.Task(conn.publish, "foobar", "new message!") elif response[0] == "message": result["message_count"] += 1 if result["message_count"] < 100: count = result["message_count"] value = yield gen.Task(conn.publish, "foobar", "new message %s!" % count) result["message"] = response[2] self.assertTrue("sub" in result) self.assertTrue("message" in result) self.assertTrue(result["message"], "new message 99!")
def setUp(self): super(TestSentinelClient, self).setUp() self.client = SentinelClient(io_loop=self.io_loop)