def test_add_subject_assignments(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")
subject_id = mock_data.create_subject(policy_id)
data_id = mock_data.create_subject_data(policy_id=policy_id,
category_id=subject_category_id)
subject_assignments = assignment_helper.add_subject_assignment(
policy_id, subject_id, subject_category_id, data_id)
assert subject_assignments
subject_id_1 = list(subject_assignments.keys())[0]
assert subject_assignments[subject_id_1]["policy_id"] == policy_id
assert subject_assignments[subject_id_1]["subject_id"] == subject_id
assert subject_assignments[subject_id_1][
"category_id"] == subject_category_id
assert len(subject_assignments[subject_id_1].get("assignments")) == 1
assert data_id in subject_assignments[subject_id_1].get("assignments")
with pytest.raises(SubjectAssignmentExisting) as exception_info:
assignment_helper.add_subject_assignment(policy_id, subject_id,
subject_category_id, data_id)
assert str(exception_info.value) == '409: Subject Assignment Existing'
assert str(exception_info.value.description
) == 'The given subject assignment value is existing.'
def test_delete_policies_with_subject_assignment(db):
subject_category_id, object_category_id, action_category_id, meta_rule_id, policy_id = mock_data.create_new_policy()
subject_id = mock_data.create_subject(policy_id)
data_id = mock_data.create_subject_data(policy_id=policy_id, category_id=subject_category_id)
assignment_helper.add_subject_assignment(policy_id, subject_id, subject_category_id, data_id)
with pytest.raises(DeletePolicyWithPerimeter) as exception_info:
policy_helper.delete_policies(policy_id)
assert '400: Policy With Perimeter Error' == str(exception_info.value)
def test_delete_subject_category_with_assignment(db):
subject_category_id, object_category_id, action_category_id, meta_rule_id, policy_id = mock_data.create_new_policy(
)
subject_id = mock_data.create_subject(policy_id)
data_id = mock_data.create_subject_data(policy_id, subject_category_id)
assignment_helper.add_subject_assignment(policy_id, subject_id,
subject_category_id, data_id)
with pytest.raises(DeleteSubjectCategoryWithMetaRule) as exception_info:
category_helper.delete_subject_category(subject_category_id)
assert str(
exception_info.value) == '400: Subject Category With Meta Rule Error'
def test_get_subject_assignments_by_policy_id(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")
subject_id = mock_data.create_subject(policy_id)
data_id = mock_data.create_subject_data(policy_id=policy_id,
category_id=subject_category_id)
assignment_helper.add_subject_assignment(policy_id, subject_id,
subject_category_id, data_id)
subj_assignments = assignment_helper.get_subject_assignments(policy_id)
assert len(subj_assignments) == 1
def test_delete_subject_with_assignment(db):
subject_category_id, object_category_id, action_category_id, meta_rule_id, policy_id = mock_data.create_new_policy(
subject_category_name="subject_category"+uuid4().hex,
object_category_name="object_category"+uuid4().hex,
action_category_name="action_category"+uuid4().hex,
meta_rule_name="meta_rule_"+uuid4().hex)
subject_id = mock_data.create_subject(policy_id)
data_id = mock_data.create_subject_data(policy_id=policy_id, category_id=subject_category_id)
assignment_helper.add_subject_assignment(policy_id, subject_id, subject_category_id, data_id)
with pytest.raises(DeletePerimeterWithAssignment) as exception_info:
data_helper.delete_subject(policy_id, subject_id)
assert '400: Perimeter With Assignment Error' == str(exception_info.value)
def test_get_subject_assignments(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")
subject_id = mock_data.create_subject(policy_id)
data_id = mock_data.create_subject_data(policy_id=policy_id,
category_id=subject_category_id)
assignment_helper.add_subject_assignment(policy_id, subject_id,
subject_category_id, data_id)
subj_assignments = assignment_helper.get_subject_assignments(
policy_id, subject_id, subject_category_id)
subject_id_1 = list(subj_assignments.keys())[0]
assert subj_assignments[subject_id_1]["policy_id"] == policy_id
assert subj_assignments[subject_id_1]["subject_id"] == subject_id
assert subj_assignments[subject_id_1]["category_id"] == subject_category_id
assert len(subj_assignments[subject_id_1].get("assignments")) == 1
assert data_id in subj_assignments[subject_id_1].get("assignments")