aboutsummaryrefslogtreecommitdiffstats
path: root/python_moondb/tests/unit_python/helpers/meta_rule_helper.py
blob: 87af250a425ff0a5f8ce0f301f75f7f50a22ec38 (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
# Copyright 2015 Open Platform for NFV Project, Inc. and its contributors
# This software is distributed under the terms and conditions of the 'Apache-2.0'
# license which can be found in the file 'LICENSE' in this package distribution
# or at 'http://www.apache.org/licenses/LICENSE-2.0'.

from helpers import mock_data


def update_meta_rule(meta_rule_id, value=None):
    from python_moondb.core import ModelManager
    if not value:
        action_category_id = mock_data.create_action_category("action_category_id1")
        subject_category_id = mock_data.create_subject_category("subject_category_id1")
        object_category_id = mock_data.create_object_category("object_category_id1")
        value = {
            "name": "MLS_meta_rule",
            "description": "test",
            "subject_categories": [subject_category_id],
            "object_categories": [object_category_id],
            "action_categories": [action_category_id]
        }
    return ModelManager.update_meta_rule(user_id=None, meta_rule_id=meta_rule_id, value=value)


def add_meta_rule(meta_rule_id=None, value=None):
    from python_moondb.core import ModelManager
    if not value:
        action_category_id = mock_data.create_action_category("action_category_id1")
        subject_category_id = mock_data.create_subject_category("subject_category_id1")
        object_category_id = mock_data.create_object_category("object_category_id1")
        value = {
            "name": "MLS_meta_rule",
            "description": "test",
            "subject_categories": [subject_category_id],
            "object_categories": [object_category_id],
            "action_categories": [action_category_id]
        }
    return ModelManager.add_meta_rule(user_id=None, meta_rule_id=meta_rule_id, value=value)


def get_meta_rules(meta_rule_id=None):
    from python_moondb.core import ModelManager
    return ModelManager.get_meta_rules(user_id=None, meta_rule_id=meta_rule_id)


def delete_meta_rules(meta_rule_id=None):
    from python_moondb.core import ModelManager
    ModelManager.delete_meta_rule(user_id=None, meta_rule_id=meta_rule_id)