tiennm99/DocsGPT
1
0 Fork 0
mirror of https://github.com/tiennm99/DocsGPT.git synced 2026-10-14 14:17:08 +00:00
Code Issues Packages Projects Releases Wiki Activity

Pg 4 (#2390)

Browse Source 
* feat: postgres tests

* feat: mongo cutoff

* feat: mongo cutoff

* feat: adjust docs and compose files

* fix: mini code mongo removals

* fix: tests and k8s mongo stuff

* feat: test fixes

* fix: ruff

* fix: vale

* Potential fix for pull request finding 'CodeQL / Clear-text logging of sensitive information'

Co-authored-by: Copilot Autofix powered by AI <62310815+github-advanced-security[bot]@users.noreply.github.com>

* fix: mini suggestions

* vale lint fix 2

* fix: codeql columns thing

* fix: test mongo

* fix: tests coverage

* feat: better tests 4

* feat: more tests

* feat: decent coverage

* fix: ruff fixes

* fix: remove mongo mock

* feat: enhance workflow engine and API routes; add document retrieval and source handling

* feat: e2e tests

* fix: mcp, mongo and more

* fix: mini codeql warning

* fix: agent chunk view

* fix: mini issues

* fix: more pg fixes

* feat: postgres prep on start

* feat: qa tests

* fix: mini improvements

* fix: tests

---------

Co-authored-by: Copilot Autofix powered by AI <62310815+github-advanced-security[bot]@users.noreply.github.com>
Co-authored-by: Siddhant Rai <siddhant.rai.5686@gmail.com>
This commit is contained in:
AlexCopilot Autofix powered by AISiddhant Rai
 authored and GitHub committed 2026-04-18 13:13:57 +01:00
1 parent 7f6360b4ff
commit 81b6ee5daa
312 files changed
+763725 -25159
Download Patch File Download Diff File Expand all files Collapse all files

No files matched your search

tests/test_logging.py
+3 -49
View File
@@ -1,4 +1,4 @@
from unittest.mock import Mock, patch
from unittest.mock import patch
import pytest
@@ -131,7 +131,7 @@ class TestLogActivity:
yield "chunk1"
yield "chunk2"
with patch("application.logging._log_to_mongodb"):
with patch("application.logging._log_activity_to_db"):
result = list(my_gen(FakeAgent()))
assert result == ["chunk1", "chunk2"]
@@ -148,55 +148,9 @@ class TestLogActivity:
yield "ok"
raise RuntimeError("boom")
with patch("application.logging._log_to_mongodb"), pytest.raises(
with patch("application.logging._log_activity_to_db"), pytest.raises(
RuntimeError, match="boom"
):
list(failing_gen(FakeAgent()))
@pytest.mark.unit
class TestLogToMongoDB:
def test_logs_entry(self):
from application.logging import _log_to_mongodb
mock_collection = Mock()
mock_db = {"stack_logs": mock_collection}
with patch(
"application.logging.MongoDB.get_client",
return_value={"docsgpt": mock_db},
), patch("application.logging.settings") as mock_settings:
mock_settings.MONGO_DB_NAME = "docsgpt"
_log_to_mongodb("ep", "aid", "user", "key", "q", [], "info")
mock_collection.insert_one.assert_called_once()
doc = mock_collection.insert_one.call_args[0][0]
assert doc["endpoint"] == "ep"
assert doc["level"] == "info"
def test_truncates_long_strings(self):
from application.logging import _log_to_mongodb
mock_collection = Mock()
mock_db = {"stack_logs": mock_collection}
with patch(
"application.logging.MongoDB.get_client",
return_value={"docsgpt": mock_db},
), patch("application.logging.settings") as mock_settings:
mock_settings.MONGO_DB_NAME = "docsgpt"
_log_to_mongodb("ep", "aid", "user", "key", "x" * 20000, [], "info")
doc = mock_collection.insert_one.call_args[0][0]
assert len(doc["query"]) == 10000
def test_handles_mongo_error(self):
from application.logging import _log_to_mongodb
with patch(
"application.logging.MongoDB.get_client",
side_effect=Exception("DB down"),
):
# Should not raise
_log_to_mongodb("ep", "aid", "user", "key", "q", [], "info")