示例#1
0
    def test__create_objects(self, mock_clients):
        tenants = 2
        containers_per_tenant = 1
        objects_per_container = 5
        context = {
            "task": {
                "uuid": "id123"
            },
            "tenants": {
                "1001": {
                    "name":
                    "t1_name",
                    "containers": [{
                        "user": {
                            "id": "u1",
                            "tenant_id": "1001",
                            "endpoint": "e1"
                        },
                        "container": "c1",
                        "objects": []
                    }]
                },
                "1002": {
                    "name":
                    "t2_name",
                    "containers": [{
                        "user": {
                            "id": "u2",
                            "tenant_id": "1002",
                            "endpoint": "e2"
                        },
                        "container": "c2",
                        "objects": []
                    }]
                }
            }
        }

        mixin = utils.SwiftObjectMixin()
        objects_list = mixin._create_objects(context, objects_per_container,
                                             1024, 25)

        self.assertEqual(
            tenants * containers_per_tenant * objects_per_container,
            len(objects_list))
        chunk = containers_per_tenant * objects_per_container
        for index, obj in enumerate(sorted(objects_list)):
            offset = int(index / chunk) + 1
            self.assertEqual(str(1000 + offset), obj[0])
            self.assertEqual("c%d" % offset, obj[1])
            self.assertIn("rally_object_", obj[2])

        for tenant_id in context["tenants"]:
            for container in context["tenants"][tenant_id]["containers"]:
                self.assertEqual(objects_per_container,
                                 len(container["objects"]))
                for obj in container["objects"]:
                    self.assertIn("rally_object_", obj)
示例#2
0
    def test__create_objects(self, mock_clients):
        tenants = 2
        containers_per_tenant = 1
        objects_per_container = 5
        context = test.get_test_context()
        context.update({
            "tenants": {
                "1001": {
                    "name":
                    "t1_name",
                    "containers": [{
                        "user": {
                            "id": "u1",
                            "tenant_id": "1001",
                            "credential": mock.MagicMock()
                        },
                        "container": "c1",
                        "objects": []
                    }]
                },
                "1002": {
                    "name":
                    "t2_name",
                    "containers": [{
                        "user": {
                            "id": "u2",
                            "tenant_id": "1002",
                            "credential": mock.MagicMock()
                        },
                        "container": "c2",
                        "objects": []
                    }]
                }
            }
        })

        mixin = utils.SwiftObjectMixin()
        objects_list = mixin._create_objects(context, objects_per_container,
                                             1024, 25)

        self.assertEqual(
            tenants * containers_per_tenant * objects_per_container,
            len(objects_list))
        chunk = containers_per_tenant * objects_per_container
        for index, obj in enumerate(sorted(objects_list)):
            offset = int(index / chunk) + 1
            self.assertEqual(str(1000 + offset), obj[0])
            self.assertEqual("c%d" % offset, obj[1])

        for tenant_id in context["tenants"]:
            for container in context["tenants"][tenant_id]["containers"]:
                self.assertEqual(objects_per_container,
                                 len(container["objects"]))
示例#3
0
    def test__delete_objects(self, mock_clients):
        context = {
            "task": {
                "uuid": "id123"
            },
            "tenants": {
                "1001": {
                    "name":
                    "t1_name",
                    "containers": [{
                        "user": {
                            "id": "u1",
                            "tenant_id": "1001",
                            "endpoint": "e1"
                        },
                        "container": "c1",
                        "objects": ["o1", "o2", "o3"]
                    }]
                },
                "1002": {
                    "name":
                    "t2_name",
                    "containers": [{
                        "user": {
                            "id": "u2",
                            "tenant_id": "1002",
                            "endpoint": "e2"
                        },
                        "container": "c2",
                        "objects": ["o4", "o5", "o6"]
                    }]
                }
            }
        }

        mixin = utils.SwiftObjectMixin()
        mixin._delete_objects(context, 1)

        mock_swift = mock_clients.return_value.swift.return_value
        expected_objects = [("c1", "o1"), ("c1", "o2"), ("c1", "o3"),
                            ("c2", "o4"), ("c2", "o5"), ("c2", "o6")]
        mock_swift.delete_object.assert_has_calls(
            [mock.call(con, obj) for con, obj in expected_objects],
            any_order=True)

        for tenant_id in context["tenants"]:
            for container in context["tenants"][tenant_id]["containers"]:
                self.assertEqual(0, len(container["objects"]))
示例#4
0
    def test__delete_objects(self, mock_clients):
        context = test.get_test_context()
        context.update({
            "tenants": {
                "1001": {
                    "name":
                    "t1_name",
                    "containers": [{
                        "user": {
                            "id": "u1",
                            "tenant_id": "1001",
                            "credential": mock.MagicMock()
                        },
                        "container": "c1",
                        "objects": ["o1", "o2", "o3"]
                    }]
                },
                "1002": {
                    "name":
                    "t2_name",
                    "containers": [{
                        "user": {
                            "id": "u2",
                            "tenant_id": "1002",
                            "credential": mock.MagicMock()
                        },
                        "container": "c2",
                        "objects": ["o4", "o5", "o6"]
                    }]
                }
            }
        })

        mixin = utils.SwiftObjectMixin()
        mixin._delete_objects(context, 1)

        mock_swift = mock_clients.return_value.swift.return_value
        expected_objects = [("c1", "o1"), ("c1", "o2"), ("c1", "o3"),
                            ("c2", "o4"), ("c2", "o5"), ("c2", "o6")]
        mock_swift.delete_object.assert_has_calls(
            [mock.call(con, obj) for con, obj in expected_objects],
            any_order=True)

        for tenant_id in context["tenants"]:
            for container in context["tenants"][tenant_id]["containers"]:
                self.assertEqual(0, len(container["objects"]))
    def test__delete_containers(self, mock_clients):
        context = test.get_test_context()
        context.update({
            "tenants": {
                "1001": {
                    "name":
                    "t1_name",
                    "containers": [{
                        "user": {
                            "id": "u1",
                            "tenant_id": "1001",
                            "endpoint": "e1"
                        },
                        "container": "c1",
                        "objects": []
                    }]
                },
                "1002": {
                    "name":
                    "t2_name",
                    "containers": [{
                        "user": {
                            "id": "u2",
                            "tenant_id": "1002",
                            "endpoint": "e2"
                        },
                        "container": "c2",
                        "objects": []
                    }]
                }
            }
        })

        mixin = utils.SwiftObjectMixin()
        mixin._delete_containers(context, 1)

        mock_swift = mock_clients.return_value.swift.return_value
        expected_containers = ["c1", "c2"]
        mock_swift.delete_container.assert_has_calls(
            [mock.call(con) for con in expected_containers], any_order=True)

        for tenant_id in context["tenants"]:
            self.assertEqual(0,
                             len(context["tenants"][tenant_id]["containers"]))
示例#6
0
    def test__create_containers(self, mock_clients):
        tenants = 2
        containers_per_tenant = 2
        context = {
            "task": {
                "uuid": "id123"
            },
            "tenants": {
                "1001": {
                    "name": "t1_name"
                },
                "1002": {
                    "name": "t2_name"
                }
            },
            "users": [{
                "id": "u1",
                "tenant_id": "1001",
                "endpoint": "e1"
            }, {
                "id": "u2",
                "tenant_id": "1002",
                "endpoint": "e2"
            }]
        }

        mixin = utils.SwiftObjectMixin()
        containers = mixin._create_containers(context, containers_per_tenant,
                                              15)

        self.assertEqual(tenants * containers_per_tenant, len(containers))
        for index, container in enumerate(sorted(containers)):
            offset = int(index / containers_per_tenant) + 1
            self.assertEqual(str(1000 + offset), container[0])
            self.assertIn("rally_container_", container[1])

        for index, tenant_id in enumerate(sorted(context["tenants"]), start=1):
            containers = context["tenants"][tenant_id]["containers"]
            self.assertEqual(containers_per_tenant, len(containers))
            for container in containers:
                self.assertEqual("u%d" % index, container["user"]["id"])
                self.assertEqual("e%d" % index, container["user"]["endpoint"])
                self.assertIn("rally_container_", container["container"])
                self.assertEqual(0, len(container["objects"]))
示例#7
0
    def test__create_containers(self, mock_clients):
        tenants = 2
        containers_per_tenant = 2
        context = test.get_test_context()
        c = [mock.MagicMock(), mock.MagicMock()]
        context.update({
            "tenants": {
                "1001": {
                    "name": "t1_name"
                },
                "1002": {
                    "name": "t2_name"
                }
            },
            "users": [{
                "id": "u1",
                "tenant_id": "1001",
                "credential": c[0]
            }, {
                "id": "u2",
                "tenant_id": "1002",
                "credential": c[1]
            }]
        })

        mixin = utils.SwiftObjectMixin()
        containers = mixin._create_containers(context, containers_per_tenant,
                                              15)

        self.assertEqual(tenants * containers_per_tenant, len(containers))
        for index, container in enumerate(sorted(containers)):
            offset = int(index / containers_per_tenant) + 1
            self.assertEqual(str(1000 + offset), container[0])

        for index, tenant_id in enumerate(sorted(context["tenants"]), start=1):
            containers = context["tenants"][tenant_id]["containers"]
            self.assertEqual(containers_per_tenant, len(containers))
            for container in containers:
                self.assertEqual("u%d" % index, container["user"]["id"])
                self.assertEqual(c[index - 1], container["user"]["credential"])
                self.assertEqual(0, len(container["objects"]))