Issue #218: add error type

ai-cfia · Nov 29, 2024 · ecdbccd · ecdbccd
1 parent 8ba6ab4
commit ecdbccd
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/fertiscan/db/queries/organization/__init__.py b/fertiscan/db/queries/organization/__init__.py
@@ -308,7 +308,9 @@ def upsert_organization(cursor: Cursor, organization_info_id: UUID):
         SELECT upsert_organization(%s);
         """
     cursor.execute(query, (str(organization_info_id),))
-    return cursor.fetchone()[0]
+    if result := cursor.fetchone():
+        return result[0]
+    raise OrganizationUpdateError("Failed to update Organization. No data returned.")
 
 
 @handle_query_errors(OrganizationRetrievalError)

diff --git a/tests/fertiscan/test_datastore.py b/tests/fertiscan/test_datastore.py
@@ -190,6 +190,7 @@ def test_register_analysis(self):
         ingredients = ingredient.get_ingredient_json(
             self.cursor, str(analysis["product"]["label_id"])
         )
+        self.assertIsNotNone(ingredients)
         # print(ingredients)
 
         # specifications = specification.get_all_specifications(