def test_delete_execution(self): scenario = utils.MistralScenario(context=self.context) execution = fakes.FakeMistralClient().execution.create() scenario._delete_execution(execution) self.clients("mistral").executions.delete.assert_called_once_with( execution.id) self._test_atomic_action_timer(scenario.atomic_actions(), "mistral.delete_execution")
def test_mistral(self): fake_mistral = fakes.FakeMistralClient() mock_mistral = mock.Mock() mock_mistral.client.client.return_value = fake_mistral self.assertNotIn("mistral", self.clients.cache) with mock.patch.dict("sys.modules", { "mistralclient": mock_mistral, "mistralclient.api": mock_mistral }): client = self.clients.mistral() self.assertEqual(fake_mistral, client) self.service_catalog.url_for.assert_called_once_with( service_type="workflowv2", region_name=self.credential.region_name) fake_mistral_url = self.service_catalog.url_for.return_value mock_mistral.client.client.assert_called_once_with( mistral_url=fake_mistral_url, service_type="workflowv2", auth_token=self.auth_ref.auth_token) self.assertEqual(fake_mistral, self.clients.cache["mistral"])