gmakstutis/cognee
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

fix: remove async from ontology endpoint test functions

Browse source
This commit is contained in:
Fahad Shoaib 2025-11-20 15:54:09 +05:00
parent 30e3971d44
commit 8cfb6c41ee
1 changed files with 6 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

12
cognee/tests/unit/api/test_ontology_endpoint.py
View file

@ -104,7 +104,7 @@ def test_upload_ontology_unauthorized(mock_get_default_user, client, mock_defaul
@patch.object(gau_mod, "get_default_user", new_callable=AsyncMock) @patch.object(gau_mod, "get_default_user", new_callable=AsyncMock)
async def test_upload_multiple_ontologies(mock_get_default_user, client, mock_default_user): def test_upload_multiple_ontologies(mock_get_default_user, client, mock_default_user):
"""Test uploading multiple ontology files in single request""" """Test uploading multiple ontology files in single request"""
import io import io
@ -132,7 +132,7 @@ async def test_upload_multiple_ontologies(mock_get_default_user, client, mock_de
@patch.object(gau_mod, "get_default_user", new_callable=AsyncMock) @patch.object(gau_mod, "get_default_user", new_callable=AsyncMock)
async def test_upload_endpoint_accepts_arrays(mock_get_default_user, client, mock_default_user): def test_upload_endpoint_accepts_arrays(mock_get_default_user, client, mock_default_user):
"""Test that upload endpoint accepts array parameters""" """Test that upload endpoint accepts array parameters"""
import io import io
import json import json
@ -153,7 +153,7 @@ async def test_upload_endpoint_accepts_arrays(mock_get_default_user, client, moc
@patch.object(gau_mod, "get_default_user", new_callable=AsyncMock) @patch.object(gau_mod, "get_default_user", new_callable=AsyncMock)
async def test_cognify_with_multiple_ontologies(mock_get_default_user, client, mock_default_user): def test_cognify_with_multiple_ontologies(mock_get_default_user, client, mock_default_user):
"""Test cognify endpoint accepts multiple ontology keys""" """Test cognify endpoint accepts multiple ontology keys"""
payload = { payload = {
"datasets": ["test_dataset"], "datasets": ["test_dataset"],
@ -168,7 +168,7 @@ async def test_cognify_with_multiple_ontologies(mock_get_default_user, client, m
@patch.object(gau_mod, "get_default_user", new_callable=AsyncMock) @patch.object(gau_mod, "get_default_user", new_callable=AsyncMock)
async def test_complete_multifile_workflow(mock_get_default_user, client, mock_default_user): def test_complete_multifile_workflow(mock_get_default_user, client, mock_default_user):
"""Test complete workflow: upload multiple ontologies → cognify with multiple keys""" """Test complete workflow: upload multiple ontologies → cognify with multiple keys"""
import io import io
import json import json
@ -218,7 +218,7 @@ async def test_complete_multifile_workflow(mock_get_default_user, client, mock_d
@patch.object(gau_mod, "get_default_user", new_callable=AsyncMock) @patch.object(gau_mod, "get_default_user", new_callable=AsyncMock)
async def test_multifile_error_handling(mock_get_default_user, client, mock_default_user): def test_multifile_error_handling(mock_get_default_user, client, mock_default_user):
"""Test error handling for invalid multifile uploads""" """Test error handling for invalid multifile uploads"""
import io import io
import json import json
@ -251,7 +251,7 @@ async def test_multifile_error_handling(mock_get_default_user, client, mock_defa
@patch.object(gau_mod, "get_default_user", new_callable=AsyncMock) @patch.object(gau_mod, "get_default_user", new_callable=AsyncMock)
async def test_cognify_missing_ontology_key(mock_get_default_user, client, mock_default_user): def test_cognify_missing_ontology_key(mock_get_default_user, client, mock_default_user):
"""Test cognify with non-existent ontology key""" """Test cognify with non-existent ontology key"""
payload = { payload = {
"datasets": ["test_dataset"], "datasets": ["test_dataset"],