mirror of
https://github.com/tiennm99/DocsGPT.git
synced 2026-10-04 00:13:14 +00:00
81b6ee5daa
* 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>
100 lines
3.3 KiB
Python
100 lines
3.3 KiB
Python
"""Tests for Confluence auth get_token_info_from_session using pg_conn."""
|
|
|
|
from contextlib import contextmanager
|
|
from unittest.mock import patch
|
|
|
|
import pytest
|
|
|
|
|
|
@contextmanager
|
|
def _patch_db(conn):
|
|
@contextmanager
|
|
def _yield():
|
|
yield conn
|
|
|
|
with patch(
|
|
"application.storage.db.session.db_readonly", _yield
|
|
):
|
|
yield
|
|
|
|
|
|
class TestGetTokenInfoFromSession:
|
|
def test_invalid_session_token_raises(self, pg_conn):
|
|
from application.parser.connectors.confluence.auth import (
|
|
ConfluenceAuth,
|
|
)
|
|
|
|
auth = ConfluenceAuth.__new__(ConfluenceAuth)
|
|
with _patch_db(pg_conn), pytest.raises(ValueError):
|
|
auth.get_token_info_from_session("no-such-token")
|
|
|
|
def test_missing_token_info_raises(self, pg_conn):
|
|
from application.parser.connectors.confluence.auth import (
|
|
ConfluenceAuth,
|
|
)
|
|
from application.storage.db.repositories.connector_sessions import (
|
|
ConnectorSessionsRepository,
|
|
)
|
|
|
|
repo = ConnectorSessionsRepository(pg_conn)
|
|
repo.upsert("u", "confluence", status="authorized")
|
|
# Set session_token but no token_info
|
|
session = repo.get_by_user_provider("u", "confluence")
|
|
repo.update(str(session["id"]), {"session_token": "tok-no-info"})
|
|
|
|
auth = ConfluenceAuth.__new__(ConfluenceAuth)
|
|
with _patch_db(pg_conn), pytest.raises(ValueError):
|
|
auth.get_token_info_from_session("tok-no-info")
|
|
|
|
def test_missing_required_fields_raises(self, pg_conn):
|
|
from application.parser.connectors.confluence.auth import (
|
|
ConfluenceAuth,
|
|
)
|
|
from application.storage.db.repositories.connector_sessions import (
|
|
ConnectorSessionsRepository,
|
|
)
|
|
|
|
repo = ConnectorSessionsRepository(pg_conn)
|
|
repo.upsert("u", "confluence", status="authorized")
|
|
session = repo.get_by_user_provider("u", "confluence")
|
|
repo.update(
|
|
str(session["id"]),
|
|
{
|
|
"session_token": "tok-partial",
|
|
"token_info": {"access_token": "at"}, # missing refresh + cloud_id
|
|
},
|
|
)
|
|
|
|
auth = ConfluenceAuth.__new__(ConfluenceAuth)
|
|
with _patch_db(pg_conn), pytest.raises(ValueError):
|
|
auth.get_token_info_from_session("tok-partial")
|
|
|
|
def test_complete_token_info_returned(self, pg_conn):
|
|
from application.parser.connectors.confluence.auth import (
|
|
ConfluenceAuth,
|
|
)
|
|
from application.storage.db.repositories.connector_sessions import (
|
|
ConnectorSessionsRepository,
|
|
)
|
|
|
|
repo = ConnectorSessionsRepository(pg_conn)
|
|
repo.upsert("u", "confluence", status="authorized")
|
|
session = repo.get_by_user_provider("u", "confluence")
|
|
repo.update(
|
|
str(session["id"]),
|
|
{
|
|
"session_token": "tok-good",
|
|
"token_info": {
|
|
"access_token": "at",
|
|
"refresh_token": "rt",
|
|
"cloud_id": "cid-1",
|
|
},
|
|
},
|
|
)
|
|
|
|
auth = ConfluenceAuth.__new__(ConfluenceAuth)
|
|
with _patch_db(pg_conn):
|
|
got = auth.get_token_info_from_session("tok-good")
|
|
assert got["access_token"] == "at"
|
|
assert got["cloud_id"] == "cid-1"
|