aboutsummaryrefslogtreecommitdiffstats
path: root/python_moondb/tests/unit_python/policies/mock_data.py
blob: 3e9bea93d9ecb750e5132115affecc8c23cd3075 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
def create_meta_rule(meta_rule_name="meta_rule1", category_prefix=""):
    meta_rule_value = {
        "name": meta_rule_name,
        "algorithm": "name of the meta rule algorithm",
        "subject_categories": [category_prefix + "subject_category_id1",
                               category_prefix + "subject_category_id2"],
        "object_categories": [category_prefix +"object_category_id1"],
        "action_categories": [category_prefix +"action_category_id1"]
    }
    return meta_rule_value


def create_model(meta_rule_id, model_name="test_model"):
    value = {
        "name": model_name,
        "description": "test",
        "meta_rules": [meta_rule_id]

    }
    return value


def create_policy(model_id, policy_name="policy_1"):
    value = {
        "name": policy_name,
        "model_id": model_id,
        "genre": "authz",
        "description": "test",
    }
    return value


def create_pdp(pdp_ids):
    value = {
        "name": "test_pdp",
        "security_pipeline": pdp_ids,
        "keystone_project_id": "keystone_project_id1",
        "description": "...",
    }
    return value


def get_policy_id(model_name="test_model", policy_name="policy_1", meta_rule_name="meta_rule1", category_prefix=""):
    import policies.test_policies as test_policies
    import models.test_models as test_models
    import models.test_meta_rules as test_meta_rules
    meta_rule = test_meta_rules.add_meta_rule(value=create_meta_rule(meta_rule_name, category_prefix))
    meta_rule_id = list(meta_rule.keys())[0]
    model = test_models.add_model(value=create_model(meta_rule_id, model_name))
    model_id = list(model.keys())[0]
    value = create_policy(model_id, policy_name)
    policy = test_policies.add_policies(value=value)
    assert policy
    policy_id = list(policy.keys())[0]
    return policy_id