diff --git a/cognee/tests/unit/api/test_ontology_endpoint.py b/cognee/tests/unit/api/test_ontology_endpoint.py
index d53c5ab44..af3a4d90e 100644
--- a/cognee/tests/unit/api/test_ontology_endpoint.py
+++ b/cognee/tests/unit/api/test_ontology_endpoint.py
@@ -25,7 +25,10 @@ def mock_user():
def mock_default_user():
"""Mock default user for testing."""
return SimpleNamespace(
- id=uuid.uuid4(), email="default@example.com", is_active=True, tenant_id=uuid.uuid4()
+ id=str(uuid.uuid4()),
+ email="default@example.com",
+ is_active=True,
+ tenant_id=str(uuid.uuid4()),
)
@@ -108,6 +111,7 @@ def test_upload_multiple_ontologies(mock_get_default_user, client, mock_default_
"""Test uploading multiple ontology files in single request"""
import io
+ mock_get_default_user.return_value = mock_default_user
# Create mock files
file1_content = b""
file2_content = b""
@@ -137,6 +141,7 @@ def test_upload_endpoint_accepts_arrays(mock_get_default_user, client, mock_defa
import io
import json
+ mock_get_default_user.return_value = mock_default_user
file_content = b""
files = [("ontology_file", ("single.owl", io.BytesIO(file_content), "application/xml"))]
@@ -173,6 +178,7 @@ def test_complete_multifile_workflow(mock_get_default_user, client, mock_default
import io
import json
+ mock_get_default_user.return_value = mock_default_user
# Step 1: Upload multiple ontologies
file1_content = b"""