From 20993b48a7401efd23919ce43aed89d009338015 Mon Sep 17 00:00:00 2001 From: Yiorgis Gozadinos Date: Mon, 22 Sep 2025 12:53:52 +0300 Subject: [PATCH] Fix test after rebase --- tests/test_cli.py | 9 +++++---- 1 file changed, 5 insertions(+), 4 deletions(-) diff --git a/tests/test_cli.py b/tests/test_cli.py index f1ed1fa6..7c2b4a80 100644 --- a/tests/test_cli.py +++ b/tests/test_cli.py @@ -46,7 +46,7 @@ def test_add_document_src(): 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.add_document_from_source = AsyncMock() mock_app.return_value = mock_app_instance @@ -55,9 +55,10 @@ def test_add_document_src_with_title(): assert result.exit_code == 0 mock_app_instance.add_document_from_source.assert_called_once() - # Verify title is forwarded - await_args = mock_app_instance.add_document_from_source.await_args - assert await_args.kwargs.get("title") == "Nice Name" + # Verify title is forwarded (inspect call kwargs) + _, kwargs = mock_app_instance.add_document_from_source.call_args + assert kwargs.get("title") == "Nice Name" + assert kwargs.get("source") == "test.txt" def test_get_document():