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