feat(hip-3-pusher): set up pytest, websocket timeout and better reconnect handling

merolish · merolish · commit 5366f99f4403 · 2025-09-25T02:52:53.000-05:00
diff --git a/apps/hip-3-pusher/pyproject.toml b/apps/hip-3-pusher/pyproject.toml
@@ -14,3 +14,8 @@ dependencies = [
     "prometheus-client~=0.23.1",
     "websockets~=15.0.1",
 ]
+
+[dependency-groups]
+dev = [
+    "pytest>=8.4.2",
+]
diff --git a/apps/hip-3-pusher/src/pusher/config.py b/apps/hip-3-pusher/src/pusher/config.py
diff --git a/apps/hip-3-pusher/src/pusher/hermes_listener.py b/apps/hip-3-pusher/src/pusher/hermes_listener.py
@@ -4,8 +4,8 @@
 import time
 import websockets
 
-from config import Config
-from price_state import PriceState, PriceUpdate
+from pusher.config import Config
+from pusher.price_state import PriceState, PriceUpdate
 
 
 class HermesListener:
diff --git a/apps/hip-3-pusher/src/pusher/hyperliquid_listener.py b/apps/hip-3-pusher/src/pusher/hyperliquid_listener.py
@@ -4,8 +4,8 @@
 from loguru import logger
 import time
 
-from config import Config
-from price_state import PriceState, PriceUpdate
+from pusher.config import Config
+from pusher.price_state import PriceState, PriceUpdate
 
 # This will be in config, but note here.
 # Other RPC providers exist but so far we've seen their support is incomplete.
diff --git a/apps/hip-3-pusher/src/pusher/kms_signer.py b/apps/hip-3-pusher/src/pusher/kms_signer.py
@@ -10,7 +10,7 @@
 from hyperliquid.utils.signing import get_timestamp_ms, action_hash, construct_phantom_agent, l1_payload
 from loguru import logger
 
-from config import Config
+from pusher.config import Config
 
 SECP256K1_N_HALF = SECP256K1_N // 2
 
diff --git a/apps/hip-3-pusher/src/pusher/lazer_listener.py b/apps/hip-3-pusher/src/pusher/lazer_listener.py
@@ -4,8 +4,8 @@
 import time
 import websockets
 
-from config import Config
-from price_state import PriceState, PriceUpdate
+from pusher.config import Config
+from pusher.price_state import PriceState, PriceUpdate
 
 
 class LazerListener:
diff --git a/apps/hip-3-pusher/src/pusher/main.py b/apps/hip-3-pusher/src/pusher/main.py
@@ -5,13 +5,13 @@
 import sys
 import tomllib
 
-from config import Config
-from hyperliquid_listener import HyperliquidListener
-from lazer_listener import LazerListener
-from hermes_listener import HermesListener
-from price_state import PriceState
-from publisher import Publisher
-from metrics import Metrics
+from pusher.config import Config
+from pusher.hyperliquid_listener import HyperliquidListener
+from pusher.lazer_listener import LazerListener
+from pusher.hermes_listener import HermesListener
+from pusher.price_state import PriceState
+from pusher.publisher import Publisher
+from pusher.metrics import Metrics
 
 
 def load_config():
diff --git a/apps/hip-3-pusher/src/pusher/metrics.py b/apps/hip-3-pusher/src/pusher/metrics.py
@@ -3,7 +3,7 @@
 from opentelemetry.metrics import get_meter_provider, set_meter_provider
 from opentelemetry.sdk.metrics import MeterProvider
 
-from config import Config
+from pusher.config import Config
 
 METER_NAME = "hip3pusher"
 
diff --git a/apps/hip-3-pusher/src/pusher/price_state.py b/apps/hip-3-pusher/src/pusher/price_state.py
@@ -1,7 +1,7 @@
 from loguru import logger
 import time
 
-from config import Config
+from pusher.config import Config
 
 DEFAULT_STALE_PRICE_THRESHOLD_SECONDS = 5
 
diff --git a/apps/hip-3-pusher/src/pusher/publisher.py b/apps/hip-3-pusher/src/pusher/publisher.py
@@ -1,16 +1,15 @@
 import asyncio
-
 from loguru import logger
 
 from eth_account import Account
 from eth_account.signers.local import LocalAccount
 from hyperliquid.exchange import Exchange
 from hyperliquid.utils.constants import TESTNET_API_URL, MAINNET_API_URL
 
