Build 060.28: implement Persistent Checkpoint and Startup Recovery

This commit is contained in:
2026-08-01 20:55:32 +03:00
parent 58e5a12a4d
commit 8c485e32b1
63 changed files with 12430 additions and 92 deletions

View File

@@ -37,7 +37,10 @@ class RecordingJournal:
del event, message, context
def make_settings() -> SimpleNamespace:
def make_settings(
*,
storage_enabled: bool = True,
) -> SimpleNamespace:
return SimpleNamespace(
bot_token="test-token",
bot_parse_mode="HTML",
@@ -46,7 +49,9 @@ def make_settings() -> SimpleNamespace:
exchange_name="dzengi",
default_symbol="BTC/USD_LEVERAGE",
trade_stream=SimpleNamespace(enabled=True),
market_data_storage=SimpleNamespace(enabled=True),
market_data_storage=SimpleNamespace(
enabled=storage_enabled,
),
)
@@ -59,9 +64,11 @@ def test_create_app_builds_one_application_composition(
runtime = object()
storage_lifecycle = object()
storage_sink = object()
storage_repository = object()
storage = SimpleNamespace(
lifecycle=storage_lifecycle,
trade_observation_sink=storage_sink,
trade_repository=storage_repository,
)
journal = RecordingJournal()
observed_runtime_settings: list[object] = []
@@ -103,9 +110,11 @@ def test_create_app_builds_one_application_composition(
received_settings: object,
*,
trade_observation_sink: object,
checkpoint_storage: object,
) -> object:
observed_runtime_settings.append(received_settings)
assert trade_observation_sink is storage_sink
assert checkpoint_storage is storage_repository
return runtime
monkeypatch.setattr(
@@ -162,7 +171,7 @@ def test_runtime_build_error_is_fatal(
monkeypatch.setattr(
app_factory,
"load_settings",
make_settings,
lambda: make_settings(storage_enabled=False),
)
monkeypatch.setattr(
app_factory,
@@ -189,8 +198,11 @@ def test_runtime_build_error_is_fatal(
settings: object,
*,
trade_observation_sink: object,
checkpoint_storage: object,
) -> None:
del settings, trade_observation_sink
del settings
assert trade_observation_sink is None
assert checkpoint_storage is None
raise expected
monkeypatch.setattr(

View File

@@ -557,6 +557,8 @@ def test_application_cancellation_performs_full_cleanup() -> None:
"telegram-polling",
"trade-stream-runtime",
"application-shutdown",
"market-data-storage-startup",
"market-data-storage-shutdown",
}
}
@@ -637,6 +639,8 @@ def test_simultaneous_root_failures_are_awaited_deterministically() -> None:
"telegram-polling",
"trade-stream-runtime",
"application-shutdown",
"market-data-storage-startup",
"market-data-storage-shutdown",
}
}
@@ -647,6 +651,7 @@ def test_repeated_cancellation_does_not_interrupt_cleanup() -> None:
async def scenario() -> None:
dispatcher = FakeDispatcher()
runtime = BlockingStopRuntime()
storage = FakeStorageLifecycle()
bot = FakeBot()
task = asyncio.create_task(
run_application(
@@ -654,6 +659,7 @@ def test_repeated_cancellation_does_not_interrupt_cleanup() -> None:
dispatcher=dispatcher,
runtime=runtime,
bot=bot,
storage_lifecycle=storage,
)
)
)
@@ -664,6 +670,8 @@ def test_repeated_cancellation_does_not_interrupt_cleanup() -> None:
await runtime.stop_entered.wait()
task.cancel()
await asyncio.sleep(0)
assert storage.stop_calls == 0
runtime.stop_release.set()
with pytest.raises(asyncio.CancelledError):
@@ -673,6 +681,7 @@ def test_repeated_cancellation_does_not_interrupt_cleanup() -> None:
assert dispatcher.cancelled.is_set()
assert runtime.stop_calls == 1
assert runtime.stopped.is_set()
assert storage.stop_calls == 1
assert bot.session.close_calls == 1
await asyncio.sleep(0)
assert not {
@@ -685,6 +694,8 @@ def test_repeated_cancellation_does_not_interrupt_cleanup() -> None:
"telegram-polling",
"trade-stream-runtime",
"application-shutdown",
"market-data-storage-startup",
"market-data-storage-shutdown",
}
}

View File

@@ -15,6 +15,10 @@ from src.core.config import (
Settings,
TradeStreamSettings,
)
from src.market_data.storage.contracts import (
TradeCheckpointStorageProtocol,
TradeStorageProtocol,
)
def make_settings(
@@ -159,6 +163,18 @@ def test_builds_shared_graph_without_opening_pool() -> None:
composition.trade_observation_sink._trade_storage
is composition.trade_repository
)
assert (
composition.trade_observation_sink._checkpoint_storage
is composition.trade_repository
)
assert isinstance(
composition.trade_repository,
TradeStorageProtocol,
)
assert isinstance(
composition.trade_repository,
TradeCheckpointStorageProtocol,
)
assert composition.trade_observation_sink._venue == "dzengi"
conninfo = conninfo_to_dict(composition.connection_pool._conninfo)

View File

@@ -2,6 +2,10 @@ from __future__ import annotations
from typing import Any
import pytest
import src.bootstrap.trade_stream_runtime as runtime_factory
from src.bootstrap.market_data_storage import build_market_data_storage
from src.bootstrap.trade_stream_runtime import (
build_trade_stream_production_runtime,
)
@@ -21,7 +25,16 @@ class RecordingTradeObservationSink:
def __init__(self) -> None:
self.observations: list[Trade] = []
def persist(self, trade: Trade) -> None:
def persist_accepted(
self,
trade: Trade,
*,
expected_trade: Trade | None,
) -> None:
del expected_trade
self.observations.append(trade)
def persist_duplicate(self, trade: Trade) -> None:
self.observations.append(trade)
@@ -29,6 +42,9 @@ def make_settings(
*,
enabled: bool = True,
api_key: str = "api-key",
storage_enabled: bool = False,
subscription_ack_timeout_seconds: float = 12.5,
startup_market_buffer_capacity: int = 1_234,
) -> Settings:
return Settings(
bot_token="test-token",
@@ -58,6 +74,12 @@ def make_settings(
heartbeat_timeout_seconds=31.0,
scheduler_interval_seconds=6.0,
recovery_window_ms=123_456,
subscription_ack_timeout_seconds=(
subscription_ack_timeout_seconds
),
startup_market_buffer_capacity=(
startup_market_buffer_capacity
),
),
db_host="localhost",
db_port=5432,
@@ -65,7 +87,7 @@ def make_settings(
db_user="test",
db_password="test",
market_data_storage=MarketDataStorageSettings(
enabled=False,
enabled=storage_enabled,
pool_min_size=1,
pool_max_size=4,
pool_timeout_seconds=10.0,
@@ -81,6 +103,16 @@ def test_disabled_feature_does_not_build_runtime() -> None:
assert build_trade_stream_production_runtime(settings) is None
def test_enabled_storage_requires_enabled_trade_stream() -> None:
settings = make_settings(
enabled=False,
storage_enabled=True,
)
with pytest.raises(RuntimeError, match="Trade Stream"):
build_trade_stream_production_runtime(settings)
def test_builds_runtime_without_starting_lifecycle() -> None:
runtime = build_trade_stream_production_runtime(
make_settings(),
@@ -120,6 +152,116 @@ def test_uses_one_shared_stateful_dependency_graph() -> None:
assert runtime_graph._runtime_scheduler.runtime_supervisor is (
runtime_graph._runtime_supervisor
)
assert runtime_graph._startup_recovery_coordinator is None
def test_persistent_runtime_reuses_one_storage_graph_without_io() -> None:
settings = make_settings(storage_enabled=True)
storage = build_market_data_storage(settings)
assert storage is not None
runtime = build_trade_stream_production_runtime(
settings,
trade_observation_sink=storage.trade_observation_sink,
checkpoint_storage=storage.trade_repository,
)
assert isinstance(runtime, TradeStreamProductionRuntime)
runtime_graph: Any = runtime
startup_recovery = runtime_graph._startup_recovery_coordinator
live_controller = (
runtime_graph._trade_stream_service._consistency_controller
)
assert startup_recovery is not None
assert (
startup_recovery._state_hydrator._checkpoint_storage
is storage.trade_repository
)
assert (
storage.trade_observation_sink._trade_storage
is storage.trade_repository
)
assert (
storage.trade_observation_sink._checkpoint_storage
is storage.trade_repository
)
assert live_controller._trade_observation_sink is (
storage.trade_observation_sink
)
assert startup_recovery._state_hydrator._state_store is (
live_controller._state_store
)
assert startup_recovery._state_hydrator._venue == "dzengi"
assert storage.trade_observation_sink._venue == "dzengi"
assert storage.connection_pool.is_open is False
assert storage.lifecycle.started is False
assert startup_recovery._hydration_task is None
assert startup_recovery._recovery_task is None
assert runtime_graph._startup_task is None
assert runtime_graph._receive_task is None
assert runtime_graph._scheduler_task is None
@pytest.mark.parametrize(
("include_sink", "include_checkpoint"),
(
(False, False),
(True, False),
(False, True),
),
)
def test_enabled_storage_rejects_incomplete_runtime_graph_before_transport(
monkeypatch: pytest.MonkeyPatch,
include_sink: bool,
include_checkpoint: bool,
) -> None:
settings = make_settings(storage_enabled=True)
storage = build_market_data_storage(settings)
assert storage is not None
def unexpected_transport(**kwargs: object) -> None:
del kwargs
raise AssertionError("Transport graph must not be created.")
monkeypatch.setattr(
runtime_factory,
"DzengiWebSocketTransport",
unexpected_transport,
)
with pytest.raises(RuntimeError, match="requires both"):
build_trade_stream_production_runtime(
settings,
trade_observation_sink=(
storage.trade_observation_sink
if include_sink
else None
),
checkpoint_storage=(
storage.trade_repository
if include_checkpoint
else None
),
)
assert storage.connection_pool.is_open is False
def test_disabled_storage_rejects_checkpoint_dependency() -> None:
settings = make_settings()
persistent_settings = make_settings(storage_enabled=True)
storage = build_market_data_storage(persistent_settings)
assert storage is not None
with pytest.raises(RuntimeError, match="requires enabled"):
build_trade_stream_production_runtime(
settings,
checkpoint_storage=storage.trade_repository,
)
def test_optional_storage_sink_is_shared_by_live_and_recovery() -> None:
@@ -172,6 +314,8 @@ def test_applies_explicit_transport_and_runtime_settings() -> None:
"ETH/USD_LEVERAGE",
)
assert runtime_graph._runtime_scheduler.interval_seconds == 6.0
assert runtime_graph._subscription_ack_timeout_seconds == 12.5
assert runtime_graph._startup_market_buffer_capacity == 1_234
assert (
runtime_graph._runtime_supervisor._heartbeat_monitor.timeout_seconds
== 31.0

View File

@@ -15,6 +15,8 @@ _TRADE_STREAM_VARIABLES = (
"TRADE_STREAM_HEARTBEAT_TIMEOUT_SECONDS",
"TRADE_STREAM_SCHEDULER_INTERVAL_SECONDS",
"TRADE_STREAM_RECOVERY_WINDOW_MS",
"TRADE_STREAM_SUBSCRIPTION_ACK_TIMEOUT_SECONDS",
"TRADE_STREAM_STARTUP_MARKET_BUFFER_CAPACITY",
)
_MARKET_DATA_STORAGE_VARIABLES = (
@@ -69,6 +71,8 @@ def test_trade_stream_is_disabled_by_default(
assert settings.trade_stream.enabled is False
assert settings.trade_stream.websocket_url == ""
assert settings.trade_stream.symbols == ()
assert settings.trade_stream.subscription_ack_timeout_seconds == 10.0
assert settings.trade_stream.startup_market_buffer_capacity == 10_000
def test_disabled_trade_stream_ignores_dependent_values(
@@ -76,6 +80,14 @@ def test_disabled_trade_stream_ignores_dependent_values(
) -> None:
prepare_environment(monkeypatch)
monkeypatch.setenv("TRADE_STREAM_OPEN_TIMEOUT_SECONDS", "invalid")
monkeypatch.setenv(
"TRADE_STREAM_SUBSCRIPTION_ACK_TIMEOUT_SECONDS",
"invalid",
)
monkeypatch.setenv(
"TRADE_STREAM_STARTUP_MARKET_BUFFER_CAPACITY",
"invalid",
)
monkeypatch.setenv("TRADE_STREAM_SYMBOLS", ",")
settings = load_settings()
@@ -144,7 +156,7 @@ def test_enabled_trade_stream_parses_independent_settings(
enable_trade_stream(monkeypatch)
monkeypatch.setenv(
"TRADE_STREAM_SYMBOLS",
" ETH/USD_LEVERAGE, BTC/USD_LEVERAGE,ETH/USD_LEVERAGE ",
" eth/usd_leverage, BTC/USD_LEVERAGE,ETH/USD_LEVERAGE ",
)
monkeypatch.setenv("TRADE_STREAM_OPEN_TIMEOUT_SECONDS", "11.5")
monkeypatch.setenv("TRADE_STREAM_PROBE_TIMEOUT_SECONDS", "21")
@@ -152,6 +164,14 @@ def test_enabled_trade_stream_parses_independent_settings(
monkeypatch.setenv("TRADE_STREAM_HEARTBEAT_TIMEOUT_SECONDS", "31")
monkeypatch.setenv("TRADE_STREAM_SCHEDULER_INTERVAL_SECONDS", "6")
monkeypatch.setenv("TRADE_STREAM_RECOVERY_WINDOW_MS", "123456")
monkeypatch.setenv(
"TRADE_STREAM_SUBSCRIPTION_ACK_TIMEOUT_SECONDS",
"12.5",
)
monkeypatch.setenv(
"TRADE_STREAM_STARTUP_MARKET_BUFFER_CAPACITY",
"1234",
)
settings = load_settings()
trade_stream = settings.trade_stream
@@ -168,6 +188,20 @@ def test_enabled_trade_stream_parses_independent_settings(
assert trade_stream.heartbeat_timeout_seconds == 31.0
assert trade_stream.scheduler_interval_seconds == 6.0
assert trade_stream.recovery_window_ms == 123_456
assert trade_stream.subscription_ack_timeout_seconds == 12.5
assert trade_stream.startup_market_buffer_capacity == 1_234
def test_enabled_trade_stream_uses_startup_boundary_defaults(
monkeypatch: pytest.MonkeyPatch,
) -> None:
prepare_environment(monkeypatch)
enable_trade_stream(monkeypatch)
trade_stream = load_settings().trade_stream
assert trade_stream.subscription_ack_timeout_seconds == 10.0
assert trade_stream.startup_market_buffer_capacity == 10_000
@pytest.mark.parametrize(
@@ -180,6 +214,12 @@ def test_enabled_trade_stream_parses_independent_settings(
("TRADE_STREAM_SCHEDULER_INTERVAL_SECONDS", "invalid"),
("TRADE_STREAM_RECOVERY_WINDOW_MS", "1.5"),
("TRADE_STREAM_RECOVERY_WINDOW_MS", "0"),
("TRADE_STREAM_SUBSCRIPTION_ACK_TIMEOUT_SECONDS", "0"),
("TRADE_STREAM_SUBSCRIPTION_ACK_TIMEOUT_SECONDS", "nan"),
("TRADE_STREAM_SUBSCRIPTION_ACK_TIMEOUT_SECONDS", "invalid"),
("TRADE_STREAM_STARTUP_MARKET_BUFFER_CAPACITY", "0"),
("TRADE_STREAM_STARTUP_MARKET_BUFFER_CAPACITY", "-1"),
("TRADE_STREAM_STARTUP_MARKET_BUFFER_CAPACITY", "1.5"),
),
)
def test_enabled_trade_stream_rejects_invalid_numeric_settings(

View File

@@ -0,0 +1,557 @@
from __future__ import annotations
from datetime import datetime, timedelta, timezone
from decimal import Decimal
import pytest
from src.market_data.acquisition.checkpoint.trade_stream_state_hydrator import (
TradeStreamStateHydrator,
TradeStreamStateHydratorProtocol,
)
from src.market_data.acquisition.consistency.trade_stream_state_store import (
TradeStreamStateStore,
)
from src.market_data.acquisition.consistency.trade_stream_state_store_exceptions import (
TradeStreamStateStoreInitializationError,
)
from src.market_data.acquisition.models.trade import (
Trade,
TradeAggressorSide,
)
from src.market_data.storage.contracts import (
PersistentTradeCheckpoint,
TradeCheckpointStorageProtocol,
)
from src.market_data.storage.exceptions import (
MarketDataCheckpointIntegrityError,
)
VENUE = "dzengi"
SYMBOL = "BTC/USD_LEVERAGE"
SECOND_SYMBOL = "ETH/USD_LEVERAGE"
BASE_TIME = datetime(2026, 8, 1, 10, 0, tzinfo=timezone.utc)
def make_trade(
*,
symbol: str = SYMBOL,
trade_id: int = 100,
second: int = 0,
source: str = "dzengi_websocket_trade",
) -> Trade:
return Trade(
symbol=symbol,
trade_id=trade_id,
price=Decimal("65000.25"),
quantity=Decimal("0.001"),
executed_at=BASE_TIME + timedelta(seconds=second),
aggressor_side=TradeAggressorSide.BUY,
source=source,
)
def make_checkpoint(
trade: Trade,
*,
venue: str = VENUE,
revision: int = 1,
) -> PersistentTradeCheckpoint:
return PersistentTradeCheckpoint(
venue=venue,
trade=trade,
revision=revision,
updated_at=BASE_TIME + timedelta(minutes=1),
)
CheckpointResult = PersistentTradeCheckpoint | None | BaseException
TailResult = tuple[Trade, ...] | object | BaseException
AdoptionResult = PersistentTradeCheckpoint | BaseException
class RecordingCheckpointStorage:
"""Настраиваемый fake полного checkpoint-контракта."""
def __init__(
self,
*,
checkpoint_results: dict[
str,
tuple[CheckpointResult, ...],
]
| None = None,
checkpoint_tails: dict[str, TailResult] | None = None,
latest_tails: dict[str, TailResult] | None = None,
adoption_results: dict[str, AdoptionResult] | None = None,
) -> None:
self.checkpoint_results = checkpoint_results or {}
self.checkpoint_tails = checkpoint_tails or {}
self.latest_tails = latest_tails or {}
self.adoption_results = adoption_results or {}
self.calls: list[tuple[object, ...]] = []
self._checkpoint_offsets: dict[str, int] = {}
@property
def operation_names(self) -> list[str]:
return [str(call[0]) for call in self.calls]
def load_checkpoint(
self,
*,
venue: str,
symbol: str,
) -> PersistentTradeCheckpoint | None:
self.calls.append(("load_checkpoint", venue, symbol))
results = self.checkpoint_results.get(symbol, (None,))
offset = self._checkpoint_offsets.get(symbol, 0)
self._checkpoint_offsets[symbol] = offset + 1
result = results[min(offset, len(results) - 1)]
if isinstance(result, BaseException):
raise result
return result
def load_checkpoint_tail(
self,
*,
venue: str,
checkpoint: PersistentTradeCheckpoint,
limit: int,
) -> tuple[Trade, ...]:
self.calls.append(
(
"load_checkpoint_tail",
venue,
checkpoint,
limit,
)
)
result = self.checkpoint_tails.get(
checkpoint.trade.symbol,
(checkpoint.trade,),
)
if isinstance(result, BaseException):
raise result
return result # type: ignore[return-value]
def load_latest_trade_tail(
self,
*,
venue: str,
symbol: str,
limit: int,
) -> tuple[Trade, ...]:
self.calls.append(
("load_latest_trade_tail", venue, symbol, limit)
)
result = self.latest_tails.get(symbol, ())
if isinstance(result, BaseException):
raise result
return result # type: ignore[return-value]
def adopt_existing_trade_as_checkpoint(
self,
*,
venue: str,
trade: Trade,
) -> PersistentTradeCheckpoint:
self.calls.append(
("adopt_existing_trade_as_checkpoint", venue, trade)
)
result = self.adoption_results.get(
trade.symbol,
make_checkpoint(trade, venue=venue),
)
if isinstance(result, BaseException):
raise result
return result
def store_trade_and_advance_checkpoint(
self,
*,
venue: str,
expected_trade: Trade | None,
trade: Trade,
observed_at: datetime,
) -> PersistentTradeCheckpoint:
raise AssertionError(
"Hydrator не должен использовать writer checkpoint."
)
def make_hydrator(
storage: RecordingCheckpointStorage,
*,
state_store: TradeStreamStateStore | None = None,
window_size: int = 3,
) -> tuple[TradeStreamStateHydrator, TradeStreamStateStore]:
store = state_store or TradeStreamStateStore()
return (
TradeStreamStateHydrator(
checkpoint_storage=storage,
state_store=store,
venue=VENUE,
deduplication_window_size=window_size,
),
store,
)
def test_implements_protocol_uses_slots_and_constructor_has_no_io() -> None:
storage = RecordingCheckpointStorage()
hydrator, _ = make_hydrator(storage)
assert isinstance(storage, TradeCheckpointStorageProtocol)
assert isinstance(hydrator, TradeStreamStateHydratorProtocol)
assert not hasattr(hydrator, "__dict__")
assert storage.calls == []
def test_existing_checkpoint_restores_bounded_tail_and_publishes_state() -> None:
trades = (
make_trade(trade_id=100, second=0),
make_trade(trade_id=101, second=1),
make_trade(trade_id=102, second=2),
)
checkpoint = make_checkpoint(trades[-1], revision=7)
storage = RecordingCheckpointStorage(
checkpoint_results={SYMBOL: (checkpoint,)},
checkpoint_tails={SYMBOL: trades},
)
hydrator, store = make_hydrator(storage)
states = hydrator.hydrate(symbols=(SYMBOL,))
assert len(states) == 1
assert states[0].last_trade is trades[-1]
assert states[0].last_trade_id == 102
assert store.get(SYMBOL) is states[0]
assert storage.calls == [
("load_checkpoint", VENUE, SYMBOL),
("load_checkpoint_tail", VENUE, checkpoint, 3),
]
def test_first_adoption_reloads_tail_before_publishing_state() -> None:
candidate_tail = (
make_trade(trade_id=100, second=0),
make_trade(trade_id=101, second=1),
)
adopted = make_checkpoint(candidate_tail[-1])
reloaded_tail = (
make_trade(
trade_id=99,
second=-1,
source="postgres_trade_history",
),
make_trade(
trade_id=100,
second=0,
source="postgres_trade_history",
),
make_trade(
trade_id=101,
second=1,
source="postgres_trade_history",
),
)
storage = RecordingCheckpointStorage(
checkpoint_results={SYMBOL: (None,)},
latest_tails={SYMBOL: candidate_tail},
adoption_results={SYMBOL: adopted},
checkpoint_tails={SYMBOL: reloaded_tail},
)
hydrator, store = make_hydrator(storage)
(state,) = hydrator.hydrate(symbols=(SYMBOL,))
assert state.last_trade is reloaded_tail[-1]
assert store.get(SYMBOL) is state
assert storage.calls == [
("load_checkpoint", VENUE, SYMBOL),
("load_latest_trade_tail", VENUE, SYMBOL, 3),
(
"adopt_existing_trade_as_checkpoint",
VENUE,
candidate_tail[-1],
),
("load_checkpoint_tail", VENUE, adopted, 3),
]
def test_empty_history_rechecks_checkpoint_and_uses_concurrent_value() -> None:
trade = make_trade(trade_id=100)
concurrent_checkpoint = make_checkpoint(trade, revision=2)
storage = RecordingCheckpointStorage(
checkpoint_results={
SYMBOL: (None, concurrent_checkpoint),
},
latest_tails={SYMBOL: ()},
checkpoint_tails={SYMBOL: (trade,)},
)
hydrator, store = make_hydrator(storage)
(state,) = hydrator.hydrate(symbols=(SYMBOL,))
assert state.last_trade is trade
assert store.get(SYMBOL) is state
assert storage.calls == [
("load_checkpoint", VENUE, SYMBOL),
("load_latest_trade_tail", VENUE, SYMBOL, 3),
("load_checkpoint", VENUE, SYMBOL),
(
"load_checkpoint_tail",
VENUE,
concurrent_checkpoint,
3,
),
]
def test_empty_history_without_checkpoint_publishes_empty_state() -> None:
storage = RecordingCheckpointStorage(
checkpoint_results={SYMBOL: (None, None)},
latest_tails={SYMBOL: ()},
)
hydrator, store = make_hydrator(storage)
(state,) = hydrator.hydrate(symbols=(SYMBOL,))
assert state.last_trade is None
assert state.last_trade_id is None
assert store.get(SYMBOL) is state
assert storage.operation_names == [
"load_checkpoint",
"load_latest_trade_tail",
"load_checkpoint",
]
@pytest.mark.parametrize(
"malformed_tail",
(
None,
[],
),
)
def test_rejects_falsey_non_tuple_latest_tail(
malformed_tail: object,
) -> None:
storage = RecordingCheckpointStorage(
checkpoint_results={SYMBOL: (None, None)},
latest_tails={SYMBOL: malformed_tail},
)
hydrator, store = make_hydrator(storage)
with pytest.raises(
MarketDataCheckpointIntegrityError,
match="non-tuple latest Trade tail",
):
hydrator.hydrate(symbols=(SYMBOL,))
assert store.is_empty() is True
assert storage.operation_names == [
"load_checkpoint",
"load_latest_trade_tail",
]
@pytest.mark.parametrize(
("previous_trade_id", "checkpoint_trade_id"),
(
(2**31 - 1, -(2**31)),
(-1, 0),
),
)
def test_restores_tail_across_signed_rollover(
previous_trade_id: int,
checkpoint_trade_id: int,
) -> None:
trades = (
make_trade(trade_id=previous_trade_id, second=0),
make_trade(trade_id=checkpoint_trade_id, second=1),
)
checkpoint = make_checkpoint(trades[-1])
storage = RecordingCheckpointStorage(
checkpoint_results={SYMBOL: (checkpoint,)},
checkpoint_tails={SYMBOL: trades},
)
hydrator, _ = make_hydrator(storage)
(state,) = hydrator.hydrate(symbols=(SYMBOL,))
assert state.last_trade is trades[-1]
assert state.last_trade_id == checkpoint_trade_id
def test_rejects_tail_that_does_not_end_at_checkpoint() -> None:
checkpoint_trade = make_trade(trade_id=102, second=2)
checkpoint = make_checkpoint(checkpoint_trade)
storage = RecordingCheckpointStorage(
checkpoint_results={SYMBOL: (checkpoint,)},
checkpoint_tails={
SYMBOL: (
make_trade(trade_id=100, second=0),
make_trade(trade_id=101, second=1),
)
},
)
hydrator, store = make_hydrator(storage)
with pytest.raises(
MarketDataCheckpointIntegrityError,
match="does not end",
):
hydrator.hydrate(symbols=(SYMBOL,))
assert store.contains(SYMBOL) is False
def test_rejects_tail_larger_than_deduplication_window() -> None:
trades = tuple(
make_trade(trade_id=trade_id, second=trade_id - 100)
for trade_id in range(100, 103)
)
checkpoint = make_checkpoint(trades[-1])
storage = RecordingCheckpointStorage(
checkpoint_results={SYMBOL: (checkpoint,)},
checkpoint_tails={SYMBOL: trades},
)
hydrator, store = make_hydrator(storage, window_size=2)
with pytest.raises(
MarketDataCheckpointIntegrityError,
match="cannot hydrate",
):
hydrator.hydrate(symbols=(SYMBOL,))
assert store.contains(SYMBOL) is False
@pytest.mark.parametrize(
"malformed_tail",
(
[make_trade()],
(object(),),
),
)
def test_rejects_malformed_checkpoint_tail(
malformed_tail: object,
) -> None:
checkpoint = make_checkpoint(make_trade())
storage = RecordingCheckpointStorage(
checkpoint_results={SYMBOL: (checkpoint,)},
checkpoint_tails={SYMBOL: malformed_tail},
)
hydrator, store = make_hydrator(storage)
with pytest.raises(MarketDataCheckpointIntegrityError):
hydrator.hydrate(symbols=(SYMBOL,))
assert store.contains(SYMBOL) is False
def test_storage_error_is_not_swallowed_and_state_is_not_published() -> None:
failure = RuntimeError("storage failed")
storage = RecordingCheckpointStorage(
checkpoint_results={SYMBOL: (failure,)},
)
hydrator, store = make_hydrator(storage)
with pytest.raises(RuntimeError, match="storage failed"):
hydrator.hydrate(symbols=(SYMBOL,))
assert store.contains(SYMBOL) is False
assert storage.operation_names == ["load_checkpoint"]
def test_adoption_error_is_not_swallowed_and_state_is_not_published() -> None:
trade = make_trade()
failure = RuntimeError("adoption failed")
storage = RecordingCheckpointStorage(
checkpoint_results={SYMBOL: (None,)},
latest_tails={SYMBOL: (trade,)},
adoption_results={SYMBOL: failure},
)
hydrator, store = make_hydrator(storage)
with pytest.raises(RuntimeError, match="adoption failed"):
hydrator.hydrate(symbols=(SYMBOL,))
assert store.contains(SYMBOL) is False
assert storage.operation_names == [
"load_checkpoint",
"load_latest_trade_tail",
"adopt_existing_trade_as_checkpoint",
]
def test_rejects_adopted_checkpoint_for_different_candidate() -> None:
candidate = make_trade(trade_id=100)
different_trade = make_trade(trade_id=101, second=1)
storage = RecordingCheckpointStorage(
checkpoint_results={SYMBOL: (None,)},
latest_tails={SYMBOL: (candidate,)},
adoption_results={
SYMBOL: make_checkpoint(different_trade),
},
)
hydrator, store = make_hydrator(storage)
with pytest.raises(
MarketDataCheckpointIntegrityError,
match="does not match candidate",
):
hydrator.hydrate(symbols=(SYMBOL,))
assert store.contains(SYMBOL) is False
assert "load_checkpoint_tail" not in storage.operation_names
def test_second_symbol_failure_does_not_publish_first_state() -> None:
first_trade = make_trade(symbol=SYMBOL)
first_checkpoint = make_checkpoint(first_trade)
second_failure = RuntimeError("second symbol failed")
storage = RecordingCheckpointStorage(
checkpoint_results={
SYMBOL: (first_checkpoint,),
SECOND_SYMBOL: (second_failure,),
},
checkpoint_tails={SYMBOL: (first_trade,)},
)
hydrator, store = make_hydrator(storage)
with pytest.raises(RuntimeError, match="second symbol failed"):
hydrator.hydrate(symbols=(SYMBOL, SECOND_SYMBOL))
assert store.contains(SYMBOL) is False
assert store.contains(SECOND_SYMBOL) is False
assert storage.operation_names == [
"load_checkpoint",
"load_checkpoint_tail",
"load_checkpoint",
]
def test_nonempty_store_is_rejected_before_storage_io() -> None:
storage = RecordingCheckpointStorage()
store = TradeStreamStateStore()
existing_state = store.get_or_create(SYMBOL)
hydrator, _ = make_hydrator(storage, state_store=store)
with pytest.raises(TradeStreamStateStoreInitializationError):
hydrator.hydrate(symbols=(SYMBOL, SECOND_SYMBOL))
assert store.get(SYMBOL) is existing_state
assert store.contains(SECOND_SYMBOL) is False
assert storage.calls == []

View File

@@ -78,12 +78,27 @@ class RecordingTradeObservationSink:
) -> None:
self.error = error
self.observations: list[Trade] = []
self.accepted: list[tuple[Trade, Trade | None]] = []
self.duplicates: list[Trade] = []
def persist(
def persist_accepted(
self,
trade: Trade,
*,
expected_trade: Trade | None,
) -> None:
self.observations.append(trade)
self.accepted.append((trade, expected_trade))
if self.error is not None:
raise self.error
def persist_duplicate(
self,
trade: Trade,
) -> None:
self.observations.append(trade)
self.duplicates.append(trade)
if self.error is not None:
raise self.error
@@ -221,6 +236,8 @@ def test_persists_trade_before_advancing_checkpoint(
assert result is trade
assert sink.observations == [trade]
assert sink.accepted == [(trade, None)]
assert sink.duplicates == []
assert state.last_trade is trade
@@ -247,6 +264,10 @@ def test_persistence_failure_leaves_checkpoint_unchanged(
state = state_store.get(first_trade.symbol)
assert error_info.value is storage_error
assert sink.accepted == [
(first_trade, None),
(failed_trade, first_trade),
]
assert state.last_trade is first_trade
assert state.last_trade_id == first_trade.trade_id
assert failed_trade.trade_id not in state._trades
@@ -280,10 +301,59 @@ def test_valid_duplicate_is_persisted_without_checkpoint_advance(
websocket_trade,
rest_duplicate,
]
assert sink.accepted == [(websocket_trade, None)]
assert sink.duplicates == [rest_duplicate]
assert state.last_trade is websocket_trade
assert state.last_trade_id == websocket_trade.trade_id
def test_duplicate_persistence_failure_keeps_checkpoint(
state_store: TradeStreamStateStore,
) -> None:
storage_error = RuntimeError("storage failed")
sink = RecordingTradeObservationSink()
controller = TradeStreamConsistencyController(
state_store=state_store,
trade_observation_sink=sink,
)
original = _trade(source="dzengi_websocket_trade")
duplicate = _trade(source="dzengi")
controller.accept(original)
sink.error = storage_error
with pytest.raises(RuntimeError, match="storage failed") as error_info:
controller.accept(duplicate)
state = state_store.get(original.symbol)
assert error_info.value is storage_error
assert sink.accepted == [(original, None)]
assert sink.duplicates == [duplicate]
assert state.last_trade is original
assert state.last_trade_id == original.trade_id
def test_rollover_advance_uses_previous_trade_as_expected_checkpoint(
state_store: TradeStreamStateStore,
) -> None:
sink = RecordingTradeObservationSink()
controller = TradeStreamConsistencyController(
state_store=state_store,
trade_observation_sink=sink,
)
previous = _trade(trade_id=2**31 - 1)
current = _trade(trade_id=-(2**31))
controller.accept(previous)
controller.accept(current)
assert sink.accepted == [
(previous, None),
(current, previous),
]
assert sink.duplicates == []
def test_invalid_trades_do_not_reach_persistence_sink(
state_store: TradeStreamStateStore,
) -> None:

View File

@@ -420,6 +420,88 @@ def test_checkpoint_trade_id_matches_last_trade_id() -> None:
assert state.last_trade.trade_id == state.last_trade_id
def test_checkpoint_callback_receives_expected_previous_trade() -> None:
state = TradeStreamState(symbol="BTCUSD")
first_trade = _trade(trade_id=100)
second_trade = _trade(trade_id=101)
calls: list[tuple[Trade, Trade | None]] = []
def before_checkpoint(
trade: Trade,
*,
expected_trade: Trade | None,
) -> None:
calls.append((trade, expected_trade))
assert state.last_trade is expected_trade
state.accept(
first_trade,
before_checkpoint=before_checkpoint,
)
state.accept(
second_trade,
before_checkpoint=before_checkpoint,
)
assert calls == [
(first_trade, None),
(second_trade, first_trade),
]
assert state.last_trade is second_trade
def test_duplicate_uses_only_duplicate_callback() -> None:
state = TradeStreamState(
symbol="BTCUSD",
deduplication_window_size=3,
)
first_trade = _trade(trade_id=100)
latest_trade = _trade(trade_id=101)
duplicate = _trade(
trade_id=100,
source="dzengi_websocket_trade",
)
checkpoint_calls: list[tuple[Trade, Trade | None]] = []
duplicate_calls: list[Trade] = []
state.accept(first_trade)
state.accept(latest_trade)
result = state.accept(
duplicate,
before_checkpoint=lambda trade, expected_trade: (
checkpoint_calls.append((trade, expected_trade))
),
on_duplicate=duplicate_calls.append,
)
assert result is None
assert checkpoint_calls == []
assert duplicate_calls == [duplicate]
assert state.last_trade is latest_trade
def test_duplicate_callback_failure_keeps_checkpoint() -> None:
state = TradeStreamState(symbol="BTCUSD")
original = _trade(source="dzengi_websocket_trade")
duplicate = _trade(source="dzengi")
storage_error = RuntimeError("storage failed")
state.accept(original)
def fail_duplicate(trade: Trade) -> None:
assert trade is duplicate
raise storage_error
with pytest.raises(RuntimeError, match="storage failed") as error_info:
state.accept(
duplicate,
on_duplicate=fail_duplicate,
)
assert error_info.value is storage_error
assert state.last_trade is original
assert state.last_trade_id == original.trade_id
def test_rejects_empty_symbol() -> None:
with pytest.raises(ValueError):
TradeStreamState(symbol="")
@@ -440,3 +522,184 @@ def test_rejects_non_positive_window_size(
symbol="BTCUSD",
deduplication_window_size=window_size,
)
def test_from_history_builds_empty_state() -> None:
state = TradeStreamState.from_history(
symbol="BTCUSD",
trades=(),
deduplication_window_size=3,
)
assert state.symbol == "BTCUSD"
assert state.deduplication_window_size == 3
assert state.last_trade_id is None
assert state.last_trade is None
def test_from_history_restores_valid_deduplication_window() -> None:
first_trade = _trade(trade_id=100)
second_trade = _trade(trade_id=101)
last_trade = _trade(trade_id=102)
state = TradeStreamState.from_history(
symbol="BTCUSD",
trades=(
first_trade,
second_trade,
last_trade,
),
deduplication_window_size=3,
)
assert state.last_trade is last_trade
assert state.last_trade_id == last_trade.trade_id
assert state.accept(_trade(trade_id=100)) is None
assert state.last_trade is last_trade
def test_from_history_keeps_exact_window_boundary() -> None:
first_trade = _trade(trade_id=100)
second_trade = _trade(trade_id=101)
third_trade = _trade(trade_id=102)
state = TradeStreamState.from_history(
symbol="BTCUSD",
trades=(
first_trade,
second_trade,
third_trade,
),
deduplication_window_size=3,
)
next_trade = _trade(trade_id=103)
assert state.accept(next_trade) is next_trade
assert state.accept(_trade(trade_id=101)) is None
with pytest.raises(TradeOrderingError):
state.accept(_trade(trade_id=100))
@pytest.mark.parametrize(
("first_trade_id", "next_trade_id"),
(
(2**31 - 1, -(2**31)),
(-1, 0),
),
)
def test_from_history_restores_signed_rollover_sequence(
first_trade_id: int,
next_trade_id: int,
) -> None:
first_trade = _trade(trade_id=first_trade_id)
next_trade = _trade(trade_id=next_trade_id)
state = TradeStreamState.from_history(
symbol="BTCUSD",
trades=(first_trade, next_trade),
deduplication_window_size=2,
)
assert state.last_trade is next_trade
assert state.last_trade_id == next_trade_id
assert state.accept(_trade(trade_id=first_trade_id)) is None
def test_from_history_accepts_next_trade_after_restoration() -> None:
last_history_trade = _trade(trade_id=101)
state = TradeStreamState.from_history(
symbol="BTCUSD",
trades=(
_trade(trade_id=100),
last_history_trade,
),
deduplication_window_size=3,
)
next_trade = _trade(trade_id=102)
result = state.accept(next_trade)
assert result is next_trade
assert state.last_trade is next_trade
assert state.last_trade_id == next_trade.trade_id
def test_from_history_rejects_identical_duplicate_strictly() -> None:
original = _trade(trade_id=100)
duplicate = _trade(
trade_id=100,
source="dzengi_websocket_trade",
)
with pytest.raises(
TradeConsistencyError,
match="duplicate Trade",
):
TradeStreamState.from_history(
symbol="BTCUSD",
trades=(original, duplicate),
deduplication_window_size=2,
)
def test_from_history_rejects_conflicting_trade_id() -> None:
original = _trade(
trade_id=100,
price=Decimal("50000.00"),
)
conflict = _trade(
trade_id=100,
price=Decimal("50001.00"),
)
with pytest.raises(TradeConsistencyError):
TradeStreamState.from_history(
symbol="BTCUSD",
trades=(original, conflict),
deduplication_window_size=2,
)
def test_from_history_rejects_reverse_sequence() -> None:
with pytest.raises(TradeOrderingError):
TradeStreamState.from_history(
symbol="BTCUSD",
trades=(
_trade(trade_id=101),
_trade(trade_id=100),
),
deduplication_window_size=2,
)
def test_from_history_rejects_half_cycle_sequence() -> None:
with pytest.raises(
ValueError,
match="exactly half",
):
TradeStreamState.from_history(
symbol="BTCUSD",
trades=(
_trade(trade_id=0),
_trade(trade_id=-(2**31)),
),
deduplication_window_size=2,
)
def test_from_history_rejects_history_larger_than_window() -> None:
with pytest.raises(
ValueError,
match="fit deduplication_window_size",
):
TradeStreamState.from_history(
symbol="BTCUSD",
trades=(
_trade(trade_id=100),
_trade(trade_id=101),
_trade(trade_id=102),
),
deduplication_window_size=2,
)

View File

@@ -2,6 +2,8 @@
from __future__ import annotations
from typing import cast
import pytest
from src.market_data.acquisition.consistency.trade_stream_state import (
@@ -12,6 +14,7 @@ from src.market_data.acquisition.consistency.trade_stream_state_store import (
)
from src.market_data.acquisition.consistency.trade_stream_state_store_exceptions import (
TradeStreamStateNotFoundError,
TradeStreamStateStoreInitializationError,
)
from src.market_data.acquisition.consistency.trade_stream_state_store_protocol import (
TradeStreamStateStoreProtocol,
@@ -140,4 +143,106 @@ def test_clear_is_idempotent_for_empty_store() -> None:
store.clear()
store.clear()
assert store.contains("BTCUSD") is False
assert store.contains("BTCUSD") is False
def test_initialize_publishes_all_states_preserving_identity() -> None:
store = TradeStreamStateStore()
btc_state = TradeStreamState(symbol="BTCUSD")
eth_state = TradeStreamState(symbol="ETHUSD")
store.initialize((btc_state, eth_state))
assert store.get("BTCUSD") is btc_state
assert store.get("ETHUSD") is eth_state
def test_initialize_rejects_duplicate_symbols_atomically() -> None:
store = TradeStreamStateStore()
with pytest.raises(
TradeStreamStateStoreInitializationError,
match="повторно",
):
store.initialize(
(
TradeStreamState(symbol="BTCUSD"),
TradeStreamState(symbol="BTCUSD"),
)
)
assert store.contains("BTCUSD") is False
def test_initialize_rejects_invalid_item_atomically() -> None:
store = TradeStreamStateStore()
with pytest.raises(
TypeError,
match="TradeStreamState",
):
store.initialize(
cast(
tuple[TradeStreamState, ...],
(
TradeStreamState(symbol="BTCUSD"),
object(),
),
),
)
assert store.contains("BTCUSD") is False
def test_initialize_rejects_non_empty_store() -> None:
store = TradeStreamStateStore()
existing_state = store.get_or_create("BTCUSD")
with pytest.raises(
TradeStreamStateStoreInitializationError,
match="уже содержит",
):
store.initialize(
(TradeStreamState(symbol="ETHUSD"),)
)
assert store.get("BTCUSD") is existing_state
assert store.contains("ETHUSD") is False
def test_initialize_can_be_called_only_once() -> None:
store = TradeStreamStateStore()
original_state = TradeStreamState(symbol="BTCUSD")
store.initialize((original_state,))
with pytest.raises(
TradeStreamStateStoreInitializationError,
match="уже содержит",
):
store.initialize(
(TradeStreamState(symbol="ETHUSD"),)
)
assert store.get("BTCUSD") is original_state
assert store.contains("ETHUSD") is False
def test_empty_initialize_is_still_one_time_initialization() -> None:
store = TradeStreamStateStore()
store.initialize(())
with pytest.raises(TradeStreamStateStoreInitializationError):
store.initialize(())
def test_clear_allows_store_to_be_initialized_again() -> None:
store = TradeStreamStateStore()
store.initialize((TradeStreamState(symbol="BTCUSD"),))
store.clear()
eth_state = TradeStreamState(symbol="ETHUSD")
store.initialize((eth_state,))
assert store.contains("BTCUSD") is False
assert store.get("ETHUSD") is eth_state

View File

@@ -266,8 +266,8 @@ def test_reconnect_restore_boundary_and_recovery_order() -> None:
order,
) = create_coordinator(
symbols=(
f" {ETH} ",
BTC,
f" {ETH.lower()} ",
BTC.lower(),
ETH,
),
)

View File

@@ -119,6 +119,18 @@ class FakeStateStore:
self.remove_calls: list[str] = []
self.clear_calls = 0
def initialize(
self,
states: tuple[TradeStreamState, ...],
) -> None:
self._states = {
state.symbol: state
for state in states
}
def is_empty(self) -> bool:
return not self._states
def get_or_create(
self,
symbol: str,

View File

@@ -0,0 +1,462 @@
from __future__ import annotations
import asyncio
import threading
import pytest
from src.market_data.acquisition.consistency.trade_stream_state import (
TradeStreamState,
)
from src.market_data.acquisition.recovery.trade_recovery_result import (
TradeRecoveryResult,
)
from src.market_data.acquisition.runtime.live_processing_gate import (
RuntimeLiveProcessingGate,
)
from src.market_data.acquisition.runtime.runtime_startup_recovery_coordinator import (
RuntimeStartupRecoveryCoordinator,
RuntimeStartupRecoveryProtocol,
)
BTC = "BTC/USD_LEVERAGE"
ETH = "ETH/USD_LEVERAGE"
RECOVERY_END_TIME_MS = 1_785_326_405_123
HYDRATION_TASK_NAME = "trade-stream-state-hydration"
RECOVERY_TASK_NAME = "trade-stream-startup-recovery"
class FakeStateHydrator:
def __init__(
self,
*,
states: tuple[TradeStreamState, ...],
error: Exception | None = None,
started: threading.Event | None = None,
release: threading.Event | None = None,
) -> None:
self._states = states
self._error = error
self._started = started
self._release = release
self.calls: list[tuple[str, ...]] = []
self.thread_ids: list[int] = []
def hydrate(
self,
*,
symbols: tuple[str, ...],
) -> tuple[TradeStreamState, ...]:
self.calls.append(symbols)
self.thread_ids.append(threading.get_ident())
if self._started is not None:
self._started.set()
if self._release is not None and not self._release.wait(
timeout=2.0,
):
raise AssertionError("hydration release was not signalled")
if self._error is not None:
raise self._error
return self._states
class FakeRecoveryCoordinator:
def __init__(
self,
*,
error: Exception | None = None,
started: threading.Event | None = None,
release: threading.Event | None = None,
) -> None:
self._error = error
self._started = started
self._release = release
self.calls: list[tuple[str, int]] = []
self.thread_ids: list[int] = []
def recover(
self,
*,
symbol: str,
recovery_end_time: int,
) -> TradeRecoveryResult:
self.calls.append(
(
symbol,
recovery_end_time,
)
)
self.thread_ids.append(threading.get_ident())
if self._started is not None:
self._started.set()
if self._release is not None and not self._release.wait(
timeout=2.0,
):
raise AssertionError("recovery release was not signalled")
if self._error is not None:
raise self._error
return TradeRecoveryResult(
symbol=symbol,
requested_start_time=recovery_end_time,
requested_end_time=recovery_end_time,
recovered_trades=(),
)
class RecordingClock:
def __init__(
self,
value: object = RECOVERY_END_TIME_MS,
) -> None:
self._value = value
self.calls = 0
def __call__(self) -> int:
self.calls += 1
return self._value # type: ignore[return-value]
def create_coordinator(
*,
symbols: tuple[str, ...] = (BTC,),
states: tuple[TradeStreamState, ...] | None = None,
hydration_error: Exception | None = None,
hydration_started: threading.Event | None = None,
hydration_release: threading.Event | None = None,
recovery_error: Exception | None = None,
recovery_started: threading.Event | None = None,
recovery_release: threading.Event | None = None,
clock_value: object = RECOVERY_END_TIME_MS,
) -> tuple[
RuntimeStartupRecoveryCoordinator,
FakeStateHydrator,
FakeRecoveryCoordinator,
RuntimeLiveProcessingGate,
RecordingClock,
]:
hydrated_states = states or (
TradeStreamState(symbol=BTC),
)
hydrator = FakeStateHydrator(
states=hydrated_states,
error=hydration_error,
started=hydration_started,
release=hydration_release,
)
recovery = FakeRecoveryCoordinator(
error=recovery_error,
started=recovery_started,
release=recovery_release,
)
gate = RuntimeLiveProcessingGate()
clock = RecordingClock(clock_value)
coordinator = RuntimeStartupRecoveryCoordinator(
state_hydrator=hydrator,
recovery_coordinator=recovery,
live_processing_gate=gate,
symbols=symbols,
clock=clock,
)
return coordinator, hydrator, recovery, gate, clock
async def wait_until(
predicate: object,
) -> None:
for _ in range(100):
if callable(predicate) and predicate():
return
await asyncio.sleep(0)
raise AssertionError("condition was not reached")
def test_implements_protocol_uses_slots_and_constructor_has_no_io() -> None:
coordinator, hydrator, recovery, gate, clock = create_coordinator(
symbols=(
f" {ETH.lower()} ",
BTC.lower(),
ETH,
),
)
assert isinstance(coordinator, RuntimeStartupRecoveryProtocol)
assert not hasattr(coordinator, "__dict__")
assert coordinator.live_processing_gate is gate
assert coordinator.symbols == (BTC, ETH)
assert hydrator.calls == []
assert recovery.calls == []
assert clock.calls == 0
assert gate.locked is False
@pytest.mark.parametrize(
("symbols", "error_type"),
[
([], TypeError),
((), ValueError),
(("", " "), ValueError),
((BTC, 1), TypeError),
],
)
def test_rejects_invalid_symbols(
symbols: object,
error_type: type[Exception],
) -> None:
with pytest.raises(error_type):
create_coordinator(
symbols=symbols, # type: ignore[arg-type]
)
def test_hydration_is_idempotent_and_runs_off_event_loop() -> None:
async def scenario() -> tuple[
tuple[TradeStreamState, ...],
tuple[TradeStreamState, ...],
int,
]:
states = (
TradeStreamState(symbol=BTC),
)
coordinator, hydrator, *_ = create_coordinator(
states=states,
)
event_loop_thread_id = threading.get_ident()
first = await coordinator.hydrate_once()
second = await coordinator.hydrate_once()
assert hydrator.calls == [(BTC,)]
assert len(hydrator.thread_ids) == 1
return first, second, event_loop_thread_id
first, second, event_loop_thread_id = asyncio.run(scenario())
assert first is second
assert first[0].symbol == BTC
coordinator, hydrator, *_ = create_coordinator()
asyncio.run(coordinator.hydrate_once())
assert hydrator.thread_ids[0] != event_loop_thread_id
def test_recovery_uses_one_clock_boundary_and_symbol_order() -> None:
async def scenario() -> tuple[
tuple[TradeRecoveryResult, ...],
FakeRecoveryCoordinator,
RecordingClock,
int,
]:
coordinator, _, recovery, gate, clock = create_coordinator(
symbols=(
f" {ETH.lower()} ",
BTC.lower(),
ETH,
),
)
event_loop_thread_id = threading.get_ident()
async with gate:
results = await coordinator.recover_after_ack()
return results, recovery, clock, event_loop_thread_id
results, recovery, clock, event_loop_thread_id = asyncio.run(
scenario()
)
assert tuple(result.symbol for result in results) == (BTC, ETH)
assert recovery.calls == [
(BTC, RECOVERY_END_TIME_MS),
(ETH, RECOVERY_END_TIME_MS),
]
assert clock.calls == 1
assert len(set(recovery.thread_ids)) == 1
assert recovery.thread_ids[0] != event_loop_thread_id
def test_cancelled_hydration_waits_and_caches_success() -> None:
async def scenario() -> tuple[
RuntimeStartupRecoveryCoordinator,
FakeStateHydrator,
tuple[TradeStreamState, ...],
]:
started = threading.Event()
release = threading.Event()
states = (
TradeStreamState(symbol=BTC),
)
coordinator, hydrator, *_ = create_coordinator(
states=states,
hydration_started=started,
hydration_release=release,
)
task = asyncio.create_task(
coordinator.hydrate_once(),
)
await wait_until(started.is_set)
joined_task = asyncio.create_task(
coordinator.hydrate_once(),
)
await asyncio.sleep(0)
task.cancel()
await asyncio.sleep(0)
assert task.done() is False
assert joined_task.done() is False
release.set()
with pytest.raises(asyncio.CancelledError):
await task
cached = await joined_task
assert await coordinator.hydrate_once() is cached
return coordinator, hydrator, cached
coordinator, hydrator, cached = asyncio.run(scenario())
assert cached[0].symbol == BTC
assert hydrator.calls == [(BTC,)]
assert coordinator._hydration_task is None
def test_cancelled_recovery_waits_for_worker_completion() -> None:
async def scenario() -> tuple[
RuntimeStartupRecoveryCoordinator,
FakeRecoveryCoordinator,
]:
started = threading.Event()
release = threading.Event()
coordinator, _, recovery, gate, _ = create_coordinator(
recovery_started=started,
recovery_release=release,
)
async def run_recovery() -> None:
async with gate:
await coordinator.recover_after_ack()
task = asyncio.create_task(run_recovery())
await wait_until(started.is_set)
task.cancel()
await asyncio.sleep(0)
assert task.done() is False
assert gate.locked is True
release.set()
with pytest.raises(asyncio.CancelledError):
await task
assert gate.locked is False
return coordinator, recovery
coordinator, recovery = asyncio.run(scenario())
assert recovery.calls == [(BTC, RECOVERY_END_TIME_MS)]
assert coordinator._recovery_task is None
@pytest.mark.parametrize(
"operation",
[
"hydration",
"recovery",
],
)
def test_worker_error_identity_is_preserved(
operation: str,
) -> None:
error = RuntimeError(f"{operation} failed")
async def scenario() -> RuntimeStartupRecoveryCoordinator:
coordinator, *_ = create_coordinator(
hydration_error=(
error if operation == "hydration" else None
),
recovery_error=(
error if operation == "recovery" else None
),
)
with pytest.raises(RuntimeError) as raised:
if operation == "hydration":
await coordinator.hydrate_once()
else:
await coordinator.recover_after_ack()
assert raised.value is error
return coordinator
coordinator = asyncio.run(scenario())
assert coordinator._hydration_task is None
assert coordinator._recovery_task is None
@pytest.mark.parametrize(
("clock_value", "error_type"),
[
(True, TypeError),
(1.5, TypeError),
("1", TypeError),
(-1, ValueError),
],
)
def test_invalid_clock_result_fails_before_recovery(
clock_value: object,
error_type: type[Exception],
) -> None:
async def scenario() -> tuple[
FakeRecoveryCoordinator,
RecordingClock,
RuntimeStartupRecoveryCoordinator,
]:
coordinator, _, recovery, _, clock = create_coordinator(
clock_value=clock_value,
)
with pytest.raises(error_type):
await coordinator.recover_after_ack()
return recovery, clock, coordinator
recovery, clock, coordinator = asyncio.run(scenario())
assert recovery.calls == []
assert clock.calls == 1
assert coordinator._recovery_task is None
def test_completed_operations_leave_no_owned_tasks() -> None:
async def scenario() -> None:
coordinator, *_ = create_coordinator()
await coordinator.hydrate_once()
await coordinator.recover_after_ack()
pending_owned_tasks = {
task.get_name()
for task in asyncio.all_tasks()
if task is not asyncio.current_task()
and not task.done()
and task.get_name()
in {
HYDRATION_TASK_NAME,
RECOVERY_TASK_NAME,
}
}
assert pending_owned_tasks == set()
assert coordinator._hydration_task is None
assert coordinator._recovery_task is None
asyncio.run(scenario())

View File

@@ -24,10 +24,15 @@ from src.market_data.acquisition.consistency.trade_stream_consistency_controller
from src.market_data.acquisition.consistency.trade_stream_state_store import (
TradeStreamStateStore,
)
from src.market_data.acquisition.consistency.trade_stream_state import (
TradeStreamState,
)
from src.market_data.acquisition.exceptions import (
WebSocketControlMessageError,
WebSocketMessageDecodeError,
WebSocketMessageRoutingError,
WebSocketStartupMarketBufferOverflowError,
WebSocketSubscriptionAckTimeoutError,
WebSocketTransportError,
)
from src.market_data.acquisition.models.trade import Trade
@@ -37,6 +42,9 @@ from src.market_data.acquisition.runtime.runtime_events import (
DisconnectedEvent,
MessageReceivedEvent,
)
from src.market_data.acquisition.recovery.trade_recovery_result import (
TradeRecoveryResult,
)
from src.market_data.acquisition.runtime.live_processing_gate import (
RuntimeLiveProcessingGate,
)
@@ -66,6 +74,7 @@ from src.market_data.acquisition.runtime.websocket_protocol import (
from src.market_data.acquisition.trade_stream_acquisition_service import (
TradeStreamAcquisitionService,
)
from src.market_data.acquisition.symbols import normalize_symbol
SYMBOL = "BTC/USD_LEVERAGE"
@@ -153,6 +162,8 @@ class FakeTransport:
self._incoming = deque(incoming)
self._message_available = asyncio.Event()
self.receive_calls = 0
self.active_receivers = 0
self.max_active_receivers = 0
self.probe_calls = 0
self.waiting = asyncio.Event()
@@ -171,24 +182,32 @@ class FakeTransport:
async def receive(self) -> str | bytes:
self.receive_calls += 1
self._calls.append("transport.receive")
self.active_receivers += 1
self.max_active_receivers = max(
self.max_active_receivers,
self.active_receivers,
)
try:
while not self._incoming:
self.waiting.set()
await self._message_available.wait()
self._message_available.clear()
except asyncio.CancelledError:
self._calls.append(
"transport.receive.cancelled",
)
raise
try:
while not self._incoming:
self.waiting.set()
await self._message_available.wait()
self._message_available.clear()
except asyncio.CancelledError:
self._calls.append(
"transport.receive.cancelled",
)
raise
result = self._incoming.popleft()
result = self._incoming.popleft()
if isinstance(result, BaseException):
raise result
if isinstance(result, BaseException):
raise result
return result
return result
finally:
self.active_receivers -= 1
def feed(
self,
@@ -302,9 +321,9 @@ class FakeReconnectRecoveryCoordinator:
self._symbols = tuple(
sorted(
{
symbol.strip()
normalize_symbol(symbol)
for symbol in valid_symbols
if symbol.strip()
if normalize_symbol(symbol)
}
)
)
@@ -369,6 +388,80 @@ class FakeReconnectRecoveryCoordinator:
raise self._error
class FakeStartupRecoveryCoordinator:
def __init__(
self,
*,
calls: list[str],
live_processing_gate: RuntimeLiveProcessingGate,
symbols: tuple[str, ...],
hydration_error: Exception | None = None,
recovery_error: Exception | None = None,
hydration_release: asyncio.Event | None = None,
recovery_release: asyncio.Event | None = None,
) -> None:
self._calls = calls
self._live_processing_gate = live_processing_gate
self._symbols = tuple(
sorted(
{
normalize_symbol(symbol)
for symbol in symbols
if normalize_symbol(symbol)
}
)
)
self._hydration_error = hydration_error
self._recovery_error = recovery_error
self._hydration_release = hydration_release
self._recovery_release = recovery_release
self.hydration_entered = asyncio.Event()
self.recovery_entered = asyncio.Event()
self.hydrate_calls = 0
self.recover_calls = 0
@property
def live_processing_gate(
self,
) -> RuntimeLiveProcessingGate:
return self._live_processing_gate
@property
def symbols(self) -> tuple[str, ...]:
return self._symbols
async def hydrate_once(self) -> tuple[TradeStreamState, ...]:
self.hydrate_calls += 1
self._calls.append("startup_recovery.hydrate")
self.hydration_entered.set()
if self._hydration_release is not None:
await self._hydration_release.wait()
if self._hydration_error is not None:
raise self._hydration_error
return ()
async def recover_after_ack(self) -> tuple[TradeRecoveryResult, ...]:
self.recover_calls += 1
self._calls.append("startup_recovery.recover")
self.recovery_entered.set()
if not self._live_processing_gate.locked:
raise AssertionError(
"Production Runtime должен удерживать startup gate."
)
if self._recovery_release is not None:
await self._recovery_release.wait()
if self._recovery_error is not None:
raise self._recovery_error
return ()
class FakeTradeStreamService:
def __init__(
self,
@@ -623,6 +716,15 @@ class RuntimeDependencies:
scheduler_uses_different_transport: bool = False,
symbols: tuple[str, ...] = (SYMBOL,),
recovery_symbols: tuple[str, ...] | None = None,
startup_recovery_enabled: bool = False,
startup_hydration_error: Exception | None = None,
startup_recovery_error: Exception | None = None,
startup_hydration_release: asyncio.Event | None = None,
startup_recovery_release: asyncio.Event | None = None,
startup_recovery_symbols: tuple[str, ...] | None = None,
startup_recovery_uses_different_gate: bool = False,
subscription_ack_timeout_seconds: float = 10.0,
startup_market_buffer_capacity: int = 10_000,
) -> None:
self.calls: list[str] = []
self.session = FakeSession(
@@ -663,6 +765,27 @@ class RuntimeDependencies:
else recovery_symbols
),
)
self.startup_recovery = (
FakeStartupRecoveryCoordinator(
calls=self.calls,
live_processing_gate=(
RuntimeLiveProcessingGate()
if startup_recovery_uses_different_gate
else self.live_processing_gate
),
symbols=(
symbols
if startup_recovery_symbols is None
else startup_recovery_symbols
),
hydration_error=startup_hydration_error,
recovery_error=startup_recovery_error,
hydration_release=startup_hydration_release,
recovery_release=startup_recovery_release,
)
if startup_recovery_enabled
else None
)
self.supervisor = FakeRuntimeSupervisor(
calls=self.calls,
start_error=supervisor_start_error,
@@ -711,6 +834,13 @@ class RuntimeDependencies:
runtime_supervisor=self.supervisor,
runtime_scheduler=self.scheduler,
symbols=symbols,
startup_recovery_coordinator=self.startup_recovery,
subscription_ack_timeout_seconds=(
subscription_ack_timeout_seconds
),
startup_market_buffer_capacity=(
startup_market_buffer_capacity
),
)
@@ -773,6 +903,21 @@ def test_rejects_invalid_symbols(
)
def test_canonicalizes_case_variants_across_runtime_graph() -> None:
dependencies = RuntimeDependencies(
symbols=(
f" {SYMBOL.lower()} ",
SYMBOL,
),
startup_recovery_enabled=True,
)
assert dependencies.runtime._symbols == (SYMBOL,)
assert dependencies.reconnect_recovery.symbols == (SYMBOL,)
assert dependencies.startup_recovery is not None
assert dependencies.startup_recovery.symbols == (SYMBOL,)
def test_rejects_symbols_different_from_recovery_coordinator() -> None:
with pytest.raises(
ValueError,
@@ -804,6 +949,577 @@ def test_rejects_scheduler_with_different_transport() -> None:
)
def test_rejects_startup_recovery_with_different_gate() -> None:
with pytest.raises(
ValueError,
match="startup recovery coordinator must share one",
):
RuntimeDependencies(
startup_recovery_enabled=True,
startup_recovery_uses_different_gate=True,
)
def test_rejects_startup_recovery_with_different_symbols() -> None:
with pytest.raises(
ValueError,
match="startup recovery coordinator must use the same symbols",
):
RuntimeDependencies(
startup_recovery_enabled=True,
startup_recovery_symbols=(ETH_SYMBOL,),
)
@pytest.mark.parametrize(
("timeout", "error_type"),
[
(True, TypeError),
("10", TypeError),
(0.0, ValueError),
(-1.0, ValueError),
(float("inf"), ValueError),
(float("nan"), ValueError),
],
)
def test_rejects_invalid_subscription_ack_timeout(
timeout: object,
error_type: type[Exception],
) -> None:
with pytest.raises(error_type):
RuntimeDependencies(
subscription_ack_timeout_seconds=timeout, # type: ignore[arg-type]
)
@pytest.mark.parametrize(
("capacity", "error_type"),
[
(True, TypeError),
(1.0, TypeError),
(0, ValueError),
(-1, ValueError),
],
)
def test_rejects_invalid_startup_market_buffer_capacity(
capacity: object,
error_type: type[Exception],
) -> None:
with pytest.raises(error_type):
RuntimeDependencies(
startup_market_buffer_capacity=capacity, # type: ignore[arg-type]
)
def test_startup_recovery_preserves_boundary_order_and_one_receiver() -> None:
first_document = {
"destination": "internal.trade",
"payload": {
"symbol": SYMBOL,
"sequence": 1,
},
}
second_document = {
"destination": "internal.trade",
"payload": {
"symbol": SYMBOL,
"sequence": 2,
},
}
post_ack_document = {
"destination": "internal.trade",
"payload": {
"symbol": SYMBOL,
"sequence": 3,
},
}
async def scenario() -> RuntimeDependencies:
recovery_release = asyncio.Event()
dependencies = RuntimeDependencies(
startup_recovery_enabled=True,
startup_recovery_release=recovery_release,
)
runtime_task = asyncio.create_task(
dependencies.runtime.run(),
)
await dependencies.service.subscribe_entered.wait()
correlation_id = (
dependencies.service.subscribe_correlation_ids[0]
)
assert isinstance(correlation_id, str)
dependencies.transport.feed(json.dumps(first_document))
dependencies.transport.feed(json.dumps(second_document))
dependencies.transport.feed(
make_control_message(
correlation_id=correlation_id,
)
)
startup_recovery = dependencies.startup_recovery
assert startup_recovery is not None
await startup_recovery.recovery_entered.wait()
assert dependencies.live_processing_gate.locked is True
assert dependencies.service.documents == []
assert dependencies.transport.receive_calls == 3
assert dependencies.supervisor.start_calls == 0
assert dependencies.scheduler.start_calls == 0
dependencies.transport.feed(
json.dumps(post_ack_document),
)
await asyncio.sleep(0)
assert dependencies.transport.receive_calls == 3
recovery_release.set()
await dependencies.scheduler.started.wait()
await wait_until(
lambda: dependencies.service.documents
== [
first_document,
second_document,
post_ack_document,
],
)
await wait_until(
lambda: dependencies.transport.receive_calls == 5,
)
assert dependencies.service.documents == [
first_document,
second_document,
post_ack_document,
]
assert dependencies.transport.max_active_receivers == 1
await dependencies.runtime.stop()
await runtime_task
return dependencies
dependencies = asyncio.run(scenario())
assert dependencies.calls.index(
"startup_recovery.hydrate"
) < dependencies.calls.index("session.start")
assert dependencies.calls.index(
"service.subscribe"
) < dependencies.calls.index("startup_recovery.recover")
assert dependencies.calls.index(
"startup_recovery.recover"
) < dependencies.calls.index("service.handle_message")
assert dependencies.calls.index(
"service.handle_message"
) < dependencies.calls.index("supervisor.start")
def test_startup_ack_timeout_is_terminal() -> None:
async def scenario() -> RuntimeDependencies:
dependencies = RuntimeDependencies(
startup_recovery_enabled=True,
subscription_ack_timeout_seconds=0.01,
)
with pytest.raises(
WebSocketSubscriptionAckTimeoutError,
match="Истекло время",
):
await dependencies.runtime.run()
return dependencies
dependencies = asyncio.run(scenario())
startup_recovery = dependencies.startup_recovery
assert startup_recovery is not None
assert startup_recovery.recover_calls == 0
assert dependencies.live_processing_gate.failed is True
assert dependencies.supervisor.start_calls == 0
assert dependencies.scheduler.start_calls == 0
assert dependencies.transport.max_active_receivers == 1
assert dependencies.runtime.state is (
TradeStreamProductionRuntimeState.FAILED
)
def test_startup_hydration_failure_prevents_network_io() -> None:
hydration_error = RuntimeError("startup hydration failed")
async def scenario() -> RuntimeDependencies:
dependencies = RuntimeDependencies(
startup_recovery_enabled=True,
startup_hydration_error=hydration_error,
)
with pytest.raises(
RuntimeError,
match="startup hydration failed",
) as error_info:
await dependencies.runtime.run()
assert error_info.value is hydration_error
return dependencies
dependencies = asyncio.run(scenario())
startup_recovery = dependencies.startup_recovery
assert startup_recovery is not None
assert startup_recovery.hydrate_calls == 1
assert startup_recovery.recover_calls == 0
assert dependencies.session.start_calls == 0
assert dependencies.service.subscribe_calls == []
assert dependencies.live_processing_gate.failed is True
assert dependencies.runtime.state is (
TradeStreamProductionRuntimeState.FAILED
)
def test_persistent_startup_connect_failure_marks_gate_failed() -> None:
connect_error = RuntimeError("persistent connect failed")
async def scenario() -> RuntimeDependencies:
dependencies = RuntimeDependencies(
startup_recovery_enabled=True,
start_error=connect_error,
)
with pytest.raises(
RuntimeError,
match="persistent connect failed",
) as error_info:
await dependencies.runtime.run()
assert error_info.value is connect_error
return dependencies
dependencies = asyncio.run(scenario())
startup_recovery = dependencies.startup_recovery
assert startup_recovery is not None
assert startup_recovery.hydrate_calls == 1
assert startup_recovery.recover_calls == 0
assert dependencies.service.subscribe_calls == []
assert dependencies.live_processing_gate.failed is True
assert dependencies.runtime.state is (
TradeStreamProductionRuntimeState.FAILED
)
def test_negative_startup_ack_is_terminal_before_recovery() -> None:
async def scenario() -> RuntimeDependencies:
dependencies = RuntimeDependencies(
startup_recovery_enabled=True,
)
runtime_task = asyncio.create_task(
dependencies.runtime.run(),
)
await dependencies.service.subscribe_entered.wait()
correlation_id = (
dependencies.service.subscribe_correlation_ids[0]
)
assert isinstance(correlation_id, str)
dependencies.transport.feed(
make_control_message(
correlation_id=correlation_id,
status="ERROR",
)
)
with pytest.raises(
WebSocketControlMessageError,
match="отклонил",
):
await runtime_task
return dependencies
dependencies = asyncio.run(scenario())
startup_recovery = dependencies.startup_recovery
assert startup_recovery is not None
assert startup_recovery.recover_calls == 0
assert dependencies.live_processing_gate.failed is True
assert dependencies.supervisor.start_calls == 0
@pytest.mark.parametrize(
("control_document_factory", "error_match"),
[
(
lambda correlation_id: {
"correlationId": "unknown-request",
"destination": "trades.subscribe",
"status": "OK",
},
"неизвестным correlationId",
),
(
lambda correlation_id: {
"correlationId": correlation_id,
"destination": "trades.subscribe",
},
"непустой строковый status",
),
(
lambda correlation_id: {
"correlationId": correlation_id,
"destination": "unknown.destination",
"status": "OK",
},
"неизвестным destination",
),
],
)
def test_invalid_startup_control_message_is_terminal(
control_document_factory: Callable[[str], object],
error_match: str,
) -> None:
async def scenario() -> RuntimeDependencies:
dependencies = RuntimeDependencies(
startup_recovery_enabled=True,
)
runtime_task = asyncio.create_task(
dependencies.runtime.run(),
)
await dependencies.service.subscribe_entered.wait()
correlation_id = (
dependencies.service.subscribe_correlation_ids[0]
)
assert isinstance(correlation_id, str)
dependencies.transport.feed(
json.dumps(
control_document_factory(correlation_id),
)
)
with pytest.raises(
WebSocketMessageRoutingError,
match=error_match,
):
await runtime_task
return dependencies
dependencies = asyncio.run(scenario())
startup_recovery = dependencies.startup_recovery
assert startup_recovery is not None
assert startup_recovery.recover_calls == 0
assert dependencies.service.documents == []
assert dependencies.live_processing_gate.failed is True
assert dependencies.reconnect_recovery.calls == []
def test_startup_transport_error_is_terminal_without_reconnect() -> None:
transport_error = WebSocketTransportError(
"startup transport failed",
)
async def scenario() -> RuntimeDependencies:
dependencies = RuntimeDependencies(
startup_recovery_enabled=True,
incoming=(transport_error,),
)
with pytest.raises(
WebSocketTransportError,
match="startup transport failed",
) as error_info:
await dependencies.runtime.run()
assert error_info.value is transport_error
return dependencies
dependencies = asyncio.run(scenario())
startup_recovery = dependencies.startup_recovery
assert startup_recovery is not None
assert startup_recovery.recover_calls == 0
assert dependencies.reconnect_recovery.calls == []
assert dependencies.live_processing_gate.failed is True
assert dependencies.supervisor.start_calls == 0
def test_startup_market_buffer_overflow_is_terminal() -> None:
async def scenario() -> RuntimeDependencies:
dependencies = RuntimeDependencies(
startup_recovery_enabled=True,
startup_market_buffer_capacity=1,
)
runtime_task = asyncio.create_task(
dependencies.runtime.run(),
)
await dependencies.service.subscribe_entered.wait()
dependencies.transport.feed(MARKET_MESSAGE)
dependencies.transport.feed(MARKET_MESSAGE)
with pytest.raises(
WebSocketStartupMarketBufferOverflowError,
match="Переполнен буфер",
):
await runtime_task
return dependencies
dependencies = asyncio.run(scenario())
startup_recovery = dependencies.startup_recovery
assert startup_recovery is not None
assert startup_recovery.recover_calls == 0
assert dependencies.service.documents == []
assert dependencies.live_processing_gate.failed is True
def test_startup_recovery_failure_prevents_fifo_drain() -> None:
recovery_error = RuntimeError("startup recovery failed")
async def scenario() -> RuntimeDependencies:
dependencies = RuntimeDependencies(
startup_recovery_enabled=True,
startup_recovery_error=recovery_error,
)
runtime_task = asyncio.create_task(
dependencies.runtime.run(),
)
await dependencies.service.subscribe_entered.wait()
correlation_id = (
dependencies.service.subscribe_correlation_ids[0]
)
assert isinstance(correlation_id, str)
dependencies.transport.feed(MARKET_MESSAGE)
dependencies.transport.feed(
make_control_message(
correlation_id=correlation_id,
)
)
with pytest.raises(
RuntimeError,
match="startup recovery failed",
) as error_info:
await runtime_task
assert error_info.value is recovery_error
return dependencies
dependencies = asyncio.run(scenario())
assert dependencies.service.documents == []
assert dependencies.live_processing_gate.failed is True
assert dependencies.supervisor.start_calls == 0
assert dependencies.scheduler.start_calls == 0
def test_stop_during_startup_ack_wait_releases_boundary() -> None:
async def scenario() -> RuntimeDependencies:
dependencies = RuntimeDependencies(
startup_recovery_enabled=True,
)
runtime_task = asyncio.create_task(
dependencies.runtime.run(),
)
await dependencies.service.subscribe_entered.wait()
await dependencies.transport.waiting.wait()
await asyncio.wait_for(
dependencies.runtime.stop(),
timeout=1.0,
)
await runtime_task
return dependencies
dependencies = asyncio.run(scenario())
assert dependencies.live_processing_gate.locked is False
assert dependencies.live_processing_gate.failed is False
assert dependencies.supervisor.start_calls == 0
assert dependencies.transport.active_receivers == 0
assert dependencies.runtime.state is (
TradeStreamProductionRuntimeState.STOPPED
)
def test_stop_during_startup_hydration_prevents_connection() -> None:
async def scenario() -> RuntimeDependencies:
hydration_release = asyncio.Event()
dependencies = RuntimeDependencies(
startup_recovery_enabled=True,
startup_hydration_release=hydration_release,
)
runtime_task = asyncio.create_task(
dependencies.runtime.run(),
)
startup_recovery = dependencies.startup_recovery
assert startup_recovery is not None
await startup_recovery.hydration_entered.wait()
await asyncio.wait_for(
dependencies.runtime.stop(),
timeout=1.0,
)
await runtime_task
return dependencies
dependencies = asyncio.run(scenario())
assert dependencies.session.start_calls == 0
assert dependencies.live_processing_gate.failed is False
assert dependencies.runtime.state is (
TradeStreamProductionRuntimeState.STOPPED
)
def test_stop_during_startup_recovery_prevents_fifo_drain() -> None:
async def scenario() -> RuntimeDependencies:
recovery_release = asyncio.Event()
dependencies = RuntimeDependencies(
startup_recovery_enabled=True,
startup_recovery_release=recovery_release,
)
runtime_task = asyncio.create_task(
dependencies.runtime.run(),
)
await dependencies.service.subscribe_entered.wait()
correlation_id = (
dependencies.service.subscribe_correlation_ids[0]
)
assert isinstance(correlation_id, str)
dependencies.transport.feed(MARKET_MESSAGE)
dependencies.transport.feed(
make_control_message(
correlation_id=correlation_id,
)
)
startup_recovery = dependencies.startup_recovery
assert startup_recovery is not None
await startup_recovery.recovery_entered.wait()
await asyncio.wait_for(
dependencies.runtime.stop(),
timeout=1.0,
)
await runtime_task
return dependencies
dependencies = asyncio.run(scenario())
assert dependencies.service.documents == []
assert dependencies.live_processing_gate.locked is False
assert dependencies.live_processing_gate.failed is False
assert dependencies.supervisor.start_calls == 0
assert dependencies.runtime.state is (
TradeStreamProductionRuntimeState.STOPPED
)
def test_locked_gate_does_not_leave_runtime_partially_started() -> None:
async def scenario() -> RuntimeDependencies:
dependencies = RuntimeDependencies()
@@ -893,7 +1609,12 @@ def test_scheduler_claim_blocks_external_start_during_runtime_startup() -> None:
def test_startup_receive_and_shutdown_order() -> None:
async def scenario() -> RuntimeDependencies:
dependencies = RuntimeDependencies()
dependencies = RuntimeDependencies(
symbols=(
f" {SYMBOL.lower()} ",
SYMBOL,
),
)
runtime_task = asyncio.create_task(
dependencies.runtime.run(),
)

View File

@@ -1,6 +1,7 @@
from __future__ import annotations
import asyncio
from collections.abc import Awaitable
from typing import Any
import pytest
@@ -73,6 +74,12 @@ class FakeConnection:
async def recv(self) -> str | bytes:
return ""
async def ping(self) -> Awaitable[float]:
async def wait_for_pong() -> float:
return 0.001
return wait_for_pong()
class RecordingConnector:
def __init__(

View File

@@ -13,9 +13,15 @@ import pytest
from src.market_data.acquisition.adapters.dzengi.rest import (
DzengiTradesDocumentSource,
)
from src.market_data.acquisition.checkpoint.trade_stream_state_hydrator import (
TradeStreamStateHydratorProtocol,
)
from src.market_data.acquisition.consistency.trade_observation_sink_protocol import (
TradeObservationSinkProtocol,
)
from src.market_data.acquisition.consistency.trade_stream_state import (
TradeStreamState,
)
from src.market_data.acquisition.models.trade import (
Trade,
TradeAggressorSide,
@@ -44,6 +50,12 @@ from src.market_data.acquisition.runtime.runtime_events import (
from src.market_data.acquisition.runtime.runtime_recovery_protocol import (
RuntimeRecoveryProtocol,
)
from src.market_data.acquisition.recovery.trade_recovery_result import (
TradeRecoveryResult,
)
from src.market_data.acquisition.runtime.runtime_startup_recovery_coordinator import (
RuntimeStartupRecoveryProtocol,
)
from src.market_data.acquisition.runtime.scheduler import (
RuntimeSchedulerProtocol,
)
@@ -66,6 +78,10 @@ from src.market_data.acquisition.trade_stream_runtime_composition import (
TradeStreamRuntimeComposition,
build_trade_stream_runtime_composition,
)
from src.market_data.storage.contracts import (
PersistentTradeCheckpoint,
TradeCheckpointStorageProtocol,
)
SYMBOL = "BTC/USD_LEVERAGE"
@@ -326,17 +342,152 @@ class RecordingTradeObservationSink:
) -> None:
self._fail_on_trade_id = fail_on_trade_id
self.observations: list[Trade] = []
self.accepted: list[tuple[Trade, Trade | None]] = []
self.duplicates: list[Trade] = []
def persist(
def persist_accepted(
self,
trade: Trade,
*,
expected_trade: Trade | None,
) -> None:
self.observations.append(trade)
self.accepted.append((trade, expected_trade))
if trade.trade_id == self._fail_on_trade_id:
raise RuntimeError("storage failed")
def persist_duplicate(
self,
trade: Trade,
) -> None:
self.observations.append(trade)
self.duplicates.append(trade)
if trade.trade_id == self._fail_on_trade_id:
raise RuntimeError("storage failed")
class RecordingCheckpointStorage:
"""Checkpoint storage, фиксирующий нежелательный I/O."""
def __init__(self) -> None:
self.calls: list[str] = []
def load_checkpoint(
self,
*,
venue: str,
symbol: str,
) -> PersistentTradeCheckpoint | None:
self.calls.append("load_checkpoint")
raise AssertionError("Composition не должна читать checkpoint.")
def load_checkpoint_tail(
self,
*,
venue: str,
checkpoint: PersistentTradeCheckpoint,
limit: int,
) -> tuple[Trade, ...]:
self.calls.append("load_checkpoint_tail")
raise AssertionError("Composition не должна читать Trade tail.")
def load_latest_trade_tail(
self,
*,
venue: str,
symbol: str,
limit: int,
) -> tuple[Trade, ...]:
self.calls.append("load_latest_trade_tail")
raise AssertionError("Composition не должна читать Trade tail.")
def adopt_existing_trade_as_checkpoint(
self,
*,
venue: str,
trade: Trade,
) -> PersistentTradeCheckpoint:
self.calls.append("adopt_existing_trade_as_checkpoint")
raise AssertionError("Composition не должна создавать checkpoint.")
def store_trade_and_advance_checkpoint(
self,
*,
venue: str,
expected_trade: Trade | None,
trade: Trade,
observed_at: datetime,
) -> PersistentTradeCheckpoint:
self.calls.append("store_trade_and_advance_checkpoint")
raise AssertionError("Composition не должна записывать checkpoint.")
class CheckpointBackedStorage:
"""Checkpoint storage для проверки общего канонического ключа."""
def __init__(
self,
checkpoint: PersistentTradeCheckpoint,
) -> None:
self._checkpoint = checkpoint
self.calls: list[tuple[object, ...]] = []
def load_checkpoint(
self,
*,
venue: str,
symbol: str,
) -> PersistentTradeCheckpoint | None:
self.calls.append(("load_checkpoint", venue, symbol))
return self._checkpoint
def load_checkpoint_tail(
self,
*,
venue: str,
checkpoint: PersistentTradeCheckpoint,
limit: int,
) -> tuple[Trade, ...]:
self.calls.append(
(
"load_checkpoint_tail",
venue,
checkpoint,
limit,
)
)
return (checkpoint.trade,)
def load_latest_trade_tail(
self,
*,
venue: str,
symbol: str,
limit: int,
) -> tuple[Trade, ...]:
raise AssertionError("Latest Trade tail не должен запрашиваться.")
def adopt_existing_trade_as_checkpoint(
self,
*,
venue: str,
trade: Trade,
) -> PersistentTradeCheckpoint:
raise AssertionError("Checkpoint не должен создаваться.")
def store_trade_and_advance_checkpoint(
self,
*,
venue: str,
expected_trade: Trade | None,
trade: Trade,
observed_at: datetime,
) -> PersistentTradeCheckpoint:
raise AssertionError("Checkpoint не должен записываться.")
@dataclass(slots=True)
class CompositionDependencies:
session: FakeSession
@@ -352,6 +503,7 @@ class CompositionDependencies:
def create_composition(
*,
symbols: tuple[str, ...] = (SYMBOL,),
trade: Trade | None = None,
recovery_document: object = (),
heartbeat_timeout_seconds: float = 10.0,
@@ -359,6 +511,8 @@ def create_composition(
max_recovery_window_ms: int = 3_599_999,
probe_results: tuple[bool, ...] = (True,),
trade_observation_sink: TradeObservationSinkProtocol | None = None,
checkpoint_storage: TradeCheckpointStorageProtocol | None = None,
checkpoint_venue: str | None = None,
) -> tuple[
TradeStreamRuntimeComposition,
CompositionDependencies,
@@ -390,7 +544,7 @@ def create_composition(
recovery_document_source=(
dependencies.recovery_document_source
),
symbols=(SYMBOL,),
symbols=symbols,
heartbeat_timeout_seconds=heartbeat_timeout_seconds,
scheduler_interval_seconds=scheduler_interval_seconds,
trade_observation_sink=trade_observation_sink,
@@ -400,6 +554,8 @@ def create_composition(
dependencies.recovery_end_time_clock
),
scheduler_sleep=dependencies.scheduler_sleep,
checkpoint_storage=checkpoint_storage,
checkpoint_venue=checkpoint_venue,
)
return (
@@ -467,6 +623,196 @@ def test_components_implement_public_protocols() -> None:
)
def test_checkpoint_components_are_absent_without_configuration() -> None:
composition, *_ = create_composition()
assert composition.state_hydrator is None
assert composition.runtime_startup_recovery_coordinator is None
def test_checkpoint_components_implement_public_protocols() -> None:
storage = RecordingCheckpointStorage()
composition, *_ = create_composition(
checkpoint_storage=storage,
checkpoint_venue="Dzengi",
)
assert isinstance(storage, TradeCheckpointStorageProtocol)
assert isinstance(
composition.state_hydrator,
TradeStreamStateHydratorProtocol,
)
assert isinstance(
composition.runtime_startup_recovery_coordinator,
RuntimeStartupRecoveryProtocol,
)
def test_checkpoint_runtime_reuses_shared_dependency_graph() -> None:
storage = RecordingCheckpointStorage()
composition, dependencies = create_composition(
checkpoint_storage=storage,
checkpoint_venue="Dzengi",
)
state_hydrator = composition.state_hydrator
startup_recovery = (
composition.runtime_startup_recovery_coordinator
)
assert state_hydrator is not None
assert startup_recovery is not None
assert state_hydrator._state_store is composition.state_store
assert startup_recovery._state_hydrator is state_hydrator
assert (
startup_recovery._recovery_coordinator
is composition.runtime_recovery_coordinator
)
assert (
startup_recovery.live_processing_gate
is composition.live_processing_gate
)
assert (
startup_recovery.symbols
== composition.runtime_reconnect_recovery_coordinator.symbols
== (SYMBOL,)
)
assert (
startup_recovery._clock
is dependencies.recovery_end_time_clock
)
assert (
composition.runtime_reconnect_recovery_coordinator._clock
is dependencies.recovery_end_time_clock
)
def test_lowercase_symbol_uses_one_key_for_hydration_and_recovery() -> None:
checkpoint = PersistentTradeCheckpoint(
venue="dzengi",
trade=make_trade(),
revision=1,
updated_at=CHECKPOINT_TIME,
)
storage = CheckpointBackedStorage(checkpoint)
composition, dependencies = create_composition(
symbols=(f" {SYMBOL.lower()} ",),
recovery_document=[],
checkpoint_storage=storage,
checkpoint_venue="Dzengi",
)
startup_recovery = (
composition.runtime_startup_recovery_coordinator
)
assert startup_recovery is not None
async def scenario() -> tuple[
tuple[TradeStreamState, ...],
tuple[TradeRecoveryResult, ...],
]:
states = await startup_recovery.hydrate_once()
async with composition.live_processing_gate:
results = await startup_recovery.recover_after_ack()
return states, results
states, results = asyncio.run(scenario())
assert startup_recovery.symbols == (SYMBOL,)
assert (
composition.runtime_reconnect_recovery_coordinator.symbols
== (SYMBOL,)
)
assert states[0].symbol == SYMBOL
assert composition.state_store.get(SYMBOL) is states[0]
assert composition.state_store.contains(SYMBOL.lower()) is False
assert results[0].symbol == SYMBOL
assert storage.calls[0] == (
"load_checkpoint",
"dzengi",
SYMBOL,
)
assert dependencies.recovery_document_source.calls
assert dependencies.recovery_document_source.calls[0][0] == SYMBOL
def test_case_variant_duplicates_collapse_before_hydration_worker() -> None:
checkpoint = PersistentTradeCheckpoint(
venue="dzengi",
trade=make_trade(),
revision=1,
updated_at=CHECKPOINT_TIME,
)
storage = CheckpointBackedStorage(checkpoint)
composition, _ = create_composition(
symbols=(
SYMBOL.lower(),
SYMBOL,
f" {SYMBOL.lower()} ",
),
checkpoint_storage=storage,
checkpoint_venue="dzengi",
)
startup_recovery = (
composition.runtime_startup_recovery_coordinator
)
assert startup_recovery is not None
states = asyncio.run(startup_recovery.hydrate_once())
assert startup_recovery.symbols == (SYMBOL,)
assert len(states) == 1
assert states[0].symbol == SYMBOL
assert [call[0] for call in storage.calls] == [
"load_checkpoint",
"load_checkpoint_tail",
]
@pytest.mark.parametrize(
("checkpoint_storage", "checkpoint_venue"),
[
(RecordingCheckpointStorage(), None),
(None, "dzengi"),
],
)
def test_partial_checkpoint_configuration_is_rejected_without_io(
checkpoint_storage: TradeCheckpointStorageProtocol | None,
checkpoint_venue: str | None,
) -> None:
with pytest.raises(
ValueError,
match="должны быть переданы вместе",
):
create_composition(
checkpoint_storage=checkpoint_storage,
checkpoint_venue=checkpoint_venue,
)
if isinstance(checkpoint_storage, RecordingCheckpointStorage):
assert checkpoint_storage.calls == []
def test_checkpoint_composition_has_no_io_or_background_tasks() -> None:
storage = RecordingCheckpointStorage()
composition, dependencies = create_composition(
checkpoint_storage=storage,
checkpoint_venue="dzengi",
)
startup_recovery = (
composition.runtime_startup_recovery_coordinator
)
assert startup_recovery is not None
assert storage.calls == []
assert dependencies.recovery_document_source.calls == []
assert dependencies.recovery_end_time_clock.calls == 0
assert startup_recovery._hydration_task is None
assert startup_recovery._recovery_task is None
def test_external_dependencies_are_reused() -> None:
composition, dependencies = create_composition()
@@ -548,6 +894,11 @@ def test_live_and_recovery_share_optional_persistence_sink() -> None:
trade.trade_id
for trade in sink.observations
] == [100, recovered_trade_id]
assert sink.accepted == [
(live_trade, None),
(recovery_result.last_trade, live_trade),
]
assert sink.duplicates == []
def test_recovery_duplicate_updates_persistence_without_checkpoint_change(
@@ -589,6 +940,8 @@ def test_recovery_duplicate_updates_persistence_without_checkpoint_change(
assert len(sink.observations) == 2
assert sink.observations[0] is live_trade
assert sink.observations[1].source == "dzengi"
assert sink.accepted == [(live_trade, None)]
assert sink.duplicates == [sink.observations[1]]
assert state.last_trade is live_trade

View File

@@ -0,0 +1,260 @@
from __future__ import annotations
from dataclasses import replace
from datetime import datetime, timezone
from decimal import Decimal
from typing import Any
import pytest
from src.market_data.acquisition.models.trade import (
Trade,
TradeAggressorSide,
)
from src.market_data.storage import (
MarketDataCheckpointConflictError,
MarketDataCheckpointIntegrityError,
MarketDataStorageConflictError,
MarketDataStorageError,
PersistentTradeCheckpoint,
TradeCheckpointStorageProtocol,
)
VENUE = "DZENGI"
SYMBOL = "BTC/USD_LEVERAGE"
EXECUTED_AT = datetime(2026, 8, 1, 10, 0, tzinfo=timezone.utc)
UPDATED_AT = datetime(2026, 8, 1, 10, 1, tzinfo=timezone.utc)
def make_trade(
*,
symbol: str = SYMBOL,
trade_id: int = 123,
executed_at: datetime = EXECUTED_AT,
) -> Trade:
return Trade(
symbol=symbol,
trade_id=trade_id,
price=Decimal("65000.25"),
quantity=Decimal("0.001"),
executed_at=executed_at,
aggressor_side=TradeAggressorSide.BUY,
source="dzengi_websocket_trade",
)
def make_checkpoint() -> PersistentTradeCheckpoint:
return PersistentTradeCheckpoint(
venue=VENUE,
trade=make_trade(),
revision=7,
updated_at=UPDATED_AT,
)
class RecordingCheckpointStorage:
def __init__(self) -> None:
self.checkpoint = make_checkpoint()
def load_checkpoint(
self,
*,
venue: str,
symbol: str,
) -> PersistentTradeCheckpoint | None:
return self.checkpoint
def load_checkpoint_tail(
self,
*,
venue: str,
checkpoint: PersistentTradeCheckpoint,
limit: int,
) -> tuple[Trade, ...]:
return (checkpoint.trade,)
def load_latest_trade_tail(
self,
*,
venue: str,
symbol: str,
limit: int,
) -> tuple[Trade, ...]:
return (self.checkpoint.trade,)
def adopt_existing_trade_as_checkpoint(
self,
*,
venue: str,
trade: Trade,
) -> PersistentTradeCheckpoint:
return replace(
self.checkpoint,
venue=venue,
trade=trade,
revision=1,
)
def store_trade_and_advance_checkpoint(
self,
*,
venue: str,
expected_trade: Trade | None,
trade: Trade,
observed_at: datetime,
) -> PersistentTradeCheckpoint:
return replace(
self.checkpoint,
trade=trade,
revision=self.checkpoint.revision + 1,
updated_at=observed_at,
)
def test_checkpoint_preserves_full_trade_and_durable_identity() -> None:
checkpoint = make_checkpoint()
assert checkpoint.venue == VENUE
assert checkpoint.trade is not None
assert checkpoint.revision == 7
assert checkpoint.updated_at == UPDATED_AT
assert checkpoint.checkpoint_schema_version == 1
assert checkpoint.identity == (
VENUE,
SYMBOL,
123,
EXECUTED_AT,
)
def test_checkpoint_storage_protocol_is_runtime_checkable() -> None:
assert isinstance(
RecordingCheckpointStorage(),
TradeCheckpointStorageProtocol,
)
@pytest.mark.parametrize("venue", ("", " ", "\t"))
def test_checkpoint_rejects_empty_venue(venue: str) -> None:
with pytest.raises(ValueError, match="venue"):
PersistentTradeCheckpoint(
venue=venue,
trade=make_trade(),
revision=1,
updated_at=UPDATED_AT,
)
def test_checkpoint_rejects_non_string_venue() -> None:
with pytest.raises(TypeError, match="venue"):
PersistentTradeCheckpoint(
venue=123, # type: ignore[arg-type]
trade=make_trade(),
revision=1,
updated_at=UPDATED_AT,
)
def test_checkpoint_rejects_non_trade_payload() -> None:
with pytest.raises(TypeError, match="Canonical Trade"):
PersistentTradeCheckpoint(
venue=VENUE,
trade=object(), # type: ignore[arg-type]
revision=1,
updated_at=UPDATED_AT,
)
def test_checkpoint_rejects_empty_trade_symbol() -> None:
with pytest.raises(ValueError, match="trade.symbol"):
PersistentTradeCheckpoint(
venue=VENUE,
trade=make_trade(symbol=" "),
revision=1,
updated_at=UPDATED_AT,
)
def test_checkpoint_rejects_naive_trade_time() -> None:
with pytest.raises(ValueError, match="trade.executed_at"):
PersistentTradeCheckpoint(
venue=VENUE,
trade=make_trade(
executed_at=datetime(2026, 8, 1, 10, 0),
),
revision=1,
updated_at=UPDATED_AT,
)
@pytest.mark.parametrize("trade_id", (-2_147_483_649, 2_147_483_648))
def test_checkpoint_rejects_trade_id_outside_signed_range(
trade_id: int,
) -> None:
with pytest.raises(ValueError, match="signed 32-bit"):
PersistentTradeCheckpoint(
venue=VENUE,
trade=make_trade(trade_id=trade_id),
revision=1,
updated_at=UPDATED_AT,
)
@pytest.mark.parametrize("revision", (0, -1))
def test_checkpoint_rejects_non_positive_revision(revision: int) -> None:
with pytest.raises(ValueError, match="revision"):
PersistentTradeCheckpoint(
venue=VENUE,
trade=make_trade(),
revision=revision,
updated_at=UPDATED_AT,
)
@pytest.mark.parametrize("revision", (True, 1.5, "1", None))
def test_checkpoint_rejects_non_integer_revision(
revision: Any,
) -> None:
with pytest.raises(TypeError, match="revision"):
PersistentTradeCheckpoint(
venue=VENUE,
trade=make_trade(),
revision=revision,
updated_at=UPDATED_AT,
)
def test_checkpoint_rejects_naive_updated_at() -> None:
with pytest.raises(ValueError, match="updated_at"):
PersistentTradeCheckpoint(
venue=VENUE,
trade=make_trade(),
revision=1,
updated_at=datetime(2026, 8, 1, 10, 1),
)
@pytest.mark.parametrize("version", (0, -1))
def test_checkpoint_rejects_non_positive_schema_version(
version: int,
) -> None:
with pytest.raises(ValueError, match="checkpoint_schema_version"):
PersistentTradeCheckpoint(
venue=VENUE,
trade=make_trade(),
revision=1,
updated_at=UPDATED_AT,
checkpoint_schema_version=version,
)
def test_checkpoint_errors_preserve_storage_hierarchy() -> None:
assert issubclass(
MarketDataCheckpointConflictError,
MarketDataStorageConflictError,
)
assert issubclass(
MarketDataCheckpointIntegrityError,
MarketDataStorageError,
)

View File

@@ -129,6 +129,9 @@ class PartitionCursor:
if normalized.startswith("ALTER TABLE") and "ADD CONSTRAINT" in normalized:
return
if normalized.startswith("ALTER TABLE") and "DROP CONSTRAINT" in normalized:
return
if normalized.startswith("WITH moved_rows AS"):
self.rowcount = self._connection.moved_row_count
return
@@ -481,6 +484,70 @@ def test_manager_creates_partition_moves_default_rows_and_registers_it() -> None
assert all("::timestamptz" in statement for statement, _ in ddl_calls)
def test_trade_partition_rebuilds_checkpoint_foreign_key_around_move() -> None:
_, connection, provider = _dependencies()
manager = PostgresMarketDataPartitionManager(
connection_provider=provider
)
manager.ensure_month_partition(
data_type=MarketDataPartitionType.TRADES,
month=JULY,
)
statements = tuple(statement for statement, _ in connection.calls)
parent_lock_index = statements.index(
'LOCK TABLE "market_data"."trades" '
"IN SHARE ROW EXCLUSIVE MODE"
)
default_lock_index = statements.index(
'LOCK TABLE "market_data"."trades_default" '
"IN ACCESS EXCLUSIVE MODE"
)
drop_index = statements.index(
'ALTER TABLE "market_data"."trade_stream_checkpoints" '
'DROP CONSTRAINT "trade_stream_checkpoints_trade_fk"'
)
move_index = next(
index
for index, statement in enumerate(statements)
if statement.startswith("WITH moved_rows AS")
)
restore_index = next(
index
for index, statement in enumerate(statements)
if "ADD CONSTRAINT \"trade_stream_checkpoints_trade_fk\"" in statement
)
assert (
parent_lock_index
< default_lock_index
< drop_index
< move_index
< restore_index
)
restored_sql = statements[restore_index]
assert "ON UPDATE NO ACTION ON DELETE NO ACTION" in restored_sql
assert "DEFERRABLE INITIALLY DEFERRED" in restored_sql
def test_non_trade_partition_does_not_touch_checkpoint_foreign_key() -> None:
_, connection, provider = _dependencies()
manager = PostgresMarketDataPartitionManager(
connection_provider=provider
)
manager.ensure_month_partition(
data_type=MarketDataPartitionType.QUOTES,
month=JULY,
)
assert not any(
"trade_stream_checkpoints_trade_fk" in statement
for statement, _ in connection.calls
)
def test_manager_is_idempotent_after_registered_partition_exists() -> None:
_, connection, provider = _dependencies()
manager = PostgresMarketDataPartitionManager(
@@ -718,6 +785,22 @@ def test_retention_drops_complete_month_and_deletes_partial_history() -> None:
for statement, _ in connection.calls
)
statements = tuple(statement for statement, _ in connection.calls)
drop_index = statements.index(
'ALTER TABLE "market_data"."trade_stream_checkpoints" '
'DROP CONSTRAINT "trade_stream_checkpoints_trade_fk"'
)
delete_index = statements.index(
'DELETE FROM "market_data"."trades" '
'WHERE "executed_at" < %s'
)
restore_index = next(
index
for index, statement in enumerate(statements)
if "ADD CONSTRAINT \"trade_stream_checkpoints_trade_fk\"" in statement
)
assert drop_index < delete_index < restore_index
def test_unconfigured_data_type_is_unlimited_and_not_touched() -> None:
database, connection, provider = _dependencies()

View File

@@ -0,0 +1,817 @@
from __future__ import annotations
from dataclasses import dataclass
from datetime import datetime, timedelta, timezone
from decimal import Decimal
from typing import Any
import pytest
from src.market_data.acquisition.models.trade import (
Trade,
TradeAggressorSide,
)
from src.market_data.acquisition.trade_id_sequence import (
SIGNED_TRADE_ID_MAX,
SIGNED_TRADE_ID_MIN,
)
from src.market_data.storage import (
MarketDataCheckpointConflictError,
MarketDataCheckpointIntegrityError,
MarketDataStorageOperationError,
MarketDataStorageValidationError,
PersistentTradeCheckpoint,
PostgresTradeRepository,
TradeCheckpointStorageProtocol,
)
VENUE = "dzengi"
SYMBOL = "BTC/USD_LEVERAGE"
EXECUTED_AT = datetime(2026, 8, 1, 10, 0, tzinfo=timezone.utc)
OBSERVED_AT = EXECUTED_AT + timedelta(seconds=1)
def _trade(
*,
trade_id: int = 100,
executed_at: datetime = EXECUTED_AT,
price: Decimal = Decimal("65000.25"),
source: str = "dzengi_websocket_trade",
) -> Trade:
return Trade(
symbol=SYMBOL,
trade_id=trade_id,
price=price,
quantity=Decimal("0.001"),
executed_at=executed_at,
aggressor_side=TradeAggressorSide.BUY,
source=source,
)
def _trade_row(trade: Trade) -> tuple[object, ...]:
return (
trade.symbol,
trade.trade_id,
trade.executed_at,
trade.price,
trade.quantity,
trade.aggressor_side.value,
trade.source,
1,
)
def _checkpoint_row(
trade: Trade,
*,
revision: int,
updated_at: datetime = OBSERVED_AT,
) -> tuple[object, ...]:
return (
VENUE,
trade.symbol,
trade.trade_id,
trade.executed_at,
revision,
updated_at,
1,
VENUE,
trade.symbol,
trade.trade_id,
trade.executed_at,
trade.price,
trade.quantity,
trade.aggressor_side.value,
trade.source,
1,
)
def _existing_trade_row(trade: Trade) -> tuple[object, ...]:
return (
trade.price,
trade.quantity,
trade.aggressor_side.value,
OBSERVED_AT,
OBSERVED_AT,
[trade.source],
1,
)
@dataclass(frozen=True)
class SqlStep:
starts_with: str
fetchone: object = None
fetchall: tuple[tuple[object, ...], ...] = ()
error: BaseException | None = None
class ScriptedCursor:
def __init__(self, connection: ScriptedConnection) -> None:
self._connection = connection
self._fetchone: object = None
self._fetchall: tuple[tuple[object, ...], ...] = ()
def __enter__(self) -> ScriptedCursor:
return self
def __exit__(self, *args: object) -> None:
return None
def execute(
self,
statement: str,
parameters: tuple[Any, ...],
) -> None:
normalized = " ".join(statement.split())
self._connection.calls.append((normalized, parameters))
if not self._connection.steps:
raise AssertionError(f"Unexpected SQL: {normalized}")
step = self._connection.steps.pop(0)
if not normalized.startswith(step.starts_with):
raise AssertionError(
f"Expected SQL starting with {step.starts_with!r}, "
f"received {normalized!r}."
)
if step.error is not None:
raise step.error
self._fetchone = step.fetchone
self._fetchall = step.fetchall
def fetchone(self) -> object:
return self._fetchone
def fetchall(self) -> tuple[tuple[object, ...], ...]:
return self._fetchall
class ScriptedConnection:
def __init__(self, steps: tuple[SqlStep, ...]) -> None:
self.steps = list(steps)
self.calls: list[tuple[str, tuple[Any, ...]]] = []
self.exit_exception_types: list[type[BaseException] | None] = []
def __enter__(self) -> ScriptedConnection:
return self
def __exit__(
self,
exception_type: type[BaseException] | None,
exception: BaseException | None,
traceback: object,
) -> None:
self.exit_exception_types.append(exception_type)
return None
def cursor(self) -> ScriptedCursor:
return ScriptedCursor(self)
class RecordingProvider:
def __init__(self, connection: ScriptedConnection) -> None:
self.connection = connection
self.calls = 0
def __call__(self) -> ScriptedConnection:
self.calls += 1
return self.connection
def _repository(
*steps: SqlStep,
) -> tuple[PostgresTradeRepository, ScriptedConnection, RecordingProvider]:
connection = ScriptedConnection(steps)
provider = RecordingProvider(connection)
repository = PostgresTradeRepository(
connection_provider=provider,
)
return repository, connection, provider
def test_repository_matches_checkpoint_storage_protocol() -> None:
repository, _, _ = _repository()
assert isinstance(repository, TradeCheckpointStorageProtocol)
def test_load_checkpoint_returns_none_without_persistent_state() -> None:
repository, connection, provider = _repository(
SqlStep(starts_with="SELECT checkpoint.venue"),
)
result = repository.load_checkpoint(
venue=" dzengi ",
symbol=" btc/usd_leverage ",
)
assert result is None
assert provider.calls == 1
assert connection.calls[0][1] == (VENUE, SYMBOL)
assert connection.steps == []
def test_load_checkpoint_restores_exact_canonical_trade() -> None:
trade = _trade()
repository, connection, _ = _repository(
SqlStep(
starts_with="SELECT checkpoint.venue",
fetchone=_checkpoint_row(trade, revision=7),
),
)
result = repository.load_checkpoint(
venue=VENUE,
symbol=SYMBOL,
)
assert result == PersistentTradeCheckpoint(
venue=VENUE,
trade=trade,
revision=7,
updated_at=OBSERVED_AT,
)
assert "FOR UPDATE" not in connection.calls[0][0]
def test_load_checkpoint_rejects_missing_durable_trade() -> None:
trade = _trade()
orphan_row = list(_checkpoint_row(trade, revision=1))
orphan_row[7:] = [None] * 9
repository, connection, _ = _repository(
SqlStep(
starts_with="SELECT checkpoint.venue",
fetchone=tuple(orphan_row),
),
)
with pytest.raises(
MarketDataCheckpointIntegrityError,
match="exact Canonical Trade",
):
repository.load_checkpoint(
venue=VENUE,
symbol=SYMBOL,
)
assert connection.exit_exception_types == [
MarketDataCheckpointIntegrityError
]
def test_checkpoint_tail_is_returned_oldest_to_checkpoint_across_rollover(
) -> None:
previous = _trade(
trade_id=SIGNED_TRADE_ID_MAX,
executed_at=EXECUTED_AT,
)
current = _trade(
trade_id=SIGNED_TRADE_ID_MIN,
executed_at=EXECUTED_AT + timedelta(milliseconds=1),
)
checkpoint = PersistentTradeCheckpoint(
venue=VENUE,
trade=current,
revision=2,
updated_at=OBSERVED_AT,
)
repository, connection, _ = _repository(
SqlStep(
starts_with="WITH reference_trade AS",
fetchall=(
_trade_row(current),
_trade_row(previous),
),
),
)
result = repository.load_checkpoint_tail(
venue=VENUE,
checkpoint=checkpoint,
limit=2,
)
assert result == (previous, current)
parameters = connection.calls[0][1]
assert parameters[:4] == (
VENUE,
SYMBOL,
SIGNED_TRADE_ID_MIN,
current.executed_at,
)
assert parameters[-1] == 2
def test_checkpoint_tail_rejects_non_strict_trade_id_order() -> None:
previous = _trade(trade_id=100)
conflicting = _trade(
trade_id=100,
executed_at=EXECUTED_AT + timedelta(milliseconds=1),
)
checkpoint = PersistentTradeCheckpoint(
venue=VENUE,
trade=conflicting,
revision=2,
updated_at=OBSERVED_AT,
)
repository, connection, _ = _repository(
SqlStep(
starts_with="WITH reference_trade AS",
fetchall=(
_trade_row(conflicting),
_trade_row(previous),
),
),
)
with pytest.raises(
MarketDataCheckpointIntegrityError,
match="strictly rollover-ordered",
):
repository.load_checkpoint_tail(
venue=VENUE,
checkpoint=checkpoint,
limit=2,
)
assert connection.exit_exception_types == [None]
@pytest.mark.parametrize("limit", (0, -1, True, 1.5, "1"))
def test_tail_rejects_invalid_limit_without_io(limit: object) -> None:
repository, _, provider = _repository()
with pytest.raises(MarketDataStorageValidationError, match="limit"):
repository.load_latest_trade_tail(
venue=VENUE,
symbol=SYMBOL,
limit=limit, # type: ignore[arg-type]
)
assert provider.calls == 0
def test_latest_tail_returns_empty_without_trade_history() -> None:
repository, connection, _ = _repository(
SqlStep(starts_with="SELECT symbol, trade_id"),
)
result = repository.load_latest_trade_tail(
venue=VENUE,
symbol=SYMBOL,
limit=10,
)
assert result == ()
assert len(connection.calls) == 1
def test_adopts_existing_trade_without_mutating_trade_history() -> None:
trade = _trade()
repository, connection, provider = _repository(
SqlStep(
starts_with="SELECT symbol, trade_id",
fetchone=_trade_row(trade),
),
SqlStep(
starts_with="INSERT INTO market_data.trade_stream_checkpoints",
fetchone=(1,),
),
SqlStep(
starts_with="SELECT checkpoint.venue",
fetchone=_checkpoint_row(trade, revision=1),
),
)
result = repository.adopt_existing_trade_as_checkpoint(
venue=" dzengi ",
trade=trade,
)
assert result.trade == trade
assert result.revision == 1
assert provider.calls == 1
assert connection.calls[0][1] == (
VENUE,
SYMBOL,
trade.trade_id,
trade.executed_at,
)
assert "FOR SHARE" in connection.calls[0][0]
assert not any(
statement.startswith("INSERT INTO market_data.trades")
or statement.startswith("UPDATE market_data.trades")
for statement, _ in connection.calls
)
assert connection.exit_exception_types == [None]
assert connection.steps == []
def test_repeated_adoption_of_same_trade_is_idempotent() -> None:
trade = _trade()
repository, connection, _ = _repository(
SqlStep(
starts_with="SELECT symbol, trade_id",
fetchone=_trade_row(trade),
),
SqlStep(
starts_with="INSERT INTO market_data.trade_stream_checkpoints",
),
SqlStep(
starts_with="SELECT checkpoint.venue",
fetchone=_checkpoint_row(trade, revision=1),
),
)
result = repository.adopt_existing_trade_as_checkpoint(
venue=VENUE,
trade=trade,
)
assert result.trade == trade
assert result.revision == 1
assert not any(
statement.startswith(
"UPDATE market_data.trade_stream_checkpoints"
)
for statement, _ in connection.calls
)
assert connection.exit_exception_types == [None]
def test_adoption_rejects_missing_durable_trade() -> None:
repository, connection, _ = _repository(
SqlStep(starts_with="SELECT symbol, trade_id"),
)
with pytest.raises(
MarketDataCheckpointIntegrityError,
match="absent from durable",
):
repository.adopt_existing_trade_as_checkpoint(
venue=VENUE,
trade=_trade(),
)
assert len(connection.calls) == 1
assert connection.exit_exception_types == [
MarketDataCheckpointIntegrityError
]
def test_adoption_rejects_conflicting_durable_payload() -> None:
candidate = _trade()
durable = _trade(price=Decimal("65000.26"))
repository, connection, _ = _repository(
SqlStep(
starts_with="SELECT symbol, trade_id",
fetchone=_trade_row(durable),
),
)
with pytest.raises(
MarketDataCheckpointConflictError,
match="conflicts with durable",
):
repository.adopt_existing_trade_as_checkpoint(
venue=VENUE,
trade=candidate,
)
assert len(connection.calls) == 1
assert connection.exit_exception_types == [
MarketDataCheckpointConflictError
]
def test_adoption_rejects_checkpoint_of_another_trade() -> None:
candidate = _trade(trade_id=100)
existing = _trade(
trade_id=101,
executed_at=EXECUTED_AT + timedelta(milliseconds=1),
)
repository, connection, _ = _repository(
SqlStep(
starts_with="SELECT symbol, trade_id",
fetchone=_trade_row(candidate),
),
SqlStep(
starts_with="INSERT INTO market_data.trade_stream_checkpoints",
),
SqlStep(
starts_with="SELECT checkpoint.venue",
fetchone=_checkpoint_row(existing, revision=1),
),
)
with pytest.raises(
MarketDataCheckpointConflictError,
match="already points",
):
repository.adopt_existing_trade_as_checkpoint(
venue=VENUE,
trade=candidate,
)
assert connection.exit_exception_types == [
MarketDataCheckpointConflictError
]
def test_adoption_database_error_rolls_back_transaction() -> None:
trade = _trade()
repository, connection, _ = _repository(
SqlStep(
starts_with="SELECT symbol, trade_id",
fetchone=_trade_row(trade),
),
SqlStep(
starts_with="INSERT INTO market_data.trade_stream_checkpoints",
error=RuntimeError("database failed"),
),
)
with pytest.raises(
MarketDataStorageOperationError,
match="adopt existing Trade",
) as error_info:
repository.adopt_existing_trade_as_checkpoint(
venue=VENUE,
trade=trade,
)
assert isinstance(error_info.value.__cause__, RuntimeError)
assert connection.exit_exception_types == [RuntimeError]
def test_first_checkpoint_is_inserted_with_trade_in_one_transaction() -> None:
trade = _trade()
repository, connection, provider = _repository(
SqlStep(
starts_with="INSERT INTO market_data.trades",
fetchone=(1,),
),
SqlStep(
starts_with="INSERT INTO market_data.trade_stream_checkpoints",
fetchone=(1,),
),
SqlStep(
starts_with="SELECT checkpoint.venue",
fetchone=_checkpoint_row(trade, revision=1),
),
)
result = repository.store_trade_and_advance_checkpoint(
venue=VENUE,
expected_trade=None,
trade=trade,
observed_at=OBSERVED_AT,
)
assert result.trade == trade
assert result.revision == 1
assert provider.calls == 1
assert connection.exit_exception_types == [None]
assert "FOR UPDATE OF checkpoint" in connection.calls[-1][0]
assert connection.steps == []
def test_first_checkpoint_mismatch_rolls_back_transaction() -> None:
candidate = _trade(trade_id=100)
unexpected = _trade(
trade_id=101,
executed_at=EXECUTED_AT + timedelta(milliseconds=1),
)
repository, connection, _ = _repository(
SqlStep(
starts_with="INSERT INTO market_data.trades",
fetchone=(1,),
),
SqlStep(
starts_with="INSERT INTO market_data.trade_stream_checkpoints",
fetchone=(1,),
),
SqlStep(
starts_with="SELECT checkpoint.venue",
fetchone=_checkpoint_row(unexpected, revision=1),
),
)
with pytest.raises(
MarketDataCheckpointIntegrityError,
match="First checkpoint",
):
repository.store_trade_and_advance_checkpoint(
venue=VENUE,
expected_trade=None,
trade=candidate,
observed_at=OBSERVED_AT,
)
assert connection.exit_exception_types == [
MarketDataCheckpointIntegrityError
]
def test_existing_checkpoint_advances_with_revision_cas() -> None:
previous = _trade(trade_id=100)
current = _trade(
trade_id=101,
executed_at=EXECUTED_AT + timedelta(milliseconds=1),
)
repository, connection, _ = _repository(
SqlStep(
starts_with="INSERT INTO market_data.trades",
fetchone=(1,),
),
SqlStep(
starts_with="SELECT checkpoint.venue",
fetchone=_checkpoint_row(previous, revision=7),
),
SqlStep(
starts_with="UPDATE market_data.trade_stream_checkpoints",
fetchone=(1,),
),
SqlStep(
starts_with="SELECT checkpoint.venue",
fetchone=_checkpoint_row(current, revision=8),
),
)
result = repository.store_trade_and_advance_checkpoint(
venue=VENUE,
expected_trade=previous,
trade=current,
observed_at=OBSERVED_AT,
)
assert result.trade == current
assert result.revision == 8
update_parameters = connection.calls[2][1]
assert update_parameters[-1] == 7
assert connection.exit_exception_types == [None]
def test_retry_after_committed_candidate_keeps_original_revision() -> None:
previous = _trade(trade_id=100)
current = _trade(
trade_id=101,
executed_at=EXECUTED_AT + timedelta(milliseconds=1),
)
repository, connection, _ = _repository(
SqlStep(starts_with="INSERT INTO market_data.trades"),
SqlStep(
starts_with="SELECT price, quantity",
fetchone=_existing_trade_row(current),
),
SqlStep(
starts_with="SELECT checkpoint.venue",
fetchone=_checkpoint_row(current, revision=8),
),
)
result = repository.store_trade_and_advance_checkpoint(
venue=VENUE,
expected_trade=previous,
trade=current,
observed_at=OBSERVED_AT,
)
assert result.revision == 8
assert not any(
statement.startswith(
"UPDATE market_data.trade_stream_checkpoints"
)
for statement, _ in connection.calls
)
assert connection.exit_exception_types == [None]
def test_stale_expected_checkpoint_rolls_back_candidate_trade() -> None:
stale = _trade(trade_id=100)
database_current = _trade(
trade_id=101,
executed_at=EXECUTED_AT + timedelta(milliseconds=1),
)
candidate = _trade(
trade_id=102,
executed_at=EXECUTED_AT + timedelta(milliseconds=2),
)
repository, connection, _ = _repository(
SqlStep(
starts_with="INSERT INTO market_data.trades",
fetchone=(1,),
),
SqlStep(
starts_with="SELECT checkpoint.venue",
fetchone=_checkpoint_row(database_current, revision=2),
),
)
with pytest.raises(
MarketDataCheckpointConflictError,
match="differs from expected",
):
repository.store_trade_and_advance_checkpoint(
venue=VENUE,
expected_trade=stale,
trade=candidate,
observed_at=OBSERVED_AT,
)
assert connection.exit_exception_types == [
MarketDataCheckpointConflictError
]
def test_checkpoint_database_error_rolls_back_whole_transaction() -> None:
previous = _trade(trade_id=100)
current = _trade(
trade_id=101,
executed_at=EXECUTED_AT + timedelta(milliseconds=1),
)
repository, connection, _ = _repository(
SqlStep(
starts_with="INSERT INTO market_data.trades",
fetchone=(1,),
),
SqlStep(
starts_with="SELECT checkpoint.venue",
fetchone=_checkpoint_row(previous, revision=1),
),
SqlStep(
starts_with="UPDATE market_data.trade_stream_checkpoints",
error=RuntimeError("database failed"),
),
)
with pytest.raises(
MarketDataStorageOperationError,
match="atomically",
) as error_info:
repository.store_trade_and_advance_checkpoint(
venue=VENUE,
expected_trade=previous,
trade=current,
observed_at=OBSERVED_AT,
)
assert isinstance(error_info.value.__cause__, RuntimeError)
assert connection.exit_exception_types == [RuntimeError]
def test_half_cycle_candidate_is_rejected_as_ambiguous() -> None:
previous = _trade(trade_id=0)
ambiguous = _trade(
trade_id=SIGNED_TRADE_ID_MIN,
executed_at=EXECUTED_AT + timedelta(milliseconds=1),
)
repository, connection, _ = _repository(
SqlStep(
starts_with="INSERT INTO market_data.trades",
fetchone=(1,),
),
SqlStep(
starts_with="SELECT checkpoint.venue",
fetchone=_checkpoint_row(previous, revision=1),
),
)
with pytest.raises(
MarketDataCheckpointConflictError,
match="ambiguous",
):
repository.store_trade_and_advance_checkpoint(
venue=VENUE,
expected_trade=previous,
trade=ambiguous,
observed_at=OBSERVED_AT,
)
assert connection.exit_exception_types == [
MarketDataCheckpointConflictError
]
def test_invalid_expected_trade_is_rejected_without_io() -> None:
repository, _, provider = _repository()
with pytest.raises(
MarketDataStorageValidationError,
match="expected_trade",
):
repository.store_trade_and_advance_checkpoint(
venue=VENUE,
expected_trade=object(), # type: ignore[arg-type]
trade=_trade(),
observed_at=OBSERVED_AT,
)
assert provider.calls == 0

View File

@@ -17,6 +17,7 @@ from src.market_data.storage import (
MarketDataStorageValidationError,
MarketDataWriteResult,
MarketDataWriteStatus,
PersistentTradeCheckpoint,
TradeStorageObservationSink,
)
@@ -42,13 +43,18 @@ class RecordingTradeStorage:
self,
*,
result: object | None = None,
checkpoint_result: object | None = None,
error: Exception | None = None,
) -> None:
self.result = result or MarketDataWriteResult(
status=MarketDataWriteStatus.INSERTED,
)
self.checkpoint_result = checkpoint_result
self.error = error
self.calls: list[tuple[str, Trade, datetime]] = []
self.checkpoint_calls: list[
tuple[str, Trade | None, Trade, datetime]
] = []
def store_trade(
self,
@@ -73,12 +79,91 @@ class RecordingTradeStorage:
) -> MarketDataBatchWriteResult:
raise AssertionError("Runtime persists observations one by one")
def load_checkpoint(
self,
*,
venue: str,
symbol: str,
) -> PersistentTradeCheckpoint | None:
raise AssertionError((venue, symbol))
def load_checkpoint_tail(
self,
*,
venue: str,
checkpoint: PersistentTradeCheckpoint,
limit: int,
) -> tuple[Trade, ...]:
raise AssertionError((venue, checkpoint, limit))
def load_latest_trade_tail(
self,
*,
venue: str,
symbol: str,
limit: int,
) -> tuple[Trade, ...]:
raise AssertionError((venue, symbol, limit))
def adopt_existing_trade_as_checkpoint(
self,
*,
venue: str,
trade: Trade,
) -> PersistentTradeCheckpoint:
raise AssertionError((venue, trade))
def store_trade_and_advance_checkpoint(
self,
*,
venue: str,
expected_trade: Trade | None,
trade: Trade,
observed_at: datetime,
) -> PersistentTradeCheckpoint:
self.checkpoint_calls.append(
(venue, expected_trade, trade, observed_at)
)
if self.error is not None:
raise self.error
if self.checkpoint_result is not None:
return self.checkpoint_result # type: ignore[return-value]
return PersistentTradeCheckpoint(
venue=venue,
trade=trade,
revision=1,
updated_at=observed_at,
)
class RecordingWriteOnlyTradeStorage:
def store_trade(
self,
*,
venue: str,
trade: Trade,
observed_at: datetime,
) -> MarketDataWriteResult:
raise AssertionError((venue, trade, observed_at))
def store_trades(
self,
*,
venue: str,
trades: tuple[Trade, ...],
observed_at: datetime,
) -> MarketDataBatchWriteResult:
raise AssertionError((venue, trades, observed_at))
@pytest.mark.parametrize(
"status",
tuple(MarketDataWriteStatus),
)
def test_forwards_observation_and_accepts_all_success_statuses(
def test_duplicate_forwards_observation_and_accepts_all_success_statuses(
status: MarketDataWriteStatus,
) -> None:
storage = RecordingTradeStorage(
@@ -91,7 +176,7 @@ def test_forwards_observation_and_accepts_all_success_statuses(
)
trade = make_trade()
result = sink.persist(trade)
result = sink.persist_duplicate(trade)
assert result is None
assert storage.calls == [
@@ -101,6 +186,42 @@ def test_forwards_observation_and_accepts_all_success_statuses(
OBSERVED_AT,
)
]
assert storage.checkpoint_calls == []
def test_accepted_trade_atomically_advances_checkpoint() -> None:
storage = RecordingTradeStorage()
sink = TradeStorageObservationSink(
trade_storage=storage,
venue=" dzengi ",
clock=lambda: OBSERVED_AT,
)
previous_trade = make_trade()
trade = Trade(
symbol=previous_trade.symbol,
trade_id=previous_trade.trade_id + 1,
price=previous_trade.price,
quantity=previous_trade.quantity,
executed_at=previous_trade.executed_at,
aggressor_side=previous_trade.aggressor_side,
source=previous_trade.source,
)
result = sink.persist_accepted(
trade,
expected_trade=previous_trade,
)
assert result is None
assert storage.checkpoint_calls == [
(
VENUE,
previous_trade,
trade,
OBSERVED_AT,
)
]
assert storage.calls == []
def test_implements_acquisition_side_sink_protocol() -> None:
@@ -113,7 +234,11 @@ def test_implements_acquisition_side_sink_protocol() -> None:
assert not hasattr(sink, "__dict__")
def test_storage_error_is_not_wrapped() -> None:
@pytest.mark.parametrize(
"operation",
("accepted", "duplicate"),
)
def test_storage_error_is_not_wrapped(operation: str) -> None:
storage_error = RuntimeError("storage failed")
sink = TradeStorageObservationSink(
trade_storage=RecordingTradeStorage(error=storage_error),
@@ -125,7 +250,13 @@ def test_storage_error_is_not_wrapped() -> None:
RuntimeError,
match="storage failed",
) as error_info:
sink.persist(make_trade())
if operation == "accepted":
sink.persist_accepted(
make_trade(),
expected_trade=None,
)
else:
sink.persist_duplicate(make_trade())
assert error_info.value is storage_error
@@ -141,7 +272,26 @@ def test_rejects_invalid_storage_result() -> None:
TypeError,
match="MarketDataWriteResult",
):
sink.persist(make_trade())
sink.persist_duplicate(make_trade())
def test_rejects_invalid_checkpoint_result() -> None:
sink = TradeStorageObservationSink(
trade_storage=RecordingTradeStorage(
checkpoint_result=object(),
),
venue=VENUE,
clock=lambda: OBSERVED_AT,
)
with pytest.raises(
TypeError,
match="PersistentTradeCheckpoint",
):
sink.persist_accepted(
make_trade(),
expected_trade=None,
)
def test_rejects_invalid_dependencies() -> None:
@@ -151,6 +301,15 @@ def test_rejects_invalid_dependencies() -> None:
venue=VENUE,
)
with pytest.raises(
TypeError,
match="TradeCheckpointStorageProtocol",
):
TradeStorageObservationSink(
trade_storage=RecordingWriteOnlyTradeStorage(),
venue=VENUE,
)
with pytest.raises(TypeError, match="clock must be callable"):
TradeStorageObservationSink(
trade_storage=RecordingTradeStorage(),

View File

@@ -105,6 +105,7 @@ def test_default_migrations_have_stable_order_and_names() -> None:
(5, "add_trade_observation_sources"),
(6, "add_quote_and_candle_observation_sources"),
(7, "create_market_data_partition_registry"),
(8, "create_trade_stream_checkpoints"),
)
@@ -119,8 +120,8 @@ def test_default_schema_defines_partitions_identities_and_constraints() -> None:
assert "CREATE TABLE market_data.trades" in sql
assert "PRIMARY KEY (venue, symbol, trade_id, executed_at)" in sql
assert "trade_id BETWEEN -2147483648 AND 2147483647" in sql
assert sql.count("CHECK (BTRIM(venue) <> '')") == 3
assert sql.count("CHECK (BTRIM(symbol) <> '')") == 3
assert sql.count("CHECK (BTRIM(venue) <> '')") == 4
assert sql.count("CHECK (BTRIM(symbol) <> '')") == 4
assert sql.count("CHECK (BTRIM(source) <> '')") == 3
assert "PARTITION BY RANGE (executed_at)" in sql
assert "CREATE TABLE market_data.quotes" in sql
@@ -143,6 +144,18 @@ def test_default_schema_defines_partitions_identities_and_constraints() -> None:
assert "partition_bound TEXT NOT NULL" in sql
assert "BTRIM(partition_bound) <> ''" in sql
assert "range_end > range_start" in sql
assert "CREATE TABLE market_data.trade_stream_checkpoints" in sql
assert "PRIMARY KEY (venue, symbol)" in sql
assert "revision BIGINT NOT NULL" in sql
assert "checkpoint_schema_version INTEGER NOT NULL DEFAULT 1" in sql
assert "CONSTRAINT trade_stream_checkpoints_trade_fk" in sql
assert "FOREIGN KEY (" in sql
assert ") REFERENCES market_data.trades (" in sql
assert "ON UPDATE NO ACTION" in sql
assert "ON DELETE NO ACTION" in sql
assert "DEFERRABLE INITIALLY DEFERRED" in sql
assert "CHECK (revision > 0)" in sql
assert "CHECK (checkpoint_schema_version > 0)" in sql
def test_run_locks_and_applies_every_pending_migration_in_order() -> None:
@@ -150,7 +163,7 @@ def test_run_locks_and_applies_every_pending_migration_in_order() -> None:
result = runner.run()
assert result == (1, 2, 3, 4, 5, 6, 7)
assert result == (1, 2, 3, 4, 5, 6, 7, 8)
assert provider.calls == 1
assert connection.entered == 1
assert connection.exited == 1
@@ -167,7 +180,7 @@ def test_run_locks_and_applies_every_pending_migration_in_order() -> None:
)
and isinstance(parameters, tuple)
)
assert inserted_versions == (1, 2, 3, 4, 5, 6, 7)
assert inserted_versions == (1, 2, 3, 4, 5, 6, 7, 8)
def test_run_skips_already_applied_migrations() -> None:
@@ -196,7 +209,7 @@ def test_run_applies_only_migrations_after_existing_prefix() -> None:
result = runner.run()
assert result == (3, 4, 5, 6, 7)
assert result == (3, 4, 5, 6, 7, 8)
inserted_versions = tuple(
parameters[0]
for statement, parameters in cursor.calls
@@ -205,7 +218,7 @@ def test_run_applies_only_migrations_after_existing_prefix() -> None:
)
and isinstance(parameters, tuple)
)
assert inserted_versions == (3, 4, 5, 6, 7)
assert inserted_versions == (3, 4, 5, 6, 7, 8)
def test_run_rejects_unknown_applied_version() -> None:

View File

@@ -5,11 +5,14 @@ from typing import Any
import pytest
from psycopg.conninfo import conninfo_to_dict
from tests.support import postgres_market_data
from tests.support.postgres_market_data import (
POSTGRES_TEST_APPLICATION_NAME,
POSTGRES_TEST_CONTROL_APPLICATION_NAME,
count_other_test_connections,
load_postgres_test_settings,
reset_postgres_test_database,
wait_for_postgres_relation_lock_waiters,
)
@@ -59,6 +62,52 @@ class RecordingControlConnection:
)
class QueuedCursor:
def __init__(self, connection: QueuedControlConnection) -> None:
self._connection = connection
def __enter__(self) -> QueuedCursor:
return self
def __exit__(self, *_: object) -> None:
return None
def execute(
self,
statement: object,
parameters: object = None,
) -> None:
self._connection.statements.append(
(" ".join(str(statement).split()), parameters)
)
def fetchone(self) -> object:
if not self._connection.rows:
raise AssertionError("Не подготовлен ответ PostgreSQL")
return self._connection.rows.pop(0)
class QueuedControlConnection:
def __init__(self, *, rows: list[object]) -> None:
self.rows = list(rows)
self.statements: list[tuple[str, object]] = []
def cursor(self) -> QueuedCursor:
return QueuedCursor(self)
class FakeClock:
def __init__(self) -> None:
self.current = 0.0
def monotonic(self) -> float:
return self.current
def sleep(self, seconds: float) -> None:
self.current += seconds
def test_postgres_harness_is_disabled_without_explicit_flag() -> None:
assert load_postgres_test_settings({}) is None
@@ -188,3 +237,230 @@ def test_postgres_reset_refuses_unvalidated_connection_before_drop(
assert connection.statements == [
"SELECT current_database(), current_setting('application_name')",
]
def test_connection_count_includes_every_non_control_database_session() -> None:
database_name = "dzentra_test_all_connections"
connection: Any = QueuedControlConnection(
rows=[
(database_name, POSTGRES_TEST_CONTROL_APPLICATION_NAME),
(3,),
]
)
result = count_other_test_connections(connection)
assert result == 3
assert connection.statements == [
(
"SELECT current_database(), "
"current_setting('application_name')",
None,
),
(
"SELECT COUNT(*) FROM pg_catalog.pg_stat_activity "
"WHERE datname = %s "
"AND backend_type = 'client backend' "
"AND application_name IS DISTINCT FROM %s",
(database_name, POSTGRES_TEST_CONTROL_APPLICATION_NAME),
),
]
@pytest.mark.parametrize(
"identity",
(
("production", POSTGRES_TEST_CONTROL_APPLICATION_NAME),
("dzentra_test_all_connections", "another-application"),
None,
),
)
def test_connection_count_revalidates_control_connection(
identity: object,
) -> None:
connection: Any = QueuedControlConnection(rows=[identity, (0,)])
with pytest.raises(RuntimeError, match="validated test control"):
count_other_test_connections(connection)
assert len(connection.statements) == 1
@pytest.mark.parametrize(
"count_row",
(
(True,),
(-1,),
("1",),
[1],
None,
),
)
def test_connection_count_rejects_invalid_postgres_result(
count_row: object,
) -> None:
connection: Any = QueuedControlConnection(
rows=[
(
"dzentra_test_all_connections",
POSTGRES_TEST_CONTROL_APPLICATION_NAME,
),
count_row,
]
)
with pytest.raises(RuntimeError, match="invalid connection count"):
count_other_test_connections(connection)
def test_relation_waiter_uses_validated_database_and_exact_count(
monkeypatch: pytest.MonkeyPatch,
) -> None:
database_name = "dzentra_test_relation_waiter"
connection: Any = QueuedControlConnection(
rows=[
(database_name, POSTGRES_TEST_CONTROL_APPLICATION_NAME),
(16_384,),
(0,),
(2,),
]
)
clock = FakeClock()
monkeypatch.setattr(postgres_market_data.time, "monotonic", clock.monotonic)
monkeypatch.setattr(postgres_market_data.time, "sleep", clock.sleep)
wait_for_postgres_relation_lock_waiters(
connection,
relation_name="market_data.trade_stream_checkpoints",
expected_count=2,
timeout_seconds=1.0,
)
assert connection.rows == []
assert connection.statements[1] == (
"SELECT pg_catalog.to_regclass(%s)::oid",
("market_data.trade_stream_checkpoints",),
)
lock_statement, lock_parameters = connection.statements[2]
assert "locktype = 'relation'" in lock_statement
assert "AND NOT granted" in lock_statement
assert lock_parameters == (database_name, 16_384)
assert connection.statements[3] == (
lock_statement,
lock_parameters,
)
def test_relation_waiter_times_out_with_observed_count(
monkeypatch: pytest.MonkeyPatch,
) -> None:
connection: Any = QueuedControlConnection(
rows=[
(
"dzentra_test_relation_timeout",
POSTGRES_TEST_CONTROL_APPLICATION_NAME,
),
(42,),
(0,),
(0,),
]
)
clock = FakeClock()
monkeypatch.setattr(postgres_market_data.time, "monotonic", clock.monotonic)
monkeypatch.setattr(postgres_market_data.time, "sleep", clock.sleep)
with pytest.raises(
TimeoutError,
match=r"expected 1, observed 0\.",
):
wait_for_postgres_relation_lock_waiters(
connection,
relation_name="market_data.trade_stream_checkpoints",
expected_count=1,
timeout_seconds=0.01,
)
@pytest.mark.parametrize(
"relation_row",
(
None,
(),
(None,),
(True,),
("42",),
(0,),
),
)
def test_relation_waiter_rejects_unresolved_or_invalid_relation(
relation_row: object,
) -> None:
connection: Any = QueuedControlConnection(
rows=[
(
"dzentra_test_relation_shape",
POSTGRES_TEST_CONTROL_APPLICATION_NAME,
),
relation_row,
]
)
with pytest.raises(RuntimeError, match="requested test relation"):
wait_for_postgres_relation_lock_waiters(
connection,
relation_name="market_data.trade_stream_checkpoints",
expected_count=1,
)
@pytest.mark.parametrize(
"waiter_row",
(
None,
(),
(True,),
(-1,),
("1",),
[1],
),
)
def test_relation_waiter_rejects_invalid_count_result(
waiter_row: object,
) -> None:
connection: Any = QueuedControlConnection(
rows=[
(
"dzentra_test_waiter_shape",
POSTGRES_TEST_CONTROL_APPLICATION_NAME,
),
(42,),
waiter_row,
]
)
with pytest.raises(RuntimeError, match="invalid relation-lock waiter"):
wait_for_postgres_relation_lock_waiters(
connection,
relation_name="market_data.trade_stream_checkpoints",
expected_count=1,
)
@pytest.mark.parametrize(
"timeout_seconds",
(0.0, -1.0, float("inf"), float("nan"), True),
)
def test_relation_waiter_requires_positive_finite_timeout(
timeout_seconds: float,
) -> None:
connection: Any = QueuedControlConnection(rows=[])
with pytest.raises(ValueError, match="positive finite"):
wait_for_postgres_relation_lock_waiters(
connection,
relation_name="market_data.trade_stream_checkpoints",
expected_count=1,
timeout_seconds=timeout_seconds,
)
assert connection.statements == []

View File

@@ -5,6 +5,7 @@ import asyncio
import pytest
from tests.support.trade_stream_runtime import (
active_owned_task_names,
run_scenario,
wait_until_or_runtime_exit,
)
@@ -90,3 +91,30 @@ def test_wait_until_or_runtime_exit_rejects_clean_early_runtime_exit() -> None:
)
run_scenario(scenario())
def test_owned_task_detection_covers_application_and_processing_tasks() -> None:
async def scenario() -> None:
release = asyncio.Event()
names = (
"application-shutdown",
"telegram-polling",
"trade-stream-market-processing",
"persistent-application-verification",
)
tasks = tuple(
asyncio.create_task(release.wait(), name=name)
for name in names
)
try:
await asyncio.sleep(0)
assert active_owned_task_names() == tuple(sorted(names))
finally:
release.set()
await asyncio.gather(*tasks)
assert active_owned_task_names() == ()
run_scenario(scenario())