Only run `kraken` feed tests in CI, use `open_test_pikerd()`

agg_feedz
Tyler Goodlet 2023-01-09 18:16:55 -05:00
parent 06622105cd
commit 593db0ed0d
1 changed files with 16 additions and 16 deletions

View File

@ -4,15 +4,15 @@ Data feed layer APIs, performance, msg throttling.
'''
from collections import Counter
from pprint import pprint
from typing import AsyncContextManager
import pytest
# import tractor
import trio
from piker import (
open_piker_runtime,
from piker.data import (
ShmArray,
open_feed,
)
from piker.data import ShmArray
from piker.data._source import (
unpack_fqsn,
)
@ -22,25 +22,33 @@ from piker.data._source import (
'fqsns',
[
# binance
(100, {'btcusdt.binance', 'ethusdt.binance'}),
(100, {'btcusdt.binance', 'ethusdt.binance'}, False),
# kraken
(20, {'ethusdt.kraken', 'xbtusd.kraken'}),
(20, {'ethusdt.kraken', 'xbtusd.kraken'}, True),
# binance + kraken
(100, {'btcusdt.binance', 'xbtusd.kraken'}),
(100, {'btcusdt.binance', 'xbtusd.kraken'}, False),
],
ids=lambda param: f'quotes={param[0]}@fqsns={param[1]}',
)
def test_multi_fqsn_feed(
open_test_pikerd: AsyncContextManager,
fqsns: set[str],
ci_env: bool
):
'''
Start a real-time data feed for provided fqsn and pull
a few quotes then simply shut down.
'''
max_quotes, fqsns = fqsns
max_quotes, fqsns, run_in_ci = fqsns
if (
ci_env
and not run_in_ci
):
pytest.skip('Skipping CI disabled test due to feed restrictions')
brokers = set()
for fqsn in fqsns:
@ -49,15 +57,7 @@ def test_multi_fqsn_feed(
async def main():
async with (
open_piker_runtime(
'test_basic_rt_feed',
# XXX tractor BUG: this doesn't translate through to the
# ``tractor._state._runtimevars``...
# registry_addr=('127.0.0.1', 6666),
debug_mode=True,
),
open_test_pikerd(),
open_feed(
fqsns,
loglevel='info',