Files
DocsGPT/tests/api/user/agents/test_routes_gaps.py
T
81b6ee5daa Pg 4 (#2390)
* 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>
2026-04-18 13:13:57 +01:00

121 lines
3.3 KiB
Python

"""Gap-coverage tests for application.api.user.agents.routes.
No bson/ObjectId imports. The routes internally use ObjectId (patched where
needed) and Mongo collections (mocked). Repository assertions use
AgentsRepository patches for Postgres-specific paths.
"""
import uuid
from unittest.mock import Mock
import pytest
from flask import Flask
# ---------------------------------------------------------------------------
# Fixtures
# ---------------------------------------------------------------------------
@pytest.fixture
def app():
return Flask(__name__)
def _fake_oid():
"""Return a 24-character hex string that looks like a Mongo ObjectId."""
return uuid.uuid4().hex[:24]
# ---------------------------------------------------------------------------
# normalize_workflow_reference — pure utility, no bson needed
# ---------------------------------------------------------------------------
@pytest.mark.unit
class TestNormalizeWorkflowReferenceGaps:
pass
def test_handles_integer_zero(self):
from application.api.user.agents.routes import normalize_workflow_reference
assert normalize_workflow_reference(0) == "0"
def test_handles_list_converts_to_str(self):
from application.api.user.agents.routes import normalize_workflow_reference
result = normalize_workflow_reference([1, 2])
assert isinstance(result, str)
def test_json_string_with_underscore_id(self):
from application.api.user.agents.routes import normalize_workflow_reference
result = normalize_workflow_reference('{"_id": "abc"}')
assert result == "abc"
def test_json_string_with_workflow_id_key(self):
from application.api.user.agents.routes import normalize_workflow_reference
result = normalize_workflow_reference('{"workflow_id": "wf99"}')
assert result == "wf99"
def test_whitespace_only_string_returns_empty(self):
from application.api.user.agents.routes import normalize_workflow_reference
assert normalize_workflow_reference(" \t ") == ""
def test_dict_missing_all_id_keys_returns_none(self):
from application.api.user.agents.routes import normalize_workflow_reference
result = normalize_workflow_reference({"other_key": "value"})
assert result is None
# ---------------------------------------------------------------------------
# validate_workflow_access — mocked ObjectId + workflows_collection
# ---------------------------------------------------------------------------
@pytest.mark.unit
class TestValidateWorkflowAccessGaps:
pass
def _fake_objectid(self, valid: bool = True):
"""Return a Mock that mimics bson.ObjectId."""
oid_cls = Mock()
oid_cls.is_valid = Mock(return_value=valid)
oid_cls.return_value = Mock()
return oid_cls
# ---------------------------------------------------------------------------
# build_agent_document — pure logic, no DB needed
# ---------------------------------------------------------------------------
@pytest.mark.unit
class TestBuildAgentDocumentGaps:
pass
# ---------------------------------------------------------------------------
# GetAgent route — additional edge cases (uuid-based IDs, no bson)
# ---------------------------------------------------------------------------
@pytest.mark.unit
class TestGetAgentGaps:
pass