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()
