def test_context_add_2_listeners(self):
ws = self.test_ws
OTHER_WS = WebSocket(self.mock_socket, self.environ)
ctx = self.ctx
ctx.add_listener(ws)
ctx.add_listener(OTHER_WS)
eq_(len(ctx.listeners), 2)
def setUp(self):
self.mock_socket = s = mock.Mock()
self.environ = env = dict(HTTP_ORIGIN='http://localhost', HTTP_WEBSOCKET_PROTOCOL='ws',
PATH_INFO='test')
self.test_ws = WebSocket(s, env)
super(TestWebSocketObject, self).setUp()
def setUp(self):
self.ctx = WebSocketAwareResource()
self.mock_socket = s = mock.Mock()
self.environ = env = dict(HTTP_ORIGIN='http://localhost',
HTTP_WEBSOCKET_PROTOCOL='ws',
PATH_INFO='test')
self.test_ws = WebSocket(s, env)
def test_context_add_listeners(self):
ws = self.test_ws
OTHER_WS = WebSocket(self.mock_socket, self.environ)
ctx = self.ctx
eq_(ctx.listeners, set())
ctx.add_listener(ws)
eq_(ctx.listeners, set([ws]))
ctx.add_listener(ws)
eq_(ctx.listeners, set([ws]))
ctx.remove_listener(OTHER_WS)
eq_(ctx.listeners, set([ws]))
ctx.remove_listener(ws)
eq_(ctx.listeners, set())
