adding more test cases for cache

Change-Id: Ie58d6d7fa18448cef677164b97c7d89ffd69c18a
Signed-off-by: sgdt6900 <rhanafy.ext@orange.com>

3 files changed, 115 insertions, 5 deletions

diff --git a/python_moonutilities/tests/unit_python/mock_repo/__init__.py b/python_moonutilities/tests/unit_python/mock_repo/__init__.py
index 60dfbc3b..fa50edb3 100644
--- a/python_moonutilities/tests/unit_python/mock_repo/__init__.py
+++ b/python_moonutilities/tests/unit_python/mock_repo/__init__.py
@@ -15,8 +15,12 @@ def register_cache(m):
 
     register_urls.register_policy_subject(m, data_mock.shared_ids["policy"]["policy_id_1"])
     register_urls.register_policy_subject_invalid_response(m, data_mock.shared_ids["policy"]["policy_id_invalid_response"])
+
     register_urls.register_policy_object(m, data_mock.shared_ids["policy"]["policy_id_1"])
+    register_urls.register_policy_object_invalid_response(m, data_mock.shared_ids["policy"]["policy_id_invalid_response"])
+
     register_urls.register_policy_action(m, data_mock.shared_ids["policy"]["policy_id_1"])
+    register_urls.register_policy_action_invalid_response(m, data_mock.shared_ids["policy"]["policy_id_invalid_response"])
 
     register_urls.register_policy_subject_assignment(m, data_mock.shared_ids["policy"]["policy_id_1"], data_mock.shared_ids["perimeter"]["perimeter_id_1"])
 
diff --git a/python_moonutilities/tests/unit_python/mock_repo/data.py b/python_moonutilities/tests/unit_python/mock_repo/data.py
index 736d4704..d6f2613a 100644
--- a/python_moonutilities/tests/unit_python/mock_repo/data.py
+++ b/python_moonutilities/tests/unit_python/mock_repo/data.py
@@ -35,6 +35,24 @@ shared_ids = {
     },
     "model": {
         "model_id_1": "model_id_1"
+    },
+    "subject": {
+        "subject_id_1": "subject_id_1",
+        "invalid_subject_id": "invalid_subject_id",
+        "invalid_category_id": "invalid_category_id",
+        "invalid_assignment_id": "invalid_assignment_id"
+    },
+    "object": {
+        "object_id_1": "object_id_1",
+        "invalid_object_id": "invalid_object_id",
+        "invalid_category_id": "invalid_category_id",
+        "invalid_assignment_id": "invalid_assignment_id"
+    },
+    "action": {
+        "action_id_1": "action_id_1",
+        "invalid_action_id": "invalid_action_id",
+        "invalid_category_id": "invalid_category_id",
+        "invalid_assignment_id": "invalid_assignment_id"
     }
 }
 
