class TwistedTests(twisted.trial.unittest.TestCase): def setUp(self): super(TwistedTests, self).setUp() self.channel_layer = RedisChannelLayer(expiry=1, group_expiry=2, capacity=5) @defer.inlineCallbacks def test_receive_twisted(self): self.channel_layer.send("sr_test", {"value": "blue"}) self.channel_layer.send("sr_test", {"value": "green"}) self.channel_layer.send("sr_test2", {"value": "red"}) # Get just one first channel, message = yield self.channel_layer.receive_twisted(["sr_test"]) self.assertEqual(channel, "sr_test") self.assertEqual(message, {"value": "blue"}) # And the second channel, message = yield self.channel_layer.receive_twisted(["sr_test"]) self.assertEqual(channel, "sr_test") self.assertEqual(message, {"value": "green"}) # And the other channel with multi select channel, message = yield self.channel_layer.receive_twisted(["sr_test", "sr_test2"]) self.assertEqual(channel, "sr_test2") self.assertEqual(message, {"value": "red"}) def tearDown(self): del self.channel_layer reactor.removeAll() super(TwistedTests, self).tearDown()
class TwistedTests(twisted.trial.unittest.TestCase): def setUp(self): super(TwistedTests, self).setUp() self.channel_layer = RedisChannelLayer(expiry=1, group_expiry=2, capacity=5) @defer.inlineCallbacks def test_receive_many_twisted(self): self.channel_layer.send("sr_test", {"value": "blue"}) self.channel_layer.send("sr_test", {"value": "green"}) self.channel_layer.send("sr_test2", {"value": "red"}) # Get just one first channel, message = yield self.channel_layer.receive_many_twisted(["sr_test"]) self.assertEqual(channel, "sr_test") self.assertEqual(message, {"value": "blue"}) # And the second channel, message = yield self.channel_layer.receive_many_twisted(["sr_test"]) self.assertEqual(channel, "sr_test") self.assertEqual(message, {"value": "green"}) # And the other channel with multi select channel, message = yield self.channel_layer.receive_many_twisted(["sr_test", "sr_test2"]) self.assertEqual(channel, "sr_test2") self.assertEqual(message, {"value": "red"}) def tearDown(self): del self.channel_layer reactor.removeAll() super(TwistedTests, self).tearDown()