diff options
Diffstat (limited to 'python_moondb/tests/unit_python/test_pdp.py')
-rwxr-xr-x | python_moondb/tests/unit_python/test_pdp.py | 108 |
1 files changed, 66 insertions, 42 deletions
diff --git a/python_moondb/tests/unit_python/test_pdp.py b/python_moondb/tests/unit_python/test_pdp.py index 942d98a3..4d245e4d 100755 --- a/python_moondb/tests/unit_python/test_pdp.py +++ b/python_moondb/tests/unit_python/test_pdp.py @@ -1,125 +1,149 @@ import pytest - - -def update_pdp(pdp_id, value): - from python_moondb.core import PDPManager - return PDPManager.update_pdp("", pdp_id, value) - - -def delete_pdp(pdp_id): - from python_moondb.core import PDPManager - PDPManager.delete_pdp("", pdp_id) - - -def add_pdp(pdp_id=None, value=None): - from python_moondb.core import PDPManager - return PDPManager.add_pdp("", pdp_id, value) - - -def get_pdp(pdp_id=None): - from python_moondb.core import PDPManager - return PDPManager.get_pdp("", pdp_id) +import helpers.mock_data as mock_data +import helpers.pdp_helper as pdp_helper def test_update_pdp(db): pdp_id = "pdp_id1" + subject_category_id, object_category_id, action_category_id, meta_rule_id, policy_id = mock_data.create_new_policy( + subject_category_name="subject_category1", + object_category_name="object_category1", + action_category_name="action_category1", + meta_rule_name="meta_rule_1", + model_name="model1") value = { "name": "test_pdp", - "security_pipeline": ["policy_id_1", "policy_id_2"], + "security_pipeline": [policy_id], "keystone_project_id": "keystone_project_id1", "description": "...", } - add_pdp(pdp_id, value) - pdp = update_pdp(pdp_id, value) + pdp_helper.add_pdp(pdp_id, value) + pdp = pdp_helper.update_pdp(pdp_id, value) assert pdp def test_update_pdp_with_invalid_id(db): pdp_id = "pdp_id1" + subject_category_id, object_category_id, action_category_id, meta_rule_id, policy_id = mock_data.create_new_policy( + subject_category_name="subject_category1", + object_category_name="object_category1", + action_category_name="action_category1", + meta_rule_name="meta_rule_1", + model_name="model1") value = { "name": "test_pdp", - "security_pipeline": ["policy_id_1", "policy_id_2"], + "security_pipeline": [policy_id], "keystone_project_id": "keystone_project_id1", "description": "...", } with pytest.raises(Exception) as exception_info: - update_pdp(pdp_id, value) + pdp_helper.update_pdp(pdp_id, value) assert str(exception_info.value) == '400: Pdp Unknown' def test_delete_pdp(db): pdp_id = "pdp_id1" + subject_category_id, object_category_id, action_category_id, meta_rule_id, policy_id = mock_data.create_new_policy( + subject_category_name="subject_category1", + object_category_name="object_category1", + action_category_name="action_category1", + meta_rule_name="meta_rule_1", + model_name="model1") value = { "name": "test_pdp", - "security_pipeline": ["policy_id_1", "policy_id_2"], + "security_pipeline": [policy_id], "keystone_project_id": "keystone_project_id1", "description": "...", } - add_pdp(pdp_id, value) - delete_pdp(pdp_id) - assert len(get_pdp(pdp_id)) == 0 + pdp_helper.add_pdp(pdp_id, value) + pdp_helper.delete_pdp(pdp_id) + assert len(pdp_helper.get_pdp(pdp_id)) == 0 def test_delete_pdp_with_invalid_id(db): pdp_id = "pdp_id1" with pytest.raises(Exception) as exception_info: - delete_pdp(pdp_id) + pdp_helper.delete_pdp(pdp_id) assert str(exception_info.value) == '400: Pdp Unknown' def test_add_pdp(db): pdp_id = "pdp_id1" + subject_category_id, object_category_id, action_category_id, meta_rule_id, policy_id = mock_data.create_new_policy( + subject_category_name="subject_category1", + object_category_name="object_category1", + action_category_name="action_category1", + meta_rule_name="meta_rule_1", + model_name="model1") value = { "name": "test_pdp", - "security_pipeline": ["policy_id_1", "policy_id_2"], + "security_pipeline": [policy_id], "keystone_project_id": "keystone_project_id1", "description": "...", } - pdp = add_pdp(pdp_id, value) + pdp = pdp_helper.add_pdp(pdp_id, value) assert pdp def test_add_pdp_twice_with_same_id(db): pdp_id = "pdp_id1" + subject_category_id, object_category_id, action_category_id, meta_rule_id, policy_id = mock_data.create_new_policy( + subject_category_name="subject_category1", + object_category_name="object_category1", + action_category_name="action_category1", + meta_rule_name="meta_rule_1", + model_name="model1") value = { "name": "test_pdp", - "security_pipeline": ["policy_id_1", "policy_id_2"], + "security_pipeline": [policy_id], "keystone_project_id": "keystone_project_id1", "description": "...", } - add_pdp(pdp_id, value) + pdp_helper.add_pdp(pdp_id, value) with pytest.raises(Exception) as exception_info: - add_pdp(pdp_id, value) + pdp_helper.add_pdp(pdp_id, value) assert str(exception_info.value) == '409: Pdp Error' def test_add_pdp_twice_with_same_name(db): + subject_category_id, object_category_id, action_category_id, meta_rule_id, policy_id = mock_data.create_new_policy( + subject_category_name="subject_category1", + object_category_name="object_category1", + action_category_name="action_category1", + meta_rule_name="meta_rule_1", + model_name="model1") value = { "name": "test_pdp", - "security_pipeline": ["policy_id_1", "policy_id_2"], + "security_pipeline": [policy_id], "keystone_project_id": "keystone_project_id1", "description": "...", } - add_pdp(value=value) + pdp_helper.add_pdp(value=value) with pytest.raises(Exception) as exception_info: - add_pdp(value=value) + pdp_helper.add_pdp(value=value) assert str(exception_info.value) == '409: Pdp Error' def test_get_pdp(db): pdp_id = "pdp_id1" + subject_category_id, object_category_id, action_category_id, meta_rule_id, policy_id = mock_data.create_new_policy( + subject_category_name="subject_category1", + object_category_name="object_category1", + action_category_name="action_category1", + meta_rule_name="meta_rule_1", + model_name="model1") value = { "name": "test_pdp", - "security_pipeline": ["policy_id_1", "policy_id_2"], + "security_pipeline": [policy_id], "keystone_project_id": "keystone_project_id1", "description": "...", } - add_pdp(pdp_id, value) - pdp = get_pdp(pdp_id) + pdp_helper.add_pdp(pdp_id, value) + pdp = pdp_helper.get_pdp(pdp_id) assert len(pdp) == 1 def test_get_pdp_with_invalid_id(db): pdp_id = "invalid" - pdp = get_pdp(pdp_id) + pdp = pdp_helper.get_pdp(pdp_id) assert len(pdp) == 0 |