From 36868bb86e52231d2a1cea7ac00fbc63190c7b5a Mon Sep 17 00:00:00 2001
From: Tyler Goodlet <jgbt@protonmail.com>
Date: Thu, 10 Nov 2022 10:38:45 -0500
Subject: [PATCH] Add `kraken` test, ensure single broker-provider for now

---
 tests/test_feeds.py | 21 ++++++++++++++++++---
 1 file changed, 18 insertions(+), 3 deletions(-)

diff --git a/tests/test_feeds.py b/tests/test_feeds.py
index 2fb5f693..5bb9a510 100644
--- a/tests/test_feeds.py
+++ b/tests/test_feeds.py
@@ -13,14 +13,18 @@ from piker import (
     open_feed,
 )
 from piker.data import ShmArray
+from piker.data._source import (
+    unpack_fqsn,
+)
 
 
 @pytest.mark.parametrize(
     'fqsns',
     [
-        {'btcusdt.binance', 'ethusdt.binance'}
+        (100, {'btcusdt.binance', 'ethusdt.binance'}),
+        (50, {'xbteur.kraken', 'xbtusd.kraken'}),
     ],
-    ids=lambda param: f'fqsns={param}',
+    ids=lambda param: f'quotes={param[0]}@fqsns={param[1]}',
 )
 def test_basic_rt_feed(
     fqsns: set[str],
@@ -30,6 +34,17 @@ def test_basic_rt_feed(
     a few quotes then simply shut down.
 
     '''
+    max_quotes, fqsns = fqsns
+
+    brokers = set()
+    for fqsn in fqsns:
+        brokername, key, suffix = unpack_fqsn(fqsn)
+        brokers.add(brokername)
+
+    # NOTE: we only have single broker-backed multi-symbol streams
+    # currently.
+    assert len(brokers) == 1
+
     async def main():
         async with (
             open_piker_runtime(
@@ -60,7 +75,7 @@ def test_basic_rt_feed(
 
             # stream some ticks and ensure we see data from both symbol
             # subscriptions.
-            stream = feed.streams['binance']
+            stream = feed.streams[brokername]
 
             # pull the first startup quotes, one for each fqsn, and
             # ensure they match each flume's startup quote value.