From 03c57ceece22be29c97274446a1bd3296de8c8ad Mon Sep 17 00:00:00 2001
From: Tyler Goodlet <tgoodlet@gmail.com>
Date: Thu, 7 Jun 2018 00:29:17 -0400
Subject: [PATCH] Add an initial `tractor` price streaming test

---
 tests/test_tractor.py | 57 +++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 57 insertions(+)
 create mode 100644 tests/test_tractor.py

diff --git a/tests/test_tractor.py b/tests/test_tractor.py
new file mode 100644
index 0000000..ed9e184
--- /dev/null
+++ b/tests/test_tractor.py
@@ -0,0 +1,57 @@
+"""
+Actor model API testing
+"""
+import pytest
+from piker import tractor
+
+
+@pytest.fixture
+def us_symbols():
+    return ['TSLA', 'AAPL', 'CGC', 'CRON']
+
+
+@pytest.mark.trio
+async def test_rx_price_quotes_from_brokerd(us_symbols):
+    """Verify we can spawn a daemon actor and retrieve streamed price data.
+    """
+    async with tractor.open_nursery() as nursery:
+
+        # only one per host address, spawns an actor if None
+        async with tractor.find_actors('brokerd') as portals:
+            if not portals:
+                # no brokerd actor found
+                portal = await nursery.start_actor(
+                    'brokerd',
+                    ['piker.brokers.core'],
+                    statespace={
+                        'brokers2tickersubs': {},
+                        'clients': {},
+                        'dtasks': set()
+                    },
+                    main=None,
+                )
+
+            # gotta expose in a broker agnostic way...
+            # retrieve initial symbol data
+            # sd = await portal.run(
+            #     'piker.brokers.core', 'symbol_data', symbols=us_symbols)
+            # assert list(sd.keys()) == us_symbols
+
+            gen = await portal.run(
+                'piker.brokers.core',
+                '_test_price_stream',
+                broker='robinhood',
+                symbols=us_symbols,
+            )
+            # it'd sure be nice to have an asyncitertools here...
+            async for quotes in gen:
+                assert quotes
+                for key in quotes:
+                    assert key in us_symbols
+                break
+                # terminate far-end async-gen
+                # await gen.asend(None)
+                # break
+
+        await nursery.cancel()
+        # arbitter is cancelled here due to `find_actors()` internals