test publish

Author: jhunsaker

pyth/tests/conftest.py

@@ -303,10 +303,28 @@ def pyth_add_price(solana_test_validator, pyth_dir, pyth_init_product):
     return result
 
 
-@pytest.fixture(scope='function')
-def pyth_init_price(solana_test_validator, pyth_dir, pyth_add_price):
+@pytest.fixture(scope='session')
+def pyth_add_publisher(
+    solana_test_validator, solana_keygen, pyth_dir, pyth_add_price
+):
 
     for product, key in pyth_add_price.items():
+        cmd = [
+            'pyth_admin', 'add_publisher',
+            solana_keygen[0], key,
+            '-r', 'localhost',
+            '-k', pyth_dir,
+            '-c', 'finalized',
+            '-n',
+        ]
+        check_call(cmd)
+    return pyth_add_price
+
+
+@pytest.fixture(scope='function')
+def pyth_init_price(solana_test_validator, pyth_dir, pyth_add_publisher):
+
+    for product, key in pyth_add_publisher.items():
         cmd = [
             'pyth_admin', 'init_price',
             key, '-e', '-5',
@@ -316,7 +334,7 @@ def pyth_init_price(solana_test_validator, pyth_dir, pyth_add_price):
             '-n',
         ]
         check_call(cmd)
-    return pyth_add_price
+    return pyth_add_publisher
 
 
 @pytest.fixture(scope='session')

pyth/tests/test_publish.py

@@ -0,0 +1,56 @@
+import json
+import time
+from subprocess import check_call, check_output
+
+
+def test_publish(solana_test_validator, pyth_dir,
+                 pyth_init_product, pyth_init_price):
+
+    def get_price_acct():
+        cmd = [
+            'pyth', 'get_product',
+            pyth_init_product['LTC'],  # TODO get_price method
+            '-r', 'localhost',
+            '-k', pyth_dir,
+            '-c', 'finalized',
+            '-j',
+        ]
+        output = check_output(cmd)
+        output = output.decode('ascii')
+        output = json.loads(output)
+        return output['price_accounts'][0]
+
+    before = get_price_acct()
+    assert before['publisher_accounts'][0]['price'] == 0
+    assert before['publisher_accounts'][0]['conf'] == 0
+    assert before['publisher_accounts'][0]['status'] == 'unknown'
+
+    cmd = [
+        'pyth', 'upd_price_val',
+        pyth_init_price['LTC'],
+        '150', '10', 'trading',
+        '-r', 'localhost',
+        '-k', pyth_dir,
+        '-c', 'finalized',
+        '-x',
+    ]
+    check_call(cmd)
+
+    time.sleep(20)
+
+    cmd = [
+        'pyth', 'upd_price',
+        pyth_init_price['LTC'],
+        '-r', 'localhost',
+        '-k', pyth_dir,
+        '-c', 'finalized',
+        '-x',
+    ]
+    check_call(cmd)
+
+    time.sleep(20)
+
+    after = get_price_acct()
+    assert after['publisher_accounts'][0]['price'] == 150
+    assert after['publisher_accounts'][0]['conf'] == 10
+    assert after['publisher_accounts'][0]['status'] == 'trading'