@@ -94,7 +112,7 @@ subject_mock = {
 }
 
 subject_assignment_mock = {
-    "subject_id_1": {
+    shared_ids["subject"]["subject_id_1"]: {
         "policy_id": shared_ids["policy"]["policy_id_1"],
         "subject_id": "subject_id_1",
         "category_id": shared_ids["category"]["category_id_1"],
@@ -102,6 +120,33 @@ subject_assignment_mock = {
     }
 }
 
+subject_assignment_mock_invalid_subject_id = {
+    shared_ids["subject"]["invalid_subject_id"]: {
+        "policy_id": shared_ids["policy"]["policy_id_1"],
+        "subject_id_invalid": "subject_id_1",
+        "category_id": shared_ids["category"]["category_id_1"],
+        "assignments": ["data_id_1, data_id_2"],
+    }
+}
+
+subject_assignment_mock_invalid_category_id = {
+    shared_ids["subject"]["invalid_category_id"]: {
+        "policy_id": shared_ids["policy"]["policy_id_1"],
+        "subject_id": "subject_id_1",
+        "category_id_invalid": shared_ids["category"]["category_id_1"],
+        "assignments": ["data_id_1, data_id_2"],
+    }
+}
+
+subject_assignment_mock_invalid_assignment_id = {
+    shared_ids["subject"]["invalid_assignment_id"]: {
+        "policy_id": shared_ids["policy"]["policy_id_1"],
+        "subject_id": "subject_id_1",
+        "category_id": shared_ids["category"]["category_id_1"],
+        "assignments_invalid": ["data_id_1, data_id_2"],
+    }
+}
+
 object_mock = {
     shared_ids["policy"]["policy_id_1"]: {
         "object_id": {
@@ -112,11 +157,38 @@ object_mock = {
 }
 
 object_assignment_mock = {
-    "object_id_1": {
+    shared_ids["object"]["object_id_1"]: {
         "policy_id": shared_ids["policy"]["policy_id_1"],
         "object_id": "object_id_1",
         "category_id": shared_ids["category"]["category_id_1"],
-        "assignments": ["data_id_1, data_id_2"],
+        "assignments": ["data_id_1, data_id_2"]
+    }
+}
+
+object_assignment_mock_invalid_object_id = {
+    shared_ids["object"]["invalid_object_id"]: {
+        "policy_id": shared_ids["policy"]["policy_id_1"],
+        "object_id": "object_id_1",
+        "category_id": shared_ids["category"]["category_id_1"],
+        "assignments": ["data_id_1, data_id_2"]
+    }
+}
+
+object_assignment_mock_invalid_category_id = {
+    shared_ids["object"]["invalid_category_id"]: {
+        "policy_id": shared_ids["policy"]["policy_id_1"],
+        "object_id": "object_id_1",
+        "category_id": shared_ids["category"]["category_id_1"],
+        "assignments": ["data_id_1, data_id_2"]
+    }
+}
+
+object_assignment_mock_invalid_assignment_id = {
+    shared_ids["object"]["invalid_assignment_id"]: {
+        "policy_id": shared_ids["policy"]["policy_id_1"],
+        "object_id": "object_id_1",
+        "category_id": shared_ids["category"]["category_id_1"],
+        "assignments": ["data_id_1, data_id_2"]
     }
 }
 
@@ -130,14 +202,42 @@ action_mock = {
 }
 
 action_assignment_mock = {
-    "action_id_1": {
+    shared_ids["action"]["action_id_1"]: {
         "policy_id": shared_ids["policy"]["policy_id_1"],
         "action_id": "action_id_1",
         "category_id": shared_ids["category"]["category_id_1"],
-        "assignments": ["data_id_1, data_id_2"],
+        "assignments": ["data_id_1, data_id_2"]
+    }
+}
+
+action_assignment_mock_invalid_action_id = {
+    shared_ids["action"]["invalid_action_id"]: {
+        "policy_id": shared_ids["policy"]["policy_id_1"],
+        "action_id": "action_id_1",
+        "category_id": shared_ids["category"]["category_id_1"],
+        "assignments": ["data_id_1, data_id_2"]
+    }
+}
+
+action_assignment_mock_invalid_category_id = {
+    shared_ids["action"]["invalid_category_id"]: {
+        "policy_id": shared_ids["policy"]["policy_id_1"],
+        "action_id": "action_id_1",
+        "category_id": shared_ids["category"]["category_id_1"],
+        "assignments": ["data_id_1, data_id_2"]
     }
 }
 
+action_assignment_mock_invalid_assignment_id = {
+    shared_ids["action"]["invalid_assignment_id"]: {
+        "policy_id": shared_ids["policy"]["policy_id_1"],
+        "action_id": "action_id_1",
+        "category_id": shared_ids["category"]["category_id_1"],
+        "assignments": ["data_id_1, data_id_2"]
+    }
+}
+
+
 models_mock = {
     shared_ids["model"]["model_id_1"]: {
         "name": "test_model",
diff --git a/python_moonutilities/tests/unit_python/mock_repo/urls.py b/python_moonutilities/tests/unit_python/mock_repo/urls.py
index ff7976da..4170213c 100644
--- a/python_moonutilities/tests/unit_python/mock_repo/urls.py
+++ b/python_moonutilities/tests/unit_python/mock_repo/urls.py
@@ -106,9 +106,15 @@ def register_policy_subject_invalid_response(m, policy_id):
 def register_policy_object(m, policy_id):
     register_policy_any(m, policy_id, 'objects', data_mock.object_mock[policy_id])
 
+def register_policy_object_invalid_response(m, policy_id):
+    register_policy_any(m, policy_id, 'objects', data_mock.subject_mock[policy_id],'objects_invalid_key')
+
 def register_policy_action(m, policy_id):
     register_policy_any(m, policy_id, 'actions', data_mock.action_mock[policy_id])
 
+def register_policy_action_invalid_response(m, policy_id):
+    register_policy_any(m, policy_id, 'actions', data_mock.subject_mock[policy_id],'actions_invalid_key')
+
 def register_policy_subject_assignment_list(m, policy_id):
     register_policy_any(m, policy_id, 'subject_assignments', data_mock.subject_assignment_mock)