def test_compose_pinned_cores(self): user = yield deferToDatabase(factory.make_admin) handler = PodHandler(user, {}, None) pod = yield deferToDatabase(self.make_pod_with_hints) # Mock the RPC client. client = MagicMock() mock_getClient = self.patch(pods, "getClientFromIdentifiers") mock_getClient.return_value = succeed(client) # Mock the result of the composed machine. node = yield deferToDatabase(factory.make_Node) orig_init = ComposeMachineForm.__init__ def wrapped_init(obj, *args, **kwargs): self.form = obj return orig_init(obj, *args, **kwargs) self.patch(ComposeMachineForm, "__init__", wrapped_init) mock_compose_machine = self.patch(ComposeMachineForm, "compose") mock_compose_machine.return_value = succeed(node) yield handler.compose({ "id": pod.id, "skip_commissioning": True, "pinned_cores": [1, 2, 4], }) self.assertEqual(self.form.get_value_for("pinned_cores"), [1, 2, 4])
def test__compose(self): user = yield deferToDatabase(factory.make_admin) handler = PodHandler(user, {}, None) pod = yield deferToDatabase(self.make_pod_with_hints) # Mock the RPC client. client = MagicMock() mock_getClient = self.patch(pods, "getClientFromIdentifiers") mock_getClient.return_value = succeed(client) # Mock the result of the composed machine. node = yield deferToDatabase(factory.make_Node) mock_compose_machine = self.patch(ComposeMachineForm, "compose") mock_compose_machine.return_value = succeed(node) observed_data = yield handler.compose( {"id": pod.id, "skip_commissioning": True} ) self.assertEqual(pod.id, observed_data["id"])