def test_get_available_metadata(db):
policy_id = mock_data.get_policy_id()
metadata = get_available_metadata(policy_id=policy_id)
assert metadata
assert metadata['object'][0] == "object_category_id1"
assert metadata['subject'][0] == "subject_category_id1"
assert metadata['subject'][1] == "subject_category_id2"
def test_delete_subject_data(db):
policy_id = mock_data.get_policy_id()
get_available_metadata(policy_id)
data_id = "data_id_1"
category_id = "subject_category_id1"
value = {
"name": "subject-security-level",
"description": {"low": "", "medium": "", "high": ""},
}
subject_data = add_subject_data(policy_id, data_id, category_id, value).get('data')
subject_data_id = list(subject_data.keys())[0]
delete_subject_data(subject_data[subject_data_id].get('policy_id'), data_id)
new_subject_data = get_subject_data(policy_id)
assert len(new_subject_data[0]['data']) == 0
def test_delete_action_data(db):
policy_id = mock_data.get_policy_id()
get_available_metadata(policy_id)
data_id = "data_id_1"
category_id = "category_id_1"
value = {
"name": "action-type",
"description": {"vm-action": "", "storage-action": "", },
}
action_data = add_action_data(policy_id, data_id, category_id, value).get('data')
action_data_id = list(action_data.keys())[0]
delete_action_data(action_data[action_data_id].get('policy_id'), None)
new_action_data = get_action_data(policy_id)
assert len(new_action_data[0]['data']) == 0
def test_get_action_data(db):
policy_id = mock_data.get_policy_id()
get_available_metadata(policy_id)
policy_id = policy_id
data_id = "data_id_1"
category_id = "action_category_id1"
value = {
"name": "action-type",
"description": {"vm-action": "", "storage-action": "", },
}
add_action_data(policy_id, data_id, category_id, value)
action_data = get_action_data(policy_id, data_id, category_id)
assert action_data
assert len(action_data[0]['data']) == 1
def test_get_subject_data_with_invalid_category_id(db):
policy_id = mock_data.get_policy_id()
get_available_metadata(policy_id)
policy_id = policy_id
data_id = "data_id_1"
category_id = "subject_category_id1"
value = {
"name": "subject-security-level",
"description": {"low": "", "medium": "", "high": ""},
}
add_subject_data(policy_id, data_id, category_id, value)
subject_data = get_subject_data(policy_id)
assert subject_data
assert len(subject_data[0]['data']) == 1
