From 15dbb0a8f43a003c6e27569c204dbfdc2a6dd22a Mon Sep 17 00:00:00 2001
From: Morty Space <morty.space@gmail.com>
Date: Sun, 27 Sep 2020 13:09:39 +0300
Subject: [PATCH] Fixed balance test, added pairs

---
 README.md                          | 1 +
 src/cryptocom/exchange/__init__.py | 2 +-
 src/cryptocom/exchange/private.py  | 2 +-
 src/cryptocom/exchange/structs.py  | 6 ++++++
 tests/test_private.py              | 4 ++--
 5 files changed, 11 insertions(+), 4 deletions(-)

diff --git a/README.md b/README.md
index 9a422fe..25b922b 100644
--- a/README.md
+++ b/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
diff --git a/src/cryptocom/exchange/__init__.py b/src/cryptocom/exchange/__init__.py
index 4cd56d2..2950d92 100644
--- a/src/cryptocom/exchange/__init__.py
+++ b/src/cryptocom/exchange/__init__.py
@@ -13,4 +13,4 @@ __all__ = [
     'ApiError', 'ApiProvider'
 ]
 
-__version__ = '0.4.5'
+__version__ = '0.5'
diff --git a/src/cryptocom/exchange/private.py b/src/cryptocom/exchange/private.py
index ecf9750..6bd8b0d 100644
--- a/src/cryptocom/exchange/private.py
+++ b/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': {
diff --git a/src/cryptocom/exchange/structs.py b/src/cryptocom/exchange/structs.py
index 253b900..c822717 100644
--- a/src/cryptocom/exchange/structs.py
+++ b/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'
 
diff --git a/tests/test_private.py b/tests/test_private.py
index 5d4f15e..8b63646 100644
--- a/tests/test_private.py
+++ b/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