def test_dhcpHandler_doesnt_add_to_needsDHCPUpdate(self): rack_id = random.randint(0, 100) listener = PostgresListenerService() service = RackControllerService(sentinel.ipcWorker, listener) mock_startProcessing = self.patch(service, "startProcessing") service.dhcpHandler("sys_dhcp_%d" % rack_id, "") self.assertEquals(set(), service.needsDHCPUpdate) self.assertThat(mock_startProcessing, MockNotCalled())
def test_dhcpHandler_adds_to_needsDHCPUpdate(self): rack_id = random.randint(0, 100) listener = self.make_listener_without_delay() service = RackControllerService(sentinel.ipcWorker, listener) service.watching = set([rack_id]) mock_startProcessing = self.patch(service, "startProcessing") service.dhcpHandler("sys_dhcp_%d" % rack_id, "") self.assertEqual(set([rack_id]), service.needsDHCPUpdate) self.assertThat(mock_startProcessing, MockCalledOnceWith())