Merge pull request #5 from cctdaniel/daniel-tests

add test for SolanaAccount

Author: Jeff Schroeder

setup.py

@@ -10,7 +10,8 @@
     author_email='contact@pyth.network',
     install_requires=requirements,
     extras_require={
-        'testing': requirements + ['mock', 'pytest', 'pytest-cov', 'pytest-socket'],
+        'testing': requirements + ['mock', 'pytest', 'pytest-cov', 'pytest-socket',
+                                   'pytest-mock', 'pytest-asyncio'],
     },
     python_requires='>=3.7.0',
 )

tests/conftest.py

@@ -1,5 +1,8 @@
+import logging
 import pytest
+from _pytest.logging import caplog as _caplog
 from pythclient.solana import SolanaClient
+from loguru import logger
 
 
 @pytest.fixture
@@ -8,3 +11,16 @@ def solana_client():
         endpoint="https://example.com",
         ws_endpoint="wss://example.com",
     )
+
+
+@pytest.fixture
+def caplog(_caplog):
+    logger.enable("pythclient")
+
+    class PropogateHandler(logging.Handler):
+        def emit(self, record):
+            logging.getLogger(record.name).handle(record)
+
+    handler_id = logger.add(PropogateHandler(), format="{message}")
+    yield _caplog
+    logger.remove(handler_id)

tests/test_solana_account.py

@@ -0,0 +1,102 @@
+from _pytest.logging import LogCaptureFixture
+import pytest
+
+from pytest_mock import MockerFixture
+
+from mock import AsyncMock
+
+
+from pythclient.solana import SolanaAccount, SolanaPublicKey, SolanaClient
+
+
+@pytest.fixture
+def solana_pubkey() -> SolanaPublicKey:
+    return SolanaPublicKey("AHtgzX45WTKfkPG53L6WYhGEXwQkN1BVknET3sVsLL8J")
+
+
+@pytest.fixture
+def solana_account(solana_pubkey: SolanaPublicKey, solana_client: SolanaClient) -> SolanaAccount:
+    return SolanaAccount(
+        key=solana_pubkey,
+        solana=solana_client,
+    )
+
+
+@pytest.fixture()
+def mock_get_account_info(mocker: MockerFixture) -> AsyncMock:
+    async_mock = AsyncMock()
+    mocker.patch('pythclient.solana.SolanaClient.get_account_info', side_effect=async_mock)
+    return async_mock
+
+
+def test_solana_account_update_with_rpc_response(
+    solana_pubkey: SolanaPublicKey, solana_client: SolanaClient
+) -> None:
+    actual = SolanaAccount(
+        key=solana_pubkey,
+        solana=solana_client,
+    )
+    assert actual.slot is None
+    assert actual.lamports is None
+
+    slot = 106498726
+    value = {
+        "lamports": 1000000000
+    }
+
+    actual.update_with_rpc_response(slot=slot, value=value)
+
+    assert actual.slot == slot
+    assert actual.lamports == value["lamports"]
+
+
+@pytest.mark.asyncio
+async def test_solana_account_update_success(mock_get_account_info: AsyncMock,
+                                             solana_pubkey: SolanaPublicKey, solana_client: SolanaClient) -> None:
+    actual = SolanaAccount(
+        key=solana_pubkey,
+        solana=solana_client,
+    )
+
+    mock_get_account_info.return_value = {'context': {'slot': 93752509}, 'value': {'lamports': 1000000001}}
+
+    await actual.update()
+    assert actual.slot == mock_get_account_info.return_value['context']['slot']
+    assert actual.lamports == mock_get_account_info.return_value['value']['lamports']
+
+
+@pytest.mark.asyncio
+async def test_solana_account_update_fail(mock_get_account_info: AsyncMock,
+                                          caplog: LogCaptureFixture,
+                                          solana_pubkey: SolanaPublicKey,
+                                          solana_client: SolanaClient) -> None:
+    actual = SolanaAccount(
+        key=solana_pubkey,
+        solana=solana_client,
+    )
+    mock_get_account_info.return_value = {'value': {'lamports': 1000000001}}
+    exc_message = f'error while updating account {solana_pubkey}'
+    await actual.update()
+    assert exc_message in caplog.text
+
+
+@pytest.mark.asyncio
+async def test_solana_account_update_null(mock_get_account_info: AsyncMock,
+                                          caplog: LogCaptureFixture,
+                                          solana_pubkey: SolanaPublicKey,
+                                          solana_client: SolanaClient) -> None:
+    actual = SolanaAccount(
+        key=solana_pubkey,
+        solana=solana_client,
+    )
+    mock_get_account_info.return_value = {'context': {'slot': 93752509}}
+    exc_message = f'got null value from Solana getAccountInfo for {solana_pubkey}; ' \
+        + f'non-existent account? {mock_get_account_info.return_value}'
+    await actual.update()
+    assert exc_message in caplog.text
+
+
+def test_solana_account_str(solana_account: SolanaAccount) -> None:
+    actual = str(solana_account)
+    expected = f"SolanaAccount ({solana_account.key})"
+    assert actual == expected