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_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