Fixed balance test, added pairs

README.md

@@ -28,6 +28,7 @@ Exchange original API docs: [https://exchange-docs.crypto.com](https://exchange-
 
 ### Changelog
 
+- **0.5** - missing symbols MKR, UNI, possible refactoring for simple objects
 - **0.4.5** - fixed missing CELR balances
 - **0.4.4** - fixed missing QTUM, CELR coins
 - **0.4.3** - fixed missing `fees_coin` Coin enum

src/cryptocom/exchange/__init__.py

@@ -13,4 +13,4 @@ __all__ = [
     'ApiError', 'ApiProvider'
 ]
 
-__version__ = '0.4.5'
+__version__ = '0.5'

src/cryptocom/exchange/private.py

@@ -68,7 +68,7 @@ class Account:
 
     async def get_trades(
             self, pair: Pair, page: int = 0,
-            page_size: int = 200)  -> List[PrivateTrade]:
+            page_size: int = 200) -> List[PrivateTrade]:
         """Return trades."""
         data = await self.api.post('private/get-trades', {
             'params': {

src/cryptocom/exchange/structs.py

@@ -34,6 +34,8 @@ class Coin(str, enum.Enum):
     OMG = 'OMG'
     QTUM = 'QTUM'
     CELR = 'CELR'
+    MKR = 'MKR'
+    UNI = 'UNI'
 
     ETC = 'ETC'
 
@@ -91,6 +93,8 @@ class Pair(str, enum.Enum):
     MANA_USDT = 'MANA_USDT'
     QTUM_USDT = 'QTUM_USDT'
     CELR_USDT = 'CELR_USDT'
+    MKR_USDT = 'MKR_USDT'
+    UNI_USDT = 'UNI_USDT'
 
     MCO_CRO = 'MCO_CRO'
     ETH_CRO = 'ETH_CRO'
@@ -117,6 +121,8 @@ class Pair(str, enum.Enum):
     MANA_CRO = 'MANA_CRO'
     QTUM_CRO = 'QTUM_CRO'
     CELR_CRO = 'CELR_CRO'
+    MKR_CRO = 'MKR_CRO'
+    UNI_CRO = 'UNI_CRO'
 
     CRO_USDC = 'CRO_USDC'
 

tests/test_private.py

@@ -8,8 +8,8 @@ import cryptocom.exchange as cro
 @pytest.mark.asyncio
 async def test_account_get_balance(account: cro.Account):
     balances = await account.get_balance()
-    assert balances[cro.Coin.CRO].available > 2
-    assert balances[cro.Coin.CRO].available > 2
+    assert balances[cro.Coin.CRO].available > 0.5
+    assert balances[cro.Coin.USDT].available > 0.5
     for coin in cro.Coin:
         assert coin.value in balances