danvics/haiku.rag
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Fix test after rebase

Browse source
This commit is contained in:
Yiorgis Gozadinos 2025-09-22 12:53:52 +03:00
parent 8e2829bf0e
commit 20993b48a7
No known key found for this signature in database
1 changed files with 5 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

9
tests/test_cli.py
View file

@ -46,7 +46,7 @@ def test_add_document_src():
def test_add_document_src_with_title(): def test_add_document_src_with_title():
with patch("haiku.rag.cli.HaikuRAGApp") as mock_app: with patch("haiku.rag.app.HaikuRAGApp") as mock_app:
mock_app_instance = MagicMock() mock_app_instance = MagicMock()
mock_app_instance.add_document_from_source = AsyncMock() mock_app_instance.add_document_from_source = AsyncMock()
mock_app.return_value = mock_app_instance mock_app.return_value = mock_app_instance
@ -55,9 +55,10 @@ def test_add_document_src_with_title():
assert result.exit_code == 0 assert result.exit_code == 0
mock_app_instance.add_document_from_source.assert_called_once() mock_app_instance.add_document_from_source.assert_called_once()
# Verify title is forwarded # Verify title is forwarded (inspect call kwargs)
await_args = mock_app_instance.add_document_from_source.await_args _, kwargs = mock_app_instance.add_document_from_source.call_args
assert await_args.kwargs.get("title") == "Nice Name" assert kwargs.get("title") == "Nice Name"
assert kwargs.get("source") == "test.txt"
def test_get_document(): def test_get_document():