add test for solana account

Author: cctdaniel

tests/test_solana_account.py

@@ -0,0 +1,89 @@
+import pytest
+
+from mock import AsyncMock
+
+
+from pythclient.solana import SolanaAccount, SolanaPublicKey, SolanaClient
+
+
+@pytest.fixture
+def solana_pubkey():
+    return SolanaPublicKey("AHtgzX45WTKfkPG53L6WYhGEXwQkN1BVknET3sVsLL8J")
+
+
+@pytest.fixture
+def solana_account(solana_pubkey, solana_client):
+    return SolanaAccount(
+        key=solana_pubkey,
+        solana=solana_client,
+    )
+
+@pytest.fixture()
+def mock_get_account_info(mocker):
+    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, solana_client
+):
+    actual = SolanaAccount(
+        key=solana_pubkey,
+        solana=solana_client,
+    )
+    assert actual.slot == None
+    assert actual.lamports == 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, solana_pubkey, solana_client):
+    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, caplog, solana_pubkey, solana_client):
+    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, caplog, solana_pubkey, solana_client):
+    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}; 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):
+    actual = str(solana_account)
+    expected = f"SolanaAccount ({solana_account.key})"
+    assert actual == expected