def test_coreHandler_unwatch_doesnt_call_unregister(self):
processId = random.randint(0, 100)
rack_id = random.randint(0, 100)
listener = Mock()
service = RackControllerService(sentinel.ipcWorker, listener)
service.processId = processId
service.coreHandler("sys_core_%d" % processId, "unwatch_%d" % rack_id)
self.assertThat(listener.unregister, MockNotCalled())
def test_coreHandler_raises_ValueError_for_unknown_action(self):
processId = random.randint(0, 100)
rack_id = random.randint(0, 100)
listener = PostgresListenerService()
service = RackControllerService(sentinel.ipcWorker, listener)
service.processId = processId
with ExpectedException(ValueError):
service.coreHandler("sys_core_%d" % processId,
"invalid_%d" % rack_id)
def test_stopService_handles_canceling_startup(self):
listener = PostgresListenerService()
service = RackControllerService(sentinel.ipcWorker, listener)
service.processId = random.randint(0, 100)
service.starting = Deferred()
listener.register(f"sys_core_{service.processId}", service.coreHandler)
yield service.stopService()
self.assertNotIn(f"sys_core_{service.processId}", listener.listeners)
self.assertIsNone(service.starting)
def test_stopService_calls_unregister_for_the_process(self):
processId = random.randint(0, 100)
listener = Mock()
service = RackControllerService(sentinel.ipcWorker, listener)
service.processId = processId
yield service.stopService()
self.assertThat(
listener.unregister,
MockCalledOnceWith("sys_core_%d" % processId, service.coreHandler))
def test_coreHandler_unwatch_doesnt_call_unregister(self):
processId = random.randint(0, 100)
rack_id = random.randint(0, 100)
listener = PostgresListenerService()
service = RackControllerService(sentinel.ipcWorker, listener)
self.assertNotIn(rack_id, service.watching)
listener.register(f"sys_dhcp_{rack_id}", service.dhcpHandler)
service.processId = processId
service.coreHandler("sys_core_%d" % processId, "unwatch_%d" % rack_id)
self.assertEqual({f"sys_dhcp_{rack_id}": [service.dhcpHandler]},
listener.listeners)
def test_stopService_handles_canceling_startup(self):
listener = Mock()
service = RackControllerService(sentinel.ipcWorker, listener)
service.processId = random.randint(0, 100)
service.starting = Deferred()
yield service.stopService()
self.assertThat(
listener.unregister,
MockCalledOnceWith("sys_core_%d" % service.processId,
service.coreHandler))
self.assertIsNone(service.starting)
def test_coreHandler_unwatch_calls_unregister(self):
processId = random.randint(0, 100)
rack_id = random.randint(0, 100)
listener = PostgresListenerService()
service = RackControllerService(sentinel.ipcWorker, listener)
service.processId = processId
service.watching = {rack_id}
service.needsDHCPUpdate = {rack_id}
listener.register(f"sys_dhcp_{rack_id}", service.dhcpHandler)
service.coreHandler("sys_core_%d" % processId, "unwatch_%d" % rack_id)
self.assertNotIn(f"sys_dhcp_{rack_id}", listener.listeners)
self.assertEquals(set(), service.watching)
self.assertEquals(set(), service.needsDHCPUpdate)
def test_coreHandler_watch_doesnt_call_register(self):
processId = random.randint(0, 100)
rack_id = random.randint(0, 100)
listener = Mock()
service = RackControllerService(sentinel.ipcWorker, listener)
service.processId = processId
service.watching = set([rack_id])
mock_startProcessing = self.patch(service, "startProcessing")
service.coreHandler("sys_core_%d" % processId, "watch_%d" % rack_id)
self.assertThat(listener.register, MockNotCalled())
self.assertEquals(set([rack_id]), service.watching)
self.assertEquals(set([rack_id]), service.needsDHCPUpdate)
self.assertThat(mock_startProcessing, MockCalledOnceWith())
def test_coreHandler_unwatch_calls_unregister(self):
processId = random.randint(0, 100)
rack_id = random.randint(0, 100)
listener = Mock()
service = RackControllerService(sentinel.ipcWorker, listener)
service.processId = processId
service.watching = {rack_id}
service.needsDHCPUpdate = {rack_id}
service.coreHandler("sys_core_%d" % processId, "unwatch_%d" % rack_id)
self.assertThat(
listener.unregister,
MockCalledOnceWith("sys_dhcp_%d" % rack_id, service.dhcpHandler))
self.assertEquals(set(), service.watching)
self.assertEquals(set(), service.needsDHCPUpdate)
def test_coreHandler_watch_doesnt_call_register(self):
processId = random.randint(0, 100)
rack_id = random.randint(0, 100)
listener = PostgresListenerService()
service = RackControllerService(sentinel.ipcWorker, listener)
service.processId = processId
service.watching = set([rack_id])
mock_startProcessing = self.patch(service, "startProcessing")
sys_channel = f"sys_core_{processId}"
service.coreHandler(sys_channel, "watch_%d" % rack_id)
self.assertNotIn(sys_channel, listener.listeners)
self.assertNotIn(sys_channel, listener.registeredChannels)
self.assertEquals(set([rack_id]), service.watching)
self.assertEquals(set([rack_id]), service.needsDHCPUpdate)
self.assertThat(mock_startProcessing, MockCalledOnceWith())
def test_coreHandler_watch_calls_register_and_startProcessing(self):
processId = random.randint(0, 100)
rack_id = random.randint(0, 100)
listener = self.make_listener_without_delay()
service = RackControllerService(sentinel.ipcWorker, listener)
service.processId = processId
mock_startProcessing = self.patch(service, "startProcessing")
sys_channel = f"sys_core_{processId}"
service.coreHandler(sys_channel, "watch_%d" % rack_id)
dhcp_channel = f"sys_dhcp_{rack_id}"
self.assertIn(dhcp_channel, listener.listeners)
self.assertIn(service.dhcpHandler, listener.listeners[dhcp_channel])
self.assertEqual(set([rack_id]), service.watching)
self.assertEqual(set([rack_id]), service.needsDHCPUpdate)
self.assertThat(mock_startProcessing, MockCalledOnceWith())
def test_stopService_calls_unregister_for_all_watching(self):
processId = random.randint(0, 100)
watching = {random.randint(0, 100) for _ in range(3)}
listener = Mock()
service = RackControllerService(sentinel.ipcWorker, listener)
service.processId = processId
service.watching = watching
yield service.stopService()
self.assertThat(
listener.unregister,
MockAnyCall("sys_core_%d" % processId, service.coreHandler))
for watch_id in watching:
self.assertThat(
listener.unregister,
MockAnyCall("sys_dhcp_%d" % watch_id, service.dhcpHandler))
def test_stopService_calls_unregister_for_all_watching(self):
processId = random.randint(0, 100)
watching = {random.randint(0, 100) for _ in range(3)}
listener = PostgresListenerService()
service = RackControllerService(sentinel.ipcWorker, listener)
service.processId = processId
service.watching = watching
listener.register(f"sys_core_{processId}", service.coreHandler)
for watch_id in watching:
listener.register(f"sys_dhcp_{watch_id}", service.dhcpHandler)
yield service.stopService()
self.assertNotIn(f"sys_core_{processId}", listener.listeners)
for watch_id in watching:
self.assertNotIn(f"sys_dhcp_{watch_id}", listener.listeners)
