Skip to content
Merged
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
31 changes: 19 additions & 12 deletions tests/test_ibkr_market_data.py
Original file line number Diff line number Diff line change
@@ -1,9 +1,10 @@
from __future__ import annotations

from dataclasses import dataclass
from datetime import date
from datetime import date, datetime, timezone
from types import SimpleNamespace
import unittest
from unittest.mock import patch

from quant_platform_kit.ibkr.market_data import (
fetch_historical_price_candles,
Expand Down Expand Up @@ -300,6 +301,11 @@ def reqMktData(self, contract, *_args):
self.assertEqual(ib.market_data_type_calls, [3, 4, 1, 1])

def test_fetch_option_chain_snapshot_returns_bounded_contract_rows(self) -> None:
class FrozenDateTime(datetime):
@classmethod
def now(cls, tz=None):
return cls(2026, 7, 30, tzinfo=timezone.utc)

class OptionChainIB(FakeIB):
def qualifyContracts(self, contract):
self.qualified.append(contract)
Expand All @@ -325,17 +331,18 @@ def reqMktData(self, contract, *_args):
return FakeTicker(102.5, close=101.8, bid=102.4, ask=102.6)

ib = OptionChainIB()
snapshot = fetch_option_chain_snapshot(
ib,
"TQQQ",
rights=("C",),
min_dte=540,
max_dte=930,
target_dte=730,
wait_seconds=0,
stock_factory=FakeContract,
option_factory=FakeOptionContract,
)
with patch("quant_platform_kit.ibkr.market_data.datetime", FrozenDateTime):
snapshot = fetch_option_chain_snapshot(
ib,
"TQQQ",
rights=("C",),
min_dte=540,
max_dte=930,
target_dte=730,
wait_seconds=0,
stock_factory=FakeContract,
option_factory=FakeOptionContract,
)

self.assertEqual(snapshot["underlier"], "TQQQ")
self.assertEqual(snapshot["spot"], 102.5)
Expand Down