apunkt/trustgraph
1
0
Fork 0
mirror of https://github.com/trustgraph-ai/trustgraph.git synced 2026-07-02 22:41:01 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions 5

Fix tests

Browse source
This commit is contained in:
Cyber MacGeddon 2026-04-21 13:34:52 +01:00
parent bc1a7c1c44
commit 04b336fa82
2 changed files with 9 additions and 11 deletions
Download patch file Download diff file Expand all files Collapse all files

18
tests/unit/test_cli/test_load_knowledge.py
View file

@ -48,7 +48,7 @@ def knowledge_loader():
return KnowledgeLoader( return KnowledgeLoader(
files=["test.ttl"], files=["test.ttl"],
flow="test-flow", flow="test-flow",
user="test-user", workspace="test-user",
collection="test-collection", collection="test-collection",
document_id="test-doc-123", document_id="test-doc-123",
url="http://test.example.com/", url="http://test.example.com/",
@ -64,7 +64,7 @@ class TestKnowledgeLoader:
loader = KnowledgeLoader( loader = KnowledgeLoader(
files=["file1.ttl", "file2.ttl"], files=["file1.ttl", "file2.ttl"],
flow="my-flow", flow="my-flow",
user="user1", workspace="user1",
collection="col1", collection="col1",
document_id="doc1", document_id="doc1",
url="http://example.com/", url="http://example.com/",
@ -73,7 +73,7 @@ class TestKnowledgeLoader:
assert loader.files == ["file1.ttl", "file2.ttl"] assert loader.files == ["file1.ttl", "file2.ttl"]
assert loader.flow == "my-flow" assert loader.flow == "my-flow"
assert loader.user == "user1" assert loader.workspace == "user1"
assert loader.collection == "col1" assert loader.collection == "col1"
assert loader.document_id == "doc1" assert loader.document_id == "doc1"
assert loader.url == "http://example.com/" assert loader.url == "http://example.com/"
@ -126,7 +126,7 @@ ex:mary ex:knows ex:bob .
loader = KnowledgeLoader( loader = KnowledgeLoader(
files=[f.name], files=[f.name],
flow="test-flow", flow="test-flow",
user="test-user", workspace="test-user",
collection="test-collection", collection="test-collection",
document_id="test-doc", document_id="test-doc",
url="http://test.example.com/" url="http://test.example.com/"
@ -151,7 +151,7 @@ ex:mary ex:knows ex:bob .
loader = KnowledgeLoader( loader = KnowledgeLoader(
files=[temp_turtle_file], files=[temp_turtle_file],
flow="test-flow", flow="test-flow",
user="test-user", workspace="test-user",
collection="test-collection", collection="test-collection",
document_id="test-doc", document_id="test-doc",
url="http://test.example.com/", url="http://test.example.com/",
@ -199,7 +199,7 @@ class TestCLIArgumentParsing:
'tg-load-knowledge', 'tg-load-knowledge',
'-i', 'doc-123', '-i', 'doc-123',
'-f', 'my-flow', '-f', 'my-flow',
'-U', 'my-user', '-w', 'my-user',
'-C', 'my-collection', '-C', 'my-collection',
'-u', 'http://custom.example.com/', '-u', 'http://custom.example.com/',
'-t', 'my-token', '-t', 'my-token',
@ -217,7 +217,7 @@ class TestCLIArgumentParsing:
token='my-token', token='my-token',
flow='my-flow', flow='my-flow',
files=['file1.ttl', 'file2.ttl'], files=['file1.ttl', 'file2.ttl'],
user='my-user', workspace='my-user',
collection='my-collection' collection='my-collection'
) )
@ -243,7 +243,7 @@ class TestCLIArgumentParsing:
# Verify defaults were used # Verify defaults were used
call_args = mock_loader_class.call_args[1] call_args = mock_loader_class.call_args[1]
assert call_args['flow'] == 'default' assert call_args['flow'] == 'default'
assert call_args['user'] == 'trustgraph' assert call_args['workspace'] == 'default'
assert call_args['collection'] == 'default' assert call_args['collection'] == 'default'
assert call_args['url'] == 'http://localhost:8088/' assert call_args['url'] == 'http://localhost:8088/'
assert call_args['token'] is None assert call_args['token'] is None
@ -288,7 +288,7 @@ class TestErrorHandling:
loader = KnowledgeLoader( loader = KnowledgeLoader(
files=[temp_turtle_file], files=[temp_turtle_file],
flow="test-flow", flow="test-flow",
user="test-user", workspace="test-user",
collection="test-collection", collection="test-collection",
document_id="test-doc", document_id="test-doc",
url="http://test.example.com/" url="http://test.example.com/"

2
trustgraph-cli/trustgraph/cli/load_knowledge.py
View file

@ -94,7 +94,6 @@ class KnowledgeLoader:
metadata={ metadata={
"id": self.document_id, "id": self.document_id,
"metadata": [], "metadata": [],
"user": self.workspace,
"collection": self.collection "collection": self.collection
} }
) )
@ -124,7 +123,6 @@ class KnowledgeLoader:
metadata={ metadata={
"id": self.document_id, "id": self.document_id,
"metadata": [], "metadata": [],
"user": self.workspace,
"collection": self.collection "collection": self.collection
} }
) )