def test_cleanup(self, mock_cleanup):
self.context.update({"users": mock.MagicMock()})
ct_ctx = cluster_templates.ClusterTemplateGenerator(self.context)
ct_ctx.cleanup()
mock_cleanup.assert_called_once_with(
names=["magnum.cluster_templates", "nova.keypairs"],
users=self.context["users"])
def test_cleanup(self, mock_cleanup):
self.context.update({"users": mock.MagicMock()})
ct_ctx = cluster_templates.ClusterTemplateGenerator(self.context)
ct_ctx.cleanup()
mock_cleanup.assert_called_once_with(
names=["magnum.cluster_templates"],
users=self.context["users"],
superclass=magnum_utils.MagnumScenario,
task_id=self.context["owner_id"])
def test_cleanup(self, mock_cleanup):
self.context.update({"users": mock.MagicMock()})
ct_ctx = cluster_templates.ClusterTemplateGenerator(self.context)
ct_ctx.cleanup()
mock_cleanup.assert_has_calls(
(mock.call(names=["nova.keypairs"],
users=self.context["users"],
superclass=nova_utils.NovaScenario,
task_id=self.context["owner_id"]),
mock.call(names=["magnum.cluster_templates"],
users=self.context["users"],
superclass=magnum_utils.MagnumScenario,
task_id=self.context["owner_id"])))
def test_setup(self, mock__create_cluster_template):
tenants_count = 2
users_per_tenant = 5
tenants = self._gen_tenants(tenants_count)
users = []
for ten_id in tenants:
for i in range(users_per_tenant):
users.append({
"id": i,
"tenant_id": ten_id,
"credential": mock.MagicMock()
})
self.context.update({
"config": {
"users": {
"tenants": tenants_count,
"users_per_tenant": users_per_tenant,
"concurrent": 10,
},
"cluster_templates": {
"dns_nameserver": "8.8.8.8",
"external_network_id": "public",
"flavor_id": "m1.small",
"docker_volume_size": 5,
"coe": "kubernetes",
"image_id": "fedora-atomic-latest",
"network_driver": "flannel"
}
},
"users": users,
"tenants": tenants
})
ct_ctx = cluster_templates.ClusterTemplateGenerator(self.context)
ct_ctx.setup()
ct_ctx_config = self.context["config"]["cluster_templates"]
image_id = ct_ctx_config.get("image_id")
external_network_id = ct_ctx_config.get("external_network_id")
dns_nameserver = ct_ctx_config.get("dns_nameserver")
flavor_id = ct_ctx_config.get("flavor_id")
docker_volume_size = ct_ctx_config.get("docker_volume_size")
network_driver = ct_ctx_config.get("network_driver")
coe = ct_ctx_config.get("coe")
mock_calls = [
mock.call(image_id=image_id,
external_network_id=external_network_id,
dns_nameserver=dns_nameserver,
flavor_id=flavor_id,
docker_volume_size=docker_volume_size,
network_driver=network_driver,
coe=coe) for i in range(tenants_count)
]
mock__create_cluster_template.assert_has_calls(mock_calls)
# check that stack ids have been saved in context
for ten_id in self.context["tenants"].keys():
self.assertIsNotNone(
self.context["tenants"][ten_id]["cluster_template"])