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"])