feat: Add method to retrieve savings account balance

Signed-off-by: Rodrigo Arena Câmara <[email protected]>
andreroggeri · Sep 9, 2022 · d733a7a · d733a7a
1 parent 22f457b
commit d733a7a
Show file tree

Hide file tree

Showing 5 changed files with 70 additions and 2 deletions.
diff --git a/pynubank/nubank.py b/pynubank/nubank.py
@@ -211,6 +211,11 @@ def get_account_balance(self):
         data = self._make_graphql_request('account_balance')
         return data['data']['viewer']['savingsAccount']['currentSavingsBalance']['netAmount']
 
+    @requires_auth_mode(AuthMode.APP)
+    def get_account_savings_balance(self):
+        data = self._make_graphql_request('account_savings')
+        return data['data']['viewer']['productFeatures']['buckets']['screens']['home']
+
     @requires_auth_mode(AuthMode.APP)
     def get_account_investments_details(self):
         data = self._make_graphql_request('account_investments')

diff --git a/pynubank/queries/account_savings.gql b/pynubank/queries/account_savings.gql
@@ -0,0 +1,21 @@
+query bucketsHomeInfo {
+  viewer {
+    productFeatures {
+      buckets {
+        screens {
+          home {
+            mainBalance {
+              netAmount
+            }
+            balancesList {
+              label
+              maturityDate
+              netAmount
+              href
+            }
+          }
+        }
+      }
+    }
+  }
+}
diff --git a/pynubank/utils/mock_http.py b/pynubank/utils/mock_http.py
@@ -46,6 +46,8 @@ def __init__(self):
             'pix_receipt_screen')
         self._results[GHOSTFLAME_URL, str(prepare_request_body('account_feed_paginated'))] = self._read_data(
             'account_feed_paginated')
+        self._results[(GHOSTFLAME_URL, str(prepare_request_body('account_savings')))] = self._read_data(
+            'account_savings')
 
     def add_mock_url(self, url: str, graphql_object: str, response_json_name: str):
         self._results[(url, graphql_object)] = self._read_data(response_json_name)

diff --git a/pynubank/utils/mocked_responses/account_savings.json b/pynubank/utils/mocked_responses/account_savings.json
@@ -0,0 +1,31 @@
+{
+   "data": {
+      "viewer": {
+         "productFeatures": {
+            "buckets": {
+               "screens": {
+                  "home": {
+                     "mainBalance": {
+                        "netAmount": 37.0
+                     },
+                     "balancesList": [
+                        {
+                           "label": "Disponível a qualquer momento",
+                           "maturityDate": "None",
+                           "netAmount": 35.0,
+                           "href": "nuapp://flutter/savings/buckets/daily-liquidity"
+                        },
+						{
+                           "label": "Disponível em 01/07/2023",
+                           "maturityDate": "None",
+                           "netAmount": 35.0,
+                           "href": "nuapp://flutter/savings/buckets/daily-liquidity"
+                        }
+                     ]
+                  }
+               }
+            }
+         }
+      }
+   }
+}
diff --git a/tests/test_nubank_client.py b/tests/test_nubank_client.py
@@ -520,8 +520,17 @@ def test_creditcard_methods_should_allow_web_authentication(method_name, method_
     method(**method_args)
 
 
-def test_creditcard_methods_should_allow_app_authentication():
-    pass
+def test_get_account_savings_balance():
+    nubank_client = Nubank(client=MockHttpClient())
+    nubank_client.authenticate_with_cert('1234', 'hunter12', 'some-file.p12')
+
+    savings_balance = nubank_client.get_account_savings_balance()
+    savings_balance_list = savings_balance['balancesList']
+
+    assert savings_balance['mainBalance']['netAmount'] == 37.0
+    assert len(savings_balance_list) == 2
+    for balance in savings_balance_list:
+        assert balance['netAmount'] == 35.0
 
 
 def test_get_pix_details():