-from config import Config
-from kms_signer import KMSSigner
-from metrics import Metrics
-from price_state import PriceState
+from pusher.config import Config
+from pusher.kms_signer import KMSSigner
+from pusher.metrics import Metrics
+from pusher.price_state import PriceState
 
 
 class Publisher:
diff --git a/apps/hip-3-pusher/tests/test_price_state.py b/apps/hip-3-pusher/tests/test_price_state.py
@@ -0,0 +1,70 @@
+import time
+
+from pusher.config import Config, LazerConfig, HermesConfig
+from pusher.price_state import PriceState, PriceUpdate
+
+
+def get_config():
+    config: Config = Config.model_construct()
+    config.stale_price_threshold_seconds = 5
+    config.lazer = LazerConfig.model_construct()
+    config.lazer.base_feed_exponent = -8
+    config.lazer.quote_feed_exponent = -8
+    config.hermes = HermesConfig.model_construct()
+    config.hermes.base_feed_exponent = -8
+    config.hermes.quote_feed_exponent = -8
+    return config
+
+
+def test_good_hl_price():
+    config = get_config()
+    price_state = PriceState(config)
+    now = time.time()
+    price_state.hl_oracle_price = PriceUpdate("110000.0", now - price_state.stale_price_threshold_seconds / 2.0)
+
+    oracle_px = price_state.get_current_oracle_price()
+    assert oracle_px == price_state.hl_oracle_price.price
+    assert oracle_px == "110000.0"
+
+
+
+def test_fallback_lazer():
+    config = get_config()
+    price_state = PriceState(config)
+    now = time.time()
+    price_state.hl_oracle_price = PriceUpdate("110000.0", now - price_state.stale_price_threshold_seconds - 1.0)
+    price_state.lazer_base_price = PriceUpdate("11050000000000", now - price_state.stale_price_threshold_seconds / 2.0)
+    price_state.lazer_quote_price = PriceUpdate("99000000", now - price_state.stale_price_threshold_seconds / 2.0)
+
+    oracle_px = price_state.get_current_oracle_price()
+    assert oracle_px == price_state.get_lazer_price()
+    assert oracle_px == "111616.16"
+
+
+
+def test_fallback_hermes():
+    config = get_config()
+    price_state = PriceState(config)
+    now = time.time()
+    price_state.hl_oracle_price = PriceUpdate("110000.0", now - price_state.stale_price_threshold_seconds - 1.0)
+    price_state.lazer_base_price = PriceUpdate("11050000000000", now - price_state.stale_price_threshold_seconds - 1.0)
+    price_state.lazer_quote_price = PriceUpdate("99000000", now - price_state.stale_price_threshold_seconds / 2.0)
+    price_state.hermes_base_price = PriceUpdate("11100000000000", now - price_state.stale_price_threshold_seconds / 2.0)
+    price_state.hermes_quote_price = PriceUpdate("98000000", now - price_state.stale_price_threshold_seconds / 2.0)
+
+    oracle_px = price_state.get_current_oracle_price()
+    assert oracle_px == price_state.get_hermes_price()
+    assert oracle_px == "113265.31"
+
+
+def test_all_fail():
+    config = get_config()
+    price_state = PriceState(config)
+    now = time.time()
+    price_state.hl_oracle_price = PriceUpdate("110000.0", now - price_state.stale_price_threshold_seconds - 1.0)
+    price_state.hl_oracle_price = PriceUpdate("110000.0", now - price_state.stale_price_threshold_seconds - 1.0)
+    price_state.lazer_base_price = PriceUpdate("11050000000000", now - price_state.stale_price_threshold_seconds - 1.0)
+    price_state.lazer_quote_price = PriceUpdate("99000000", now - price_state.stale_price_threshold_seconds - 1.0)
+    price_state.hermes_base_price = PriceUpdate("11100000000000", now - price_state.stale_price_threshold_seconds - 1.0)
+    price_state.hermes_quote_price = PriceUpdate("98000000", now - price_state.stale_price_threshold_seconds - 1.0)
+    assert price_state.get_current_oracle_price() is None
diff --git a/apps/hip-3-pusher/uv.lock b/apps/hip-3-pusher/uv.lock

Original file line number	Diff line number	Diff line change
`@@ -14,3 +14,8 @@ dependencies = [`
`14`	`14`	`"prometheus-client~=0.23.1",`
`15`	`15`	`"websockets~=15.0.1",`
`16`	`16`	`]`
	`17`	`+`
	`18`	`+[dependency-groups]`
	`19`	`+dev = [`
	`20`	`+ "pytest>=8.4.2",`
	`21`	`+]`