mirror of
https://github.com/MODSetter/SurfSense.git
synced 2026-04-27 17:56:25 +02:00
chore: ran linting
This commit is contained in:
parent
44f59bfb61
commit
2bc6a0c3bc
24 changed files with 166 additions and 160 deletions
|
|
@ -23,7 +23,7 @@ async def test_search_google_drive_includes_legacy_composio_docs(
|
|||
|
||||
async with patched_session_factory() as session:
|
||||
service = ConnectorService(session, search_space_id=space_id)
|
||||
result_object, raw_docs = await service.search_google_drive(
|
||||
_, raw_docs = await service.search_google_drive(
|
||||
user_query="quarterly budget",
|
||||
search_space_id=space_id,
|
||||
top_k=10,
|
||||
|
|
@ -55,7 +55,7 @@ async def test_search_files_does_not_include_google_types(
|
|||
|
||||
async with patched_session_factory() as session:
|
||||
service = ConnectorService(session, search_space_id=space_id)
|
||||
result_object, raw_docs = await service.search_files(
|
||||
_, raw_docs = await service.search_files(
|
||||
user_query="quarterly budget",
|
||||
search_space_id=space_id,
|
||||
top_k=10,
|
||||
|
|
|
|||
|
|
@ -15,11 +15,11 @@ pytestmark = pytest.mark.unit
|
|||
|
||||
|
||||
@patch("app.services.composio_service.ComposioService")
|
||||
def test_returns_credentials_with_token_and_expiry(MockComposioService):
|
||||
def test_returns_credentials_with_token_and_expiry(mock_composio_service):
|
||||
"""build_composio_credentials returns a Credentials object with the Composio access token."""
|
||||
mock_service = MagicMock()
|
||||
mock_service.get_access_token.return_value = "fake-access-token"
|
||||
MockComposioService.return_value = mock_service
|
||||
mock_composio_service.return_value = mock_service
|
||||
|
||||
from app.utils.google_credentials import build_composio_credentials
|
||||
|
||||
|
|
@ -32,14 +32,14 @@ def test_returns_credentials_with_token_and_expiry(MockComposioService):
|
|||
|
||||
|
||||
@patch("app.services.composio_service.ComposioService")
|
||||
def test_refresh_handler_fetches_fresh_token(MockComposioService):
|
||||
def test_refresh_handler_fetches_fresh_token(mock_composio_service):
|
||||
"""The refresh_handler on the returned Credentials fetches a new token from Composio."""
|
||||
mock_service = MagicMock()
|
||||
mock_service.get_access_token.side_effect = [
|
||||
"initial-token",
|
||||
"refreshed-token",
|
||||
]
|
||||
MockComposioService.return_value = mock_service
|
||||
mock_composio_service.return_value = mock_service
|
||||
|
||||
from app.utils.google_credentials import build_composio_credentials
|
||||
|
||||
|
|
|
|||
|
|
@ -208,7 +208,7 @@ async def test_drive_client_uses_prebuilt_composio_credentials(mock_build):
|
|||
credentials=creds,
|
||||
)
|
||||
|
||||
files, next_token, error = await client.list_files()
|
||||
files, _next_token, error = await client.list_files()
|
||||
|
||||
assert error is None
|
||||
assert files == []
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue