def test_cleanup(self): config = { "release_id": 42, "network_provider": "provider", "deployment_mode": "mode", "net_segment_type": "type", "resource_management_workers": 3 } context = { "task": { "uuid": "some_id" }, "config": { "fuel_environments": config }, "fuel": { "environments": ["id", "id", "id"] } } env_ctx = fuel.FuelEnvGenerator(context) env_ctx._delete_envs = mock.Mock() env_ctx.cleanup() env_ctx._delete_envs.assert_called_once_with()
def test_setup_error(self, mock_fuel_scenario, mock__create_envs): context = {} context["config"] = {"fuel_environments": {"environments": 5}} context["task"] = {"uuid": "some_uuid"} context["admin"] = {"endpoint": "some_endpoint"} env_ctx = fuel.FuelEnvGenerator(context) self.assertRaises(exceptions.ContextSetupFailure, env_ctx.setup)
def test_setup(self, mock_fuel_scenario, mock__create_envs): context = {} context["config"] = {"fuel_environments": {"environments": 1}} context["task"] = {"uuid": "some_uuid"} context["admin"] = {"endpoint": "some_endpoint"} env_ctx = fuel.FuelEnvGenerator(context) env_ctx.setup() self.assertIn("fuel", env_ctx.context) self.assertIn("environments", env_ctx.context["fuel"]) mock__create_envs.assert_called_once_with() mock_fuel_scenario.assert_called_once_with(context)
def test__create_envs(self): config = { "environments": 4, "release_id": 42, "network_provider": "provider", "deployment_mode": "mode", "net_segment_type": "type", "resource_management_workers": 3 } context = {"task": {}, "config": {"fuel_environments": config}} env_ctx = fuel.FuelEnvGenerator(context) env_ctx.fscenario = mock.Mock() env_ctx.fscenario.return_value._create_environment.return_value = "id" self.assertEqual(config["environments"], len(env_ctx._create_envs())) enves = config.pop("environments") config.pop("resource_management_workers") exp_calls = [mock.call(**config) for i in range(enves)] self.assertEqual(exp_calls, env_ctx.fscenario._create_environment.mock_calls)
def test__delete_envs(self): config = { "release_id": 42, "network_provider": "provider", "deployment_mode": "mode", "net_segment_type": "type", "resource_management_workers": 3 } context = { "task": {}, "config": { "fuel_environments": config }, "fuel": { "environments": ["id", "id", "id"] } } env_ctx = fuel.FuelEnvGenerator(context) env_ctx.fscenario = mock.Mock() env_ctx._delete_envs() self.assertEqual({}, context["fuel"])