def test_calls_create_services_for_on_create(self):
endpoint = factory.make_RegionControllerProcessEndpoint()
rack_controller = factory.make_RackController()
mock_create_for = self.patch(Service.objects, "create_services_for")
RegionRackRPCConnection.objects.create(endpoint=endpoint,
rack_controller=rack_controller)
self.assertThat(mock_create_for, MockCalledOnceWith(rack_controller))
def test_calls_update_rackd_status_on_create(self):
endpoint = factory.make_RegionControllerProcessEndpoint()
rack_controller = factory.make_RackController()
mock_update_rackd_status = self.patch(rack_controller,
"update_rackd_status")
RegionRackRPCConnection.objects.create(endpoint=endpoint,
rack_controller=rack_controller)
self.assertThat(mock_update_rackd_status, MockCalledOnceWith())
def test__calls_create_services_for_on_update(self):
endpoint = factory.make_RegionControllerProcessEndpoint()
rack_controller = factory.make_RackController()
mock_create_for = self.patch(Service.objects, "create_services_for")
connection = RegionRackRPCConnection.objects.create(
endpoint=endpoint, rack_controller=rack_controller)
connection.save()
self.assertThat(
mock_create_for,
MockCallsMatch(call(rack_controller), call(rack_controller)))
def create_region_controller_process_endpoint(self, params=None):
if params is None:
params = {}
return factory.make_RegionControllerProcessEndpoint(**params)
