def test_add_type(self, object_manager): _render_data = { 'summary': [], 'external': [], 'sections': [] } type_instance_new_1 = CmdbType(name='example2', active=True, author_id=1, creation_time=datetime.utcnow(), render_meta=_render_data, fields=[], public_id=2) # CmdbObjectType insert ack = object_manager.insert_type(type_instance_new_1) assert ack == 2 type_instance_new_2 = object_manager.get_type(public_id=2) assert type_instance_new_2.get_name() == 'example2' # dict insert type_instance_new_3 = CmdbType(name='example3', active=True, author_id=1, creation_time=datetime.utcnow(), render_meta=_render_data, fields=[], public_id=3) object_manager.insert_type(type_instance_new_3.__dict__) type_instance_new_3 = object_manager.get_type(public_id=3) assert type_instance_new_3.get_name() == 'example3' # insert error test type_instance_new_4 = CmdbType(name='example3', active=True, author_id=1, creation_time=datetime.utcnow(), render_meta=_render_data, fields=[], public_id=type_instance_new_3.get_public_id()) with pytest.raises(TypeAlreadyExists): object_manager.insert_type(type_instance_new_4)
def test_update_type(self, object_manager): _render_data = { 'summary': [], 'external': [], 'sections': [] } type_instance_update_1 = CmdbType(name='exampleX', active=True, author_id=1, creation_time=datetime.utcnow(), render_meta=_render_data, fields=[], public_id=1) object_manager.update_type(type_instance_update_1) type_get_instance = object_manager.get_type(public_id=1) assert type_get_instance.get_name() == type_instance_update_1.get_name()