def get_entitytype(conn, name, description):
try:
entitytype = get_entitytype(conn, name)
except:
entitytype = create_entitytype(conn, name, description)
return entitytype
def test_create_entitytype():
entitytype_id = 42
entitytype_name = "Dummy1"
entitytype_descr = "Description of Dummy1"
mock_conn = mock.Mock()
mock_cursor = mock.Mock()
mock_cursor.fetchone.return_value = (entitytype_id, )
mock_conn.cursor.return_value = mock_cursor
entitytype = helpers.create_entitytype(mock_conn, entitytype_name,
entitytype_descr)
assert entitytype.id == entitytype_id
assert entitytype.name == entitytype_name
assert entitytype.description == entitytype_descr
