46 lines
1.6 KiB
Python
46 lines
1.6 KiB
Python
#!/usr/bin/env python3
|
|
|
|
# This script is a CI test in tilt that verifies that prices are flowing through the entire system properly.
|
|
# It checks that all prices being published by the pyth publisher are showing up at the price service.
|
|
import base58
|
|
import logging
|
|
import time
|
|
from pyth_utils import *
|
|
|
|
logging.basicConfig(
|
|
level=logging.DEBUG, format="%(asctime)s | %(module)s | %(levelname)s | %(message)s"
|
|
)
|
|
|
|
# Where to read the set of accounts from
|
|
PYTH_TEST_ACCOUNTS_HOST = "pyth"
|
|
PYTH_TEST_ACCOUNTS_PORT = 4242
|
|
|
|
PRICE_SERVICE_HOST = "pyth-price-server"
|
|
PRICE_SERVICE_PORT = 4200
|
|
|
|
def base58_to_hex(base58_string):
|
|
asc_string = base58.b58decode(base58_string)
|
|
return asc_string.hex()
|
|
|
|
all_prices_attested = False
|
|
while not all_prices_attested:
|
|
publisher_state_map = get_pyth_accounts(PYTH_TEST_ACCOUNTS_HOST, PYTH_TEST_ACCOUNTS_PORT)
|
|
pyth_price_account_ids = sorted([base58_to_hex(x["price"]) for x in publisher_state_map["symbols"]])
|
|
price_ids = sorted(get_json(PRICE_SERVICE_HOST, PRICE_SERVICE_PORT, "/api/price_feed_ids"))
|
|
|
|
if price_ids == pyth_price_account_ids:
|
|
if publisher_state_map["all_symbols_added"]:
|
|
logging.info("Price ids match and all symbols added. Enabling readiness probe")
|
|
all_prices_attested = True
|
|
else:
|
|
logging.info("Price ids match but still waiting for more symbols to come online.")
|
|
else:
|
|
logging.info("Price ids do not match")
|
|
logging.info(f"published ids: {pyth_price_account_ids}")
|
|
logging.info(f"attested ids: {price_ids}")
|
|
|
|
time.sleep(10)
|
|
|
|
# Let k8s know the service is up
|
|
readiness()
|