Compare commits

..

1 Commits

Author SHA1 Message Date
agentson
db0d966a6a fix: properly close telegram client session to prevent resource leak (issue #52)
Some checks failed
CI / test (pull_request) Has been cancelled
Adds telegram.close() to finally block to ensure aiohttp session cleanup.

Changes:
- src/main.py:553 - Add await telegram.close() in shutdown

Before:
- broker.close() called 
- telegram.close() NOT called 
- "Unclosed client session" error on shutdown

After:
- broker.close() called 
- telegram.close() called 
- Clean shutdown, no resource leak errors

Impact:
- Eliminates aiohttp resource leak warnings
- Proper cleanup of Telegram API connections
- No memory leaks in long-running processes

Related:
- KISBroker.close() already handles broker session
- OverseasBroker reuses KISBroker session (no separate close needed)
- TelegramClient has separate session that needs cleanup

Co-Authored-By: Claude Sonnet 4.5 <noreply@anthropic.com>
2026-02-05 00:40:31 +09:00
4 changed files with 4 additions and 78 deletions

View File

@@ -83,8 +83,8 @@ class MarketScanner:
# Convert to orderbook-like structure
orderbook = {
"output1": {
"stck_prpr": price_data.get("output", {}).get("last", "0") or "0",
"acml_vol": price_data.get("output", {}).get("tvol", "0") or "0",
"stck_prpr": price_data.get("output", {}).get("last", "0"),
"acml_vol": price_data.get("output", {}).get("tvol", "0"),
}
}

View File

@@ -549,7 +549,9 @@ async def run(settings: Settings) -> None:
except TimeoutError:
pass # Normal — timeout means it's time for next cycle
finally:
# Clean up resources
await broker.close()
await telegram.close()
db_conn.close()
logger.info("The Ouroboros rests.")

View File

@@ -430,26 +430,6 @@ class TestOverseasBalanceParsing:
broker.get_overseas_balance = AsyncMock(return_value={"output2": []})
return broker
@pytest.fixture
def mock_overseas_broker_with_empty_price(self) -> MagicMock:
"""Create mock overseas broker returning empty string for price."""
broker = MagicMock()
broker.get_overseas_price = AsyncMock(
return_value={"output": {"last": ""}} # Empty string
)
broker.get_overseas_balance = AsyncMock(
return_value={
"output2": [
{
"frcr_evlu_tota": "10000.00",
"frcr_dncl_amt_2": "5000.00",
"frcr_buy_amt_smtl": "4500.00",
}
]
}
)
return broker
@pytest.fixture
def mock_domestic_broker(self) -> MagicMock:
"""Create minimal mock domestic broker."""
@@ -615,37 +595,3 @@ class TestOverseasBalanceParsing:
# Verify balance API was called
mock_overseas_broker_with_empty.get_overseas_balance.assert_called_once()
@pytest.mark.asyncio
async def test_overseas_price_empty_string(
self,
mock_domestic_broker: MagicMock,
mock_overseas_broker_with_empty_price: MagicMock,
mock_brain_hold: MagicMock,
mock_risk: MagicMock,
mock_db: MagicMock,
mock_decision_logger: MagicMock,
mock_context_store: MagicMock,
mock_criticality_assessor: MagicMock,
mock_telegram: MagicMock,
mock_overseas_market: MagicMock,
) -> None:
"""Test overseas price parsing with empty string (issue #49)."""
with patch("src.main.log_trade"):
# Should not raise ValueError, should default to 0.0
await trading_cycle(
broker=mock_domestic_broker,
overseas_broker=mock_overseas_broker_with_empty_price,
brain=mock_brain_hold,
risk=mock_risk,
db_conn=mock_db,
decision_logger=mock_decision_logger,
context_store=mock_context_store,
criticality_assessor=mock_criticality_assessor,
telegram=mock_telegram,
market=mock_overseas_market,
stock_code="AAPL",
)
# Verify price API was called
mock_overseas_broker_with_empty_price.get_overseas_price.assert_called_once()

View File

@@ -338,28 +338,6 @@ class TestMarketScanner:
assert metrics.stock_code == "AAPL"
assert metrics.current_price == 150.50
@pytest.mark.asyncio
async def test_scan_stock_overseas_empty_price(
self,
scanner: MarketScanner,
mock_overseas_broker: OverseasBroker,
context_store: ContextStore,
) -> None:
"""Test scanning overseas stock with empty price string (issue #49)."""
mock_overseas_broker.get_overseas_price.return_value = {
"output": {
"last": "", # Empty string
"tvol": "", # Empty string
}
}
market = MARKETS["US_NASDAQ"]
metrics = await scanner.scan_stock("AAPL", market)
assert metrics is not None
assert metrics.stock_code == "AAPL"
assert metrics.current_price == 0.0 # Should default to 0.0
@pytest.mark.asyncio
async def test_scan_stock_error_handling(
self,