Play unit tests and modify code accordingly. (some tests are still not OK).

Change-Id: I021081710411be12ffbbb0a3c8192626dbf9f8ce

1 files changed, 81 insertions, 0 deletions

diff --git a/keystone-moon/keystone/tests/moon/unit/__init__.py b/keystone-moon/keystone/tests/moon/unit/__init__.py
index 1b678d53..0cd835ce 100644
--- a/keystone-moon/keystone/tests/moon/unit/__init__.py
+++ b/keystone-moon/keystone/tests/moon/unit/__init__.py
@@ -2,3 +2,84 @@
 # This software is distributed under the terms and conditions of the 'Apache-2.0'
 # license which can be found in the file 'LICENSE' in this package distribution
 # or at 'http://www.apache.org/licenses/LICENSE-2.0'.
+import uuid
+from keystone.contrib.moon.core import ADMIN_ID
+
+USER = {
+    'name': 'admin',
+    'domain_id': "default",
+    'password': 'admin'
+}
+IE = {
+    "name": "test IE",
+    "policymodel": "policy_authz",
+    "description": "a simple description."
+}
+
+
+def create_intra_extension(self, policy_model="policy_authz"):
+
+    IE["model"] = policy_model
+    IE["name"] = uuid.uuid4().hex
+    genre = "admin"
+    if "authz" in policy_model:
+        genre = "authz"
+    IE["genre"] = genre
+    # force re-initialization of the ADMIN_ID variable
+    from keystone.contrib.moon.core import ADMIN_ID
+    self.ADMIN_ID = ADMIN_ID
+    ref = self.admin_api.load_intra_extension_dict(self.ADMIN_ID, intra_extension_dict=IE)
+    self.assertIsInstance(ref, dict)
+    return ref
+
+
+# def create_tenant(self, authz_uuid):
+#     tenant = {
+#         "id": uuid.uuid4().hex,
+#         "name": "TestAuthzIntraExtensionManager",
+#         "enabled": True,
+#         "description": "",
+#         "domain_id": "default"
+#     }
+#     project = self.resource_api.create_project(tenant["id"], tenant)
+#     mapping = self.tenant_api.set_tenant_dict(project["id"], project["name"], authz_uuid, None)
+#     self.assertIsInstance(mapping, dict)
+#     self.assertIn("authz", mapping)
+#     self.assertEqual(mapping["authz"], authz_uuid)
+#     return mapping
+
+
+def create_user(self, username="TestAdminIntraExtensionManagerUser"):
+    user = {
+        "id": uuid.uuid4().hex,
+        "name": username,
+        "enabled": True,
+        "description": "",
+        "domain_id": "default"
+    }
+    _user = self.identity_api.create_user(user)
+    return _user
+
+def create_mapping(self, tenant_name=None, authz_id=None, admin_id=None):
+
+    from keystone.contrib.moon.core import ADMIN_ID
+    if not tenant_name:
+        tenant_name = uuid.uuid4().hex
+
+    tenant = {
+        "id": uuid.uuid4().hex,
+        "name": tenant_name,
+        "description": uuid.uuid4().hex,
+        "intra_authz_extension_id": authz_id,
+        "intra_admin_extension_id": admin_id,
+        "enabled": True,
+        "domain_id": "default"
+    }
+    keystone_tenant = self.resource_api.create_project(tenant["id"], tenant)
+    mapping = self.tenant_api.add_tenant_dict(ADMIN_ID, tenant)
+    self.assertIsInstance(mapping, dict)
+    self.assertIn("intra_authz_extension_id", mapping[tenant["id"]])
+    self.assertIn("intra_admin_extension_id", mapping[tenant["id"]])
+    self.assertEqual(mapping[tenant["id"]]["intra_authz_extension_id"], authz_id)
+    self.assertEqual(mapping[tenant["id"]]["intra_admin_extension_id"], admin_id)
+    return tenant, mapping