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