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>
This commit is contained in:
Patrick Selamy
2026-01-04 19:04:32 -05:00
co-authored by Claude Opus 4.5
parent 95b7876057
commit 31c675bdec
3 changed files with 48 additions and 17 deletions
Generated
+11
View File
@@ -126,6 +126,15 @@ wheels = [
{ url = "https://files.pythonhosted.org/packages/fb/76/641ae371508676492379f16e2fa48f4e2c11741bd63c48be4b12a6b09cba/aiosignal-1.4.0-py3-none-any.whl", hash = "sha256:053243f8b92b990551949e63930a839ff0cf0b0ebbe0597b0f3fb19e1a0fe82e", size = 7490, upload-time = "2025-07-03T22:54:42.156Z" },
]
[[package]]
name = "aiosqlite"
version = "0.22.1"
source = { registry = "https://pypi.org/simple" }
sdist = { url = "https://files.pythonhosted.org/packages/4e/8a/64761f4005f17809769d23e518d915db74e6310474e733e3593cfc854ef1/aiosqlite-0.22.1.tar.gz", hash = "sha256:043e0bd78d32888c0a9ca90fc788b38796843360c855a7262a532813133a0650", size = 14821, upload-time = "2025-12-23T19:25:43.997Z" }
wheels = [
{ url = "https://files.pythonhosted.org/packages/00/b7/e3bf5133d697a08128598c8d0abc5e16377b51465a33756de24fa7dee953/aiosqlite-0.22.1-py3-none-any.whl", hash = "sha256:21c002eb13823fad740196c5a2e9d8e62f6243bd9e7e4a1f87fb5e44ecb4fceb", size = 17405, upload-time = "2025-12-23T19:25:42.139Z" },
]
[[package]]
name = "alembic"
version = "1.17.2"
@@ -1548,6 +1557,7 @@ dependencies = [
[package.optional-dependencies]
dev = [
{ name = "aiosqlite" },
{ name = "mypy" },
{ name = "pre-commit" },
{ name = "pytest" },
@@ -1559,6 +1569,7 @@ dev = [
[package.metadata]
requires-dist = [
{ name = "aiohttp", specifier = ">=3.9.0" },
{ name = "aiosqlite", marker = "extra == 'dev'", specifier = ">=0.19.0" },
{ name = "alembic", specifier = ">=1.13.0" },
{ name = "httpx", specifier = ">=0.25.0" },
{ name = "mypy", marker = "extra == 'dev'", specifier = ">=1.7.0" },