| """Unit tests for tools with mocked HTTP / external calls.""" |
|
|
| from unittest.mock import MagicMock, patch |
|
|
| from gaia_agent.tools import ( |
| classify_file, |
| python_repl, |
| read_spreadsheet, |
| read_text_file, |
| ) |
| from gaia_agent.tools.files import fetch_task_file |
|
|
|
|
| def test_classify_file(): |
| assert classify_file("a/b/data.xlsx") == "spreadsheet" |
| assert classify_file("clip.MP3") == "audio" |
| assert classify_file("pic.jpeg") == "image" |
| assert classify_file("script.py") == "code" |
| assert classify_file("notes.txt") == "text" |
| assert classify_file("weird.xyz") == "other" |
|
|
|
|
| def test_python_repl_arithmetic(): |
| out = python_repl.invoke({"code": "result = sum(range(1, 11))"}) |
| assert "55" in out |
|
|
|
|
| def test_python_repl_handles_error(): |
| out = python_repl.invoke({"code": "1/0"}) |
| assert "Error" in out |
|
|
|
|
| def test_read_text_file(tmp_path): |
| p = tmp_path / "f.txt" |
| p.write_text("hello gaia", encoding="utf-8") |
| assert "hello gaia" in read_text_file.invoke({"path": str(p)}) |
|
|
|
|
| def test_read_spreadsheet_csv(tmp_path): |
| p = tmp_path / "t.csv" |
| p.write_text("a,b\n1,2\n3,4\n", encoding="utf-8") |
| out = read_spreadsheet.invoke({"path": str(p)}) |
| assert "2 rows" in out and "['a', 'b']" in out |
|
|
|
|
| def test_fetch_task_file_no_file(): |
| """A non-200 response yields None rather than a path.""" |
| with patch("gaia_agent.tools.files.requests.get") as mock_get: |
| mock_get.return_value = MagicMock(status_code=404, content=b"") |
| assert fetch_task_file("missing") is None |
|
|
|
|
| def test_fetch_task_file_writes(tmp_path): |
| resp = MagicMock(status_code=200, content=b"col\n1\n", |
| headers={"content-disposition": 'attachment; filename="x.csv"'}) |
| with patch("gaia_agent.tools.files.requests.get", return_value=resp): |
| path = fetch_task_file("task1") |
| assert path is not None and path.endswith("x.csv") |
|
|
