def test_add(self):
ac = ActiveConnections()
c = MockConnection()
ac.add(c)
assert c in ac.connections
def test_add(self):
ac = ActiveConnections()
c = MockConnection()
ac.add(c)
assert c in ac.connections
def test_write_message(self, mock_connection):
ac = ActiveConnections()
c = MockConnection()
ac.connections = [c]
ac.write_message({"key": "value"})
mock_connection.assert_called_once_with(json.dumps({"key": "value"}))
def test_write_message(self, mock_connection):
ac = ActiveConnections()
c = MockConnection()
ac.connections = [c]
ac.write_message({'key': 'value'})
mock_connection.assert_called_once_with(json.dumps({'key': 'value'}))
def test_double_add(self):
ac = ActiveConnections()
c = MockConnection()
ac.add(c)
ac.add(c)
assert c in ac.connections
assert len(ac.connections) == 1
def test_double_add(self):
ac = ActiveConnections()
c = MockConnection()
ac.add(c)
ac.add(c)
assert c in ac.connections
assert len(ac.connections) == 1
def test_double_remove(self):
ac = ActiveConnections()
c = MockConnection()
ac.connections = [c]
ac.remove(c)
ac.remove(c)
assert len(ac.connections) == 0
def test_remove(self):
ac = ActiveConnections()
c1 = MockConnection()
c2 = MockConnection()
ac.connections = [c1, c2]
ac.remove(c1)
assert c1 not in ac.connections
assert c2 in ac.connections
def test_double_remove(self):
ac = ActiveConnections()
c = MockConnection()
ac.connections = [c]
ac.remove(c)
ac.remove(c)
assert len(ac.connections) == 0
def test_remove(self):
ac = ActiveConnections()
c1 = MockConnection()
c2 = MockConnection()
ac.connections = [c1, c2]
ac.remove(c1)
assert c1 not in ac.connections
assert c2 in ac.connections
def test_check_origin(self):
ac = ActiveConnections()
app = tornado.web.Application()
app.active_connections = ac
r = MockRequest()
wh = WSHandler(app, r)
assert wh.check_origin(None)
assert wh.check_origin(False)
def test_open_connection(self):
ac = ActiveConnections()
app = tornado.web.Application()
app.active_connections = ac
r = MockRequest()
wh = WSHandler(app, r)
wh.open()
assert wh in ac.connections
def test_con_close(self):
ac = ActiveConnections()
app = tornado.web.Application()
app.active_connections = ac
r = MockRequest()
wh = WSHandler(app, r)
ac.connections.append(wh)
wh.on_close()
assert wh not in ac.connections
def test_init(self):
ac = ActiveConnections()
assert isinstance(ac.connections, type([]))
assert len(ac.connections) == 0
