updates ontology endpoint call in test_cognee_server_start

This commit is contained in:
hajdul88 2025-12-15 17:41:25 +01:00
parent d4ec7b99ec
commit 738fa9c05f
2 changed files with 5 additions and 3 deletions

View file

@ -148,8 +148,8 @@ class TestCogneeServerStart(unittest.TestCase):
headers=headers, headers=headers,
files=[("ontology_file", ("test.owl", ontology_content, "application/xml"))], files=[("ontology_file", ("test.owl", ontology_content, "application/xml"))],
data={ data={
"ontology_key": json.dumps([ontology_key]), "ontology_key": ontology_key,
"description": json.dumps(["Test ontology"]), "description": "Test ontology",
}, },
) )
self.assertEqual(ontology_response.status_code, 200) self.assertEqual(ontology_response.status_code, 200)

View file

@ -179,7 +179,9 @@ def test_complete_multifile_workflow(client):
upload_response_2 = client.post( upload_response_2 = client.post(
"/api/v1/ontologies", "/api/v1/ontologies",
files=[("ontology_file", ("manufacturers.owl", io.BytesIO(file2_content), "application/xml"))], files=[
("ontology_file", ("manufacturers.owl", io.BytesIO(file2_content), "application/xml"))
],
data={"ontology_key": "manufacturers", "description": "Manufacturer ontology"}, data={"ontology_key": "manufacturers", "description": "Manufacturer ontology"},
) )
assert upload_response_2.status_code == 200 assert upload_response_2.status_code == 200