def test_add_same_model_twice(db):
subject_category_id, object_category_id, action_category_id, meta_rule_id = mock_data.create_new_meta_rule(
subject_category_name="subject_category1",
object_category_name="object_category1",
action_category_name="action_category1",
meta_rule_name="meta_rule_1")
value = {
"name": "model1",
"description": "test",
"meta_rules": [meta_rule_id]
}
# prepare
model_helper.add_model(model_id="model_1", value=value) # add model twice
# act
subject_category_id, object_category_id, action_category_id, meta_rule_id = mock_data.create_new_meta_rule(
subject_category_name="subject_category2",
object_category_name="object_category2",
action_category_name="action_category2",
meta_rule_name="meta_rule_2")
value = {
"name": "model2",
"description": "test",
"meta_rules": [meta_rule_id]
}
with pytest.raises(ModelExisting) as exception_info:
model_helper.add_model(model_id="model_1", value=value)
model_helper.delete_all_models()
assert str(exception_info.value) == '409: Model Error'
def test_add_model_generate_new_uuid(db):
subject_category_id, object_category_id, action_category_id, meta_rule_id1 = mock_data.create_new_meta_rule(
subject_category_name="subject_category3",
object_category_name="object_category3",
action_category_name="action_category3",
meta_rule_name="meta_rule_3")
model_value1 = {
"name": "MLS",
"description": "test",
"meta_rules": [meta_rule_id1]
}
model1 = model_helper.add_model(value=model_value1)
subject_category_id, object_category_id, action_category_id, meta_rule_id2 = mock_data.create_new_meta_rule(
subject_category_name="subject_category4",
object_category_name="object_category4",
action_category_name="action_category4",
meta_rule_name="meta_rule_4")
model_value2 = {
"name": "rbac",
"description": "test",
"meta_rules": [meta_rule_id2]
}
model2 = model_helper.add_model(value=model_value2)
assert list(model1)[0] != list(model2)[0]
model_helper.delete_all_models()
def test_update_model(db):
subject_category_id, object_category_id, action_category_id, meta_rule_id1 = mock_data.create_new_meta_rule(
subject_category_name="subject_category8",
object_category_name="object_category8",
action_category_name="action_category8",
meta_rule_name="meta_rule_8")
# prepare
model_value = {
"name": "MLS",
"description": "test",
"meta_rules": [meta_rule_id1]
}
model = model_helper.add_model(value=model_value)
model_id = list(model)[0]
subject_category_id, object_category_id, action_category_id, meta_rule_id2 = mock_data.create_new_meta_rule(
subject_category_name="subject_category9",
object_category_name="object_category9",
action_category_name="action_category9",
meta_rule_name="meta_rule_9")
new_model_value = {
"name": "MLS2",
"description": "test",
"meta_rules": [meta_rule_id2]
}
# act
model_helper.update_model(model_id=model_id, value=new_model_value)
# assert
model = model_helper.get_models(model_id)
for key in ("name", "meta_rules", "description"):
assert key in model[model_id]
assert model[model_id][key] == new_model_value[key]
model_helper.delete_all_models()
def test_delete_models(db):
subject_category_id, object_category_id, action_category_id, meta_rule_id1 = mock_data.create_new_meta_rule(
subject_category_name="subject_category6",
object_category_name="object_category6",
action_category_name="action_category6",
meta_rule_name="meta_rule_6")
model_value1 = {
"name": "MLS",
"description": "test",
"meta_rules": [meta_rule_id1]
}
model1 = model_helper.add_model(value=model_value1)
subject_category_id, object_category_id, action_category_id, meta_rule_id2 = mock_data.create_new_meta_rule(
subject_category_name="subject_category7",
object_category_name="object_category7",
action_category_name="action_category7",
meta_rule_name="meta_rule_7")
model_value2 = {
"name": "rbac",
"description": "test",
"meta_rules": [meta_rule_id2]
}
model_helper.add_model(value=model_value2)
id = list(model1)[0]
model_helper.delete_models(id)
# assert
models = model_helper.get_models()
assert id not in models
model_helper.delete_all_models()
def add_model(model_id=None, value=None):
from python_moondb.core import ModelManager
if not value:
subject_category_id, object_category_id, action_category_id, meta_rule_id = mock_data.create_new_meta_rule()
name = "MLS" if model_id is None else "MLS " + model_id
value = {
"name": name,
"description": "test",
"meta_rules": [meta_rule_id]
}
return ModelManager.add_model(user_id=None, model_id=model_id, value=value)
def test_add_models_with_same_name_twice(db):
subject_category_id, object_category_id, action_category_id, meta_rule_id = mock_data.create_new_meta_rule(
subject_category_name="subject_category5",
object_category_name="object_category5",
action_category_name="action_category5")
model_value1 = {
"name": "MLS",
"description": "test",
"meta_rules": [meta_rule_id]
}
models = model_helper.add_model(value=model_value1)
assert isinstance(models, dict)
assert models
with pytest.raises(Exception) as exception_info:
model_helper.add_model(value=model_value1)
model_helper.delete_all_models()
assert str(exception_info.value) == '409: Model Error'
def test_add_models(db):
subject_category_id, object_category_id, action_category_id, meta_rule_id = mock_data.create_new_meta_rule(
subject_category_name="subject_category5",
object_category_name="object_category5",
action_category_name="action_category5")
model_value1 = {
"name": "MLS",
"description": "test",
"meta_rules": [meta_rule_id]
}
models = model_helper.add_model(value=model_value1)
assert isinstance(models, dict)
assert models
assert len(models.keys()) == 1
model_id = list(models.keys())[0]
for key in ("name", "meta_rules", "description"):
assert key in models[model_id]
assert models[model_id][key] == model_value1[key]
model_helper.delete_all_models()
def test_add_model_with_existed_meta_rules_list(db):
subject_category_id, object_category_id, action_category_id, meta_rule_id = mock_data.create_new_meta_rule(
subject_category_name=uuid4().hex,
object_category_name=uuid4().hex,
action_category_name=uuid4().hex)
model_value1 = {
"name": uuid4().hex,
"description": "test",
"meta_rules": [meta_rule_id]
}
models = model_helper.add_model(value=model_value1)
assert isinstance(models, dict)
assert models
model_value1 = {
"name": uuid4().hex,
"description": "test",
"meta_rules": [meta_rule_id]
}
with pytest.raises(Exception) as exception_info:
model_helper.add_model(value=model_value1)
model_helper.delete_all_models()
assert str(exception_info.value) == '409: Model Error'
assert str(exception_info.value.description
) == 'Meta Rules List Existed in another Model'
