def test_removeConnectionFor_is_okay_if_connection_is_not_there(self): service = RegionService(sentinel.ipcWorker) uuid = factory.make_UUID() service._removeConnectionFor(uuid, DummyConnection()) self.assertEqual({uuid: set()}, service.connections)
def test_removeConnectionFor_fires_disconnected_event(self): service = RegionService(sentinel.ipcWorker) uuid = factory.make_UUID() c1 = DummyConnection() mock_fire = self.patch(service.events.disconnected, "fire") service._removeConnectionFor(uuid, c1) self.assertThat(mock_fire, MockCalledOnceWith(uuid))
def test_removeConnectionFor_removes_connection(self): service = RegionService(sentinel.ipcWorker) uuid = factory.make_UUID() c1 = DummyConnection() c2 = DummyConnection() service._addConnectionFor(uuid, c1) service._addConnectionFor(uuid, c2) service._removeConnectionFor(uuid, c1) self.assertEqual({uuid: {c2}}, service.connections)