From 0ad66dffb2fc94ac7dff3262b7b5a569021e16f1 Mon Sep 17 00:00:00 2001 From: Cyber MacGeddon Date: Wed, 30 Jul 2025 22:28:44 +0100 Subject: [PATCH] Logging strategy updates --- .../test_document_rag_integration.py | 24 +++++++++++-------- 1 file changed, 14 insertions(+), 10 deletions(-) diff --git a/tests/integration/test_document_rag_integration.py b/tests/integration/test_document_rag_integration.py index f92126fc..3655962f 100644 --- a/tests/integration/test_document_rag_integration.py +++ b/tests/integration/test_document_rag_integration.py @@ -248,9 +248,13 @@ class TestDocumentRagIntegration: @pytest.mark.asyncio async def test_document_rag_verbose_logging(self, mock_embeddings_client, mock_doc_embeddings_client, mock_prompt_client, - capsys): + caplog): """Test DocumentRAG verbose logging functionality""" - # Arrange + import logging + + # Arrange - Configure logging to capture debug messages + caplog.set_level(logging.DEBUG) + document_rag = DocumentRag( embeddings_client=mock_embeddings_client, doc_embeddings_client=mock_doc_embeddings_client, @@ -261,14 +265,14 @@ class TestDocumentRagIntegration: # Act await document_rag.query("test query for verbose logging") - # Assert - captured = capsys.readouterr() - assert "Initialised" in captured.out - assert "Construct prompt..." in captured.out - assert "Compute embeddings..." in captured.out - assert "Get docs..." in captured.out - assert "Invoke LLM..." in captured.out - assert "Done" in captured.out + # Assert - Check for new logging messages + log_messages = caplog.text + assert "DocumentRag initialized" in log_messages + assert "Constructing prompt..." in log_messages + assert "Computing embeddings..." in log_messages + assert "Getting documents..." in log_messages + assert "Invoking LLM..." in log_messages + assert "Query processing complete" in log_messages @pytest.mark.asyncio @pytest.mark.slow