Remove call to receive_many

This commit is contained in:
Andrew Godwin 2017-02-11 16:29:04 -08:00
parent ecf88ee72a
commit 9853bf1740

View File

@ -105,7 +105,7 @@ class TestWebSocketProtocol(TestCase):
) )
# Get the resulting message off of the channel layer # Get the resulting message off of the channel layer
_, message = self.channel_layer.receive_many(["websocket.connect"]) _, message = self.channel_layer.receive(["websocket.connect"])
self.assertIn((b'origin', b'file://'), message['headers']) self.assertIn((b'origin', b'file://'), message['headers'])
self.assertTrue(message['reply_channel'].startswith("websocket.send!")) self.assertTrue(message['reply_channel'].startswith("websocket.send!"))
@ -134,7 +134,7 @@ class TestWebSocketProtocol(TestCase):
) )
# Get the resulting message off of the channel layer # Get the resulting message off of the channel layer
_, message = self.channel_layer.receive_many(["websocket.connect"]) _, message = self.channel_layer.receive(["websocket.connect"])
self.assertNotIn(b'origin', [header_tuple[0] for header_tuple in message['headers']]) self.assertNotIn(b'origin', [header_tuple[0] for header_tuple in message['headers']])
self.assertTrue(message['reply_channel'].startswith("websocket.send!")) self.assertTrue(message['reply_channel'].startswith("websocket.send!"))