pikers
/
piker
3
0
Fork 0
Code Issues 4 Pull Requests 6 Packages Projects Releases Wiki Activity

Expect init startup quotes from each symbol

agg_feedz
Tyler Goodlet 2022-11-09 13:03:53 -05:00
parent e7de5404d3
commit 32b36aa042
1 changed files with 14 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
tests/test_feeds.py
View File

@ -5,7 +5,7 @@ Data feed layer APIs, performance, msg throttling.
from pprint import pprint from pprint import pprint
import pytest import pytest
import tractor # import tractor
import trio import trio
from piker import ( from piker import (
open_piker_runtime, open_piker_runtime,
@ -37,6 +37,7 @@ def test_basic_rt_feed(
# ``tractor._state._runtimevars``... # ``tractor._state._runtimevars``...
registry_addr=('127.0.0.1', 6666), registry_addr=('127.0.0.1', 6666),
debug_mode=True, debug_mode=True,
loglevel='runtime',
), ),
open_feed( open_feed(
fqsns, fqsns,
@ -55,8 +56,20 @@ def test_basic_rt_feed(
ohlcv: ShmArray = flume.rt_shm ohlcv: ShmArray = flume.rt_shm
hist_ohlcv: ShmArray = flume.hist_shm hist_ohlcv: ShmArray = flume.hist_shm
# stream some ticks and ensure we see data from both symbol
# subscriptions.
quote_count: int = 0 quote_count: int = 0
stream = feed.streams['binance'] stream = feed.streams['binance']
# pull the first couple startup quotes and ensure
# they match the history buffer last entries.
for _ in range(1):
first_quotes = await stream.receive()
for fqsn, quote in first_quotes.items():
assert fqsn in fqsns
flume = feed.flumes[fqsn]
assert quote['last'] == flume.first_quote['last']
async for quotes in stream: async for quotes in stream:
for fqsn, quote in quotes.items(): for fqsn, quote in quotes.items():