test pythd

Author: jhunsaker

docker/Dockerfile

@@ -16,6 +16,7 @@ RUN apt-get install -qq \
   libzstd1 \
   libzstd-dev \
   python3-pytest \
+  python3-websockets \
   sudo \
   zlib1g \
   zlib1g-dev

pyth/tests/conftest.py

@@ -290,3 +290,25 @@ def pyth_init_price(solana_test_validator, pyth_dir, pyth_add_price):
     ]
     check_call(cmd)
     return pyth_add_price
+
+
+@pytest.fixture(scope='session')
+def pythd(solana_test_validator, pyth_dir):
+
+    cmd = [
+        'pythd',
+        '-r', 'localhost',
+        '-k', pyth_dir,
+        '-x',
+        '-m', 'finalized',
+        '-d',
+    ]
+    kwargs = {
+        'stdin': DEVNULL,
+        'stdout': DEVNULL,
+        'stderr': DEVNULL,
+    }
+    with Popen(cmd, **kwargs) as p:
+        time.sleep(3)
+        yield
+        p.terminate()

pyth/tests/test_get_product.py

@@ -0,0 +1,40 @@
+import asyncio
+import json
+import websockets
+
+def test_get_product_list(solana_test_validator, pythd, pyth_init_price):
+
+    data = {
+      'jsonrpc': '2.0',
+      'id': 1,
+      'method': 'get_product_list',
+    }
+    data = json.dumps(data)
+    async def f():
+        async with websockets.connect('ws://localhost:8910/') as ws:
+            await ws.send(data)
+            result = await ws.recv()
+            return result
+    result = asyncio.run(f())
+    result = json.loads(result)
+    result = result['result']
+    result = result[0]
+    del result['account']  # TODO
+    expected_result = {
+        'attr_dict': {
+            'symbol': 'LTC/USD',
+            'asset_type': 'Crypto',
+            'country': 'US',
+            'quote_currency': 'USD',
+            'tenor': 'Spot',
+            'jlqd_symbol': 'LTCUSD',
+        },
+        'price': [
+            {
+                'account': pyth_init_price,
+                'price_exponent': -5,
+                'price_type': 'price',
+            },
+        ],
+    }
+    assert result == expected_result