remove public key as params from getBalance and added negative testcases.

Author: unniznd

src/lighthouseweb3/__init__.py

@@ -45,15 +45,15 @@ def uploadBlob(self, source: io.BufferedReader, filename: str, tag: str = ''):
         except Exception as e:
             raise e
 
-    def getBalance(self, publicKey: str):
+    def getBalance(self):
         """
         Retrieve the balance information of a user from the Lighthouse.
 
         :param publicKey: str, The public key of the user.
         :return: dict[str, any], A dictionary containing the data usage and data limit details.
         """
         try:
-            return getBalance.get_balance(self.token, publicKey)
+            return getBalance.get_balance(self.token)
         except Exception as e:
             raise e
 

src/lighthouseweb3/functions/get_balance.py

@@ -1,11 +1,11 @@
 from .config import Config
 import requests as req
 
-def get_balance(token:str, publicKey: str):
+def get_balance(token:str):
   headers = {
         "Authorization": f"Bearer {token}",
     }
-  url = f"{Config.lighthouse_api}/api/user/user_data_usage?publicKey={publicKey}"
+  url = f"{Config.lighthouse_api}/api/user/user_data_usage"
   try:
     response = req.get(url, headers=headers)
   except Exception as e:

tests/test_get_balance.py

@@ -11,7 +11,16 @@ def test_get_balance(self):
         """test get_balance function"""
         parse_env()
         l = Lighthouse(os.environ.get("LIGHTHOUSE_TOKEN"))
-        res = l.getBalance("0x1FB9Be96d228De94F4C57962675433Ae55a6c4a5")
+        res = l.getBalance()
         self.assertIsInstance(res, dict, "data is a dict")
         self.assertIsInstance(res.get("dataLimit"), int, "data limit is a integer")
-        self.assertIsInstance(res.get("dataUsed"), int, "data used is a integer")
+        self.assertIsInstance(res.get("dataUsed"), int, "data used is a integer")
+    
+    def test_get_balance_invalid_token(self):
+        """test get_balance function with invalid token"""
+        parse_env()
+        l = Lighthouse('invalid_token')
+        res = l.getBalance()
+        self.assertIsInstance(res, dict, "res is a dict")
+        self.assertIn("authentication failed", str(res).lower())
+