def test_add(): ac = ActiveConnections() c = MockConnection() ac.add(c) assert c in ac.connections
def test_double_add(): ac = ActiveConnections() c = MockConnection() ac.add(c) ac.add(c) assert c in ac.connections assert len(ac.connections) == 1