changes for get account when exist proxy (#78)

* changes for get account when exist proxy

* add tests

* conver dict into Account

* dict comprrenhetion

Author: gregcastro

arcus/client.py

@@ -106,9 +106,17 @@ def request(
 
     @property
     def accounts(self) -> Dict[str, Account]:
-        accounts_ = dict(primary=Account(**self.get('/account')))
-        if self.topup_key:
-            accounts_['topup'] = Account(**self.get('/account', topup=True))
+        if self.proxy:
+            accounts_dict = self.get('/account')
+            accounts_ = {
+                key: Account(**val) for key, val in accounts_dict.items()
+            }
+        else:
+            accounts_ = dict(primary=Account(**self.get('/account')))
+            if self.topup_key:
+                accounts_['topup'] = Account(
+                    **self.get('/account', topup=True)
+                )
         return accounts_
 
     @staticmethod

arcus/version.py

@@ -1 +1 @@
-__version__ = '1.2.1'  # pragma: no cover
+__version__ = '1.2.2'  # pragma: no cover

tests/conftest.py

@@ -18,10 +18,18 @@ def client_proxy():
         m.get(
             f'{proxy}/account',
             json=dict(
-                name='Cuenca',
-                balance=60454.43,
-                minimum_balance=0.0,
-                currency='MXN',
+                primary=dict(
+                    name='cuenca',
+                    balance=63869.33,
+                    minimum_balance=0.0,
+                    currency='MXN',
+                ),
+                topup=dict(
+                    name='cuenca-tae',
+                    balance=69720.0,
+                    minimum_balance=0.0,
+                    currency='MXN',
+                ),
             ),
         )
         m.post(

tests/resources/test_accounts.py

@@ -19,6 +19,16 @@ def test_get_account_info(client):
         assert account.balance > account.minimum_balance
 
 
+@pytest.mark.vcr
+def test_get_topup_account_info(client_proxy):
+    accounts = client_proxy.get('/account', topup=True)
+    for account in accounts.values():
+        assert type(account) is dict
+        assert account['currency'] == 'MXN'
+        assert type(account['balance']) is float
+        assert account['balance'] > account['minimum_balance']
+
+
 def test_get_account_info_proxy(client_proxy):
     accounts = client_proxy.accounts
     assert type(accounts) is dict