Rockachopa/Timmy-time-dashboard
1
0
Fork 2
Code Issues Pull Requests 1 Actions 6 Packages Projects Releases Wiki Activity

feat: code quality audit + autoresearch integration + infra hardening (#150)

Browse Source
This commit is contained in:
Alexander Whitestone
2026-03-08 12:50:44 -04:00
committed by GitHub
parent fd0ede0d51
commit ae3bb1cc21
186 changed files with 5129 additions and 3289 deletions
Download Patch File Download Diff File Expand all files Collapse all files

24
tests/timmy/test_ollama_timeout.py
View File

@@ -5,13 +5,12 @@ This caused socket read errors in production. The agno Ollama class uses
``timeout`` (not ``request_timeout``).
"""
from unittest.mock import patch, MagicMock
from unittest.mock import MagicMock, patch
def test_base_agent_sets_timeout():
"""BaseAgent creates Ollama with timeout=300."""
with patch("timmy.agents.base.Ollama") as mock_ollama, \
patch("timmy.agents.base.Agent"):
with patch("timmy.agents.base.Ollama") as mock_ollama, patch("timmy.agents.base.Agent"):
mock_ollama.return_value = MagicMock()
# Import after patching to get the patched version
@@ -36,19 +35,20 @@ def test_base_agent_sets_timeout():
# Verify Ollama was called with timeout
if mock_ollama.called:
_, kwargs = mock_ollama.call_args
assert kwargs.get("timeout") == 300, (
f"Expected timeout=300, got {kwargs.get('timeout')}"
)
assert (
kwargs.get("timeout") == 300
), f"Expected timeout=300, got {kwargs.get('timeout')}"
def test_main_agent_sets_timeout():
"""create_timmy() creates Ollama with timeout=300."""
with patch("timmy.agent.Ollama") as mock_ollama, \
patch("timmy.agent.SqliteDb"), \
patch("timmy.agent.Agent"):
with patch("timmy.agent.Ollama") as mock_ollama, patch("timmy.agent.SqliteDb"), patch(
"timmy.agent.Agent"
):
mock_ollama.return_value = MagicMock()
from timmy.agent import create_timmy
try:
create_timmy()
except Exception:
@@ -56,6 +56,6 @@ def test_main_agent_sets_timeout():
if mock_ollama.called:
_, kwargs = mock_ollama.call_args
assert kwargs.get("timeout") == 300, (
f"Expected timeout=300, got {kwargs.get('timeout')}"
)
assert (
kwargs.get("timeout") == 300
), f"Expected timeout=300, got {kwargs.get('timeout')}"