Commit Graph
5 Commits
Author SHA1 Message Date
Patrick SelamyandClaude Opus 4.5 31c675bdec fix: resolve flaky tests in clob_client and websocket (#49)
## Changes

### test_get_market_not_found
- Updated test to expect `RetryError` instead of `ClobClientError`
- The `get_market` method uses the `@with_retry()` decorator, so when
  the underlying API call fails repeatedly, it raises `RetryError`
  wrapping the original exception
- Removed unused `ClobClientError` import

### test_start_and_receive_trades
- Replaced `MagicMock` with a proper `MockWebSocket` class that
  implements the async iterator protocol correctly
- `MagicMock` was passing `self` as an extra argument when calling
  `__aiter__`, causing "takes 0 positional arguments but 1 was given"
- Removed unused `MagicMock` import

Closes #49

🤖 Generated with [Claude Code](https://claude.com/claude-code)

Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com>
2026-01-04 19:04:32 -05:00
Patrick SelamyandClaude Opus 4.5 63aade0a60 fix: mark 2 flaky tests as xfail pending #49
- test_get_market_not_found: retry logic wraps exception
- test_start_and_receive_trades: async mock iterator signature

🤖 Generated with [Claude Code](https://claude.com/claude-code)

Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com>
2026-01-04 17:21:43 -05:00
Patrick SelamyandClaude Opus 4.5 2a35ade607 fix: correct test fixture names and add aiosqlite dependency
- Revert underscore-prefixed fixture parameters (breaks pytest discovery)
- Use # noqa: ARG002 comments instead
- Add aiosqlite to dev dependencies for SQLite-based tests

🤖 Generated with [Claude Code](https://claude.com/claude-code)

Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com>
2026-01-04 17:18:34 -05:00
Patrick SelamyandClaude Opus 4.5 1f4f1fa557 fix: resolve linting and formatting issues for CI
- Use contextlib.suppress instead of try/except/pass (SIM105)
- Prefix unused fixture arguments with underscore (ARG002)
- Replace asyncio.TimeoutError with TimeoutError (UP041)
- Apply ruff formatting to all files

🤖 Generated with [Claude Code](https://claude.com/claude-code)

Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com>
2026-01-04 17:09:14 -05:00
Patrick SelamyandClaude Opus 4.5 85010a3ea5 feat: implement WebSocket trade stream handler with reconnection
- Add TradeEvent dataclass for trade data from WebSocket feed
- Implement TradeStreamHandler with async WebSocket streaming
- Add automatic reconnection with exponential backoff (1s-30s)
- Support event/market filtering for targeted subscriptions
- Include connection state management and statistics tracking
- Add websockets>=12.0 dependency

Acceptance Criteria:
- [x] TradeStreamHandler class using websockets library
- [x] Connects to Polymarket WSS endpoint
- [x] Subscribes to market trade channel on connection
- [x] Parses trade messages into TradeEvent dataclass
- [x] Implements heartbeat/ping-pong for connection health
- [x] Auto-reconnects on disconnect with exponential backoff
- [x] Emits events via callback pattern
- [x] Logs connection state changes

Closes #3

🤖 Generated with [Claude Code](https://claude.com/claude-code)

Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com>
2026-01-04 14:46:38 -05:00