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)
