def test_setup(self, mock_osclients, mock_flavor_resource_type_transform,
mock_image_resource_type_transform,
mock_nova_scenario__boot_servers):
tenants_count = 2
users_per_tenant = 5
servers_per_tenant = 5
tenants = self._gen_tenants(tenants_count)
users = []
for id_ in tenants.keys():
for i in range(users_per_tenant):
users.append({
"id": i,
"tenant_id": id_,
"endpoint": mock.MagicMock()
})
real_context = {
"config": {
"users": {
"tenants": 2,
"users_per_tenant": 5,
"concurrent": 10,
},
"servers": {
"servers_per_tenant": 5,
"image": {
"name": "cirros-0.3.4-x86_64-uec",
},
"flavor": {
"name": "m1.tiny",
},
},
},
"admin": {
"endpoint": mock.MagicMock()
},
"task": mock.MagicMock(),
"users": users,
"tenants": tenants
}
new_context = copy.deepcopy(real_context)
for id_ in new_context["tenants"]:
new_context["tenants"][id_].setdefault("servers", [])
for i in range(servers_per_tenant):
new_context["tenants"][id_]["servers"].append("uuid")
servers_ctx = servers.ServerGenerator(real_context)
servers_ctx.setup()
self.assertEqual(new_context, real_context)
def test_cleanup(self, mock_cleanup, mock_osclients):
tenants_count = 2
users_per_tenant = 5
servers_per_tenant = 5
tenants = self._gen_tenants(tenants_count)
users = []
for id_ in tenants.keys():
for i in range(users_per_tenant):
users.append({
"id": i,
"tenant_id": id_,
"credential": "credential"
})
tenants[id_].setdefault("servers", [])
for j in range(servers_per_tenant):
tenants[id_]["servers"].append("uuid")
self.context.update({
"config": {
"users": {
"tenants": 2,
"users_per_tenant": 5,
"concurrent": 10,
},
"servers": {
"servers_per_tenant": 5,
"image": {
"name": "cirros-0.3.4-x86_64-uec",
},
"flavor": {
"name": "m1.tiny",
},
},
},
"admin": {
"credential": mock.MagicMock()
},
"users": users,
"tenants": tenants
})
servers_ctx = servers.ServerGenerator(self.context)
servers_ctx.cleanup()
mock_cleanup.assert_called_once_with(
names=["nova.servers"],
users=self.context["users"],
superclass=nova_utils.NovaScenario,
task_id=self.context["owner_id"])
def test_init(self):
tenants_count = 2
servers_per_tenant = 5
self.context.update({
"config": {
"servers": {
"servers_per_tenant": servers_per_tenant,
}
},
"tenants": self._gen_tenants(tenants_count)
})
inst = servers.ServerGenerator(self.context)
self.assertEqual(self.context["config"]["servers"], inst.config)
def test_init(self):
tenants_count = 2
servers_per_tenant = 5
context = {}
context["task"] = mock.MagicMock()
context["config"] = {
"servers": {
"servers_per_tenant": servers_per_tenant,
}
}
context["tenants"] = self._gen_tenants(tenants_count)
inst = servers.ServerGenerator(context)
self.assertEqual(context["config"]["servers"], inst.config)
def test_init(self):
tenants_count = 2
servers_per_tenant = 5
self.context.update({
"config": {
"servers": {
"servers_per_tenant": servers_per_tenant,
}
},
"tenants": self._gen_tenants(tenants_count)
})
inst = servers.ServerGenerator(self.context)
self.assertEqual({
"auto_assign_nic": False,
"servers_per_tenant": 5
}, inst.config)
def test_setup(self, mock_osclients, mock_flavor_resource_type_transform,
mock_image_resource_type_transform,
mock_nova_scenario__boot_servers):
tenants_count = 2
users_per_tenant = 5
servers_per_tenant = 5
tenants = self._gen_tenants(tenants_count)
users = []
for id_ in tenants.keys():
for i in range(users_per_tenant):
users.append({
"id": i,
"tenant_id": id_,
"credential": mock.MagicMock()
})
self.context.update({
"config": {
"users": {
"tenants": 2,
"users_per_tenant": 5,
"concurrent": 10,
},
"servers": {
"auto_assign_nic": True,
"servers_per_tenant": 5,
"image": {
"name": "cirros-0.3.4-x86_64-uec",
},
"flavor": {
"name": "m1.tiny",
},
},
},
"admin": {
"credential": mock.MagicMock()
},
"users": users,
"tenants": tenants
})
new_context = copy.deepcopy(self.context)
for id_ in new_context["tenants"]:
new_context["tenants"][id_].setdefault("servers", [])
for i in range(servers_per_tenant):
new_context["tenants"][id_]["servers"].append("uuid")
servers_ctx = servers.ServerGenerator(self.context)
servers_ctx.setup()
self.assertEqual(new_context, self.context)
image_id = mock_image_resource_type_transform.return_value
flavor_id = mock_flavor_resource_type_transform.return_value
servers_ctx_config = self.context["config"]["servers"]
expected_auto_nic = servers_ctx_config.get("auto_assign_nic", False)
expected_requests = servers_ctx_config.get("servers_per_tenant", False)
called_times = len(tenants)
mock_calls = [
mock.call(image_id,
flavor_id,
auto_assign_nic=expected_auto_nic,
requests=expected_requests) for i in range(called_times)
]
mock_nova_scenario__boot_servers.assert_has_calls(mock_calls)