ab889f9f1d
# PR Description
* Adds/updates the following files to all toolkits:
- `.pre-commit-config.yaml`
- `.ruff.toml`
- `LICENSE`
- `Makefile`
- `pyproject.toml`
* Lint all toolkits such that they pass `make check` and `make test` (a
total doozy). This includes adding some unit tests and evals.
* Github workflow for testing toolkits before merge into main (courtesy
of @sdreyer)
* Added a QOL improvement for tool developers for when they need to get
the context's auth token.
* Minor updates to `arcade new` template.
47 lines
1.5 KiB
Python
47 lines
1.5 KiB
Python
import json
|
|
from unittest.mock import patch
|
|
|
|
import pytest
|
|
|
|
from arcade_search.tools.google import search_google
|
|
|
|
GET_SECRET_PATCH_TARGET = "arcade_search.tools.google.get_secret" # noqa: S105
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_search_google_success():
|
|
with (
|
|
patch(GET_SECRET_PATCH_TARGET, return_value="fake_api_key"),
|
|
patch("serpapi.Client") as MockClient,
|
|
):
|
|
mock_client_instance = MockClient.return_value
|
|
mock_client_instance.search.return_value.as_dict.return_value = {
|
|
"organic_results": [
|
|
{"title": "Result 1", "link": "http://example.com/1"},
|
|
{"title": "Result 2", "link": "http://example.com/2"},
|
|
{"title": "Result 3", "link": "http://example.com/3"},
|
|
]
|
|
}
|
|
|
|
result = await search_google("test query", 2)
|
|
|
|
expected_result = json.dumps([
|
|
{"title": "Result 1", "link": "http://example.com/1"},
|
|
{"title": "Result 2", "link": "http://example.com/2"},
|
|
])
|
|
assert result == expected_result
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_search_google_no_results():
|
|
with (
|
|
patch(GET_SECRET_PATCH_TARGET, return_value="fake_api_key"),
|
|
patch("serpapi.Client") as MockClient,
|
|
):
|
|
mock_client_instance = MockClient.return_value
|
|
mock_client_instance.search.return_value.as_dict.return_value = {"organic_results": []}
|
|
|
|
result = await search_google("test query", 2)
|
|
|
|
expected_result = json.dumps([])
|
|
assert result == expected_result
|