tests.test_tradehub_get_balance

Complexity

Total Complexity

2

Size/Duplication

Total Lines	62
Duplicated Lines	0 %

Importance

Changes

0

from tests import APITestCase, mainnet_client


class TestTradeHubGetBalance(APITestCase):

    def setUp(self) -> None:
        self._client = mainnet_client

    def test_get_balance_structure(self):
        """
        Check if response match expected dict structure.
        :return:
        """
        expect: dict = {
            "cel1": {
                "available": str,
                "order": str,
                "position": str,
                "denom": str
            },
            "eth1": {
                "available": str,
                "order": str,
                "position": str,
                "denom": str
            },
            "nex1": {
                "available": str,
                "order": str,
                "position": str,
                "denom": str
            },
            "nneo2": {
                "available": str,
                "order": str,
                "position": str,
                "denom": str
            },
            "swth": {
                "available": str,
                "order": str,
                "position": str,
                "denom": str
            },
            "usdc1": {
                "available": str,
                "order": str,
                "position": str,
                "denom": str
            },
            "wbtc1": {
                "available": str,
                "order": str,
                "position": str,
                "denom": str
            }
        }

        result: dict = self._client.get_balance("swth1vwges9p847l9csj8ehrlgzajhmt4fcq4sd7gzl")
        # if this may fail, check if all denoms are returned. Non zero balances are not returned
        self.assertDictStructure(expect, result)