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()
