Added more debugging for bug-fixing

example/get_tokens.py

@@ -40,6 +40,7 @@
         password=GOOGLE_PASSWORD,
         master_token=GOOGLE_MASTER_TOKEN,
         android_id=DEVICE_ID,
+        verbose=True,
     )
 
     # Get master token

glocaltokens/client.py

@@ -28,10 +28,7 @@
 from .utils import token as token_utils
 from .utils.logs import censor
 
-DEBUG = False
-
-logging_level = logging.DEBUG if DEBUG else logging.ERROR
-logging.basicConfig(level=logging_level)
+logging.basicConfig(level=logging.ERROR)
 LOGGER = logging.getLogger(__name__)
 
 
@@ -123,6 +120,7 @@ def __init__(
         password: Optional[str] = None,
         master_token: Optional[str] = None,
         android_id: Optional[str] = None,
+        verbose: Optional[bool] = False,
     ):
         """
         Initialize an GLocalAuthenticationTokens instance with google account
@@ -137,6 +135,9 @@ def __init__(
                 if not set;
 
         """
+        if verbose:
+            LOGGER.setLevel(logging.DEBUG)
+
         LOGGER.debug("Initializing new GLocalAuthenticationTokens instance.")
         self.username: Optional[str] = username
         self.password: Optional[str] = password

glocaltokens/utils/logs.py

@@ -1,8 +1,10 @@
-def censor(text: str, replace: str = "*") -> str:
+def censor(text: str, char: str = "*") -> str:
     """
     Replaces all the characters in a str by the specified [replace]
 
     text: The text to censure.
     replace: The character to instead of the content.
     """
-    return replace * len(text)
+    text = text if text else ""
+    censored_text = text[0] + (len(text) - 1) * char if len(text) else text
+    return censored_text

tests/test_client.py

@@ -1,4 +1,5 @@
 import json
+import logging
 from datetime import datetime, timedelta
 from unittest import TestCase
 
@@ -80,6 +81,18 @@ def test_initialization__valid(self, m_log):
         GLocalAuthenticationTokens(master_token=faker.master_token())
         self.assertEqual(m_log.call_count, 0)
 
+    @patch("glocaltokens.client.LOGGER.setLevel")
+    def test_initialization__valid_verbose_logger(self, m_set_level):
+        # Non verbose
+        GLocalAuthenticationTokens(username=faker.word(), password=faker.word())
+        self.assertEqual(m_set_level.call_count, 0)
+
+        # Verbose
+        GLocalAuthenticationTokens(
+            username=faker.word(), password=faker.word(), verbose=True
+        )
+        m_set_level.assert_called_once_with(logging.DEBUG)
+
     @patch("glocaltokens.client.LOGGER.error")
     def test_initialization__invalid(self, m_log):
         # Without username

tests/test_utils.py

@@ -17,5 +17,24 @@ def tearDown(self):
         pass
 
     def test_logs(self):
+        # With word
         secret_string = faker.word()
-        self.assertNotEqual(secret_string, censor(secret_string))
+        censored_string = censor(secret_string)
+        self.assertNotEqual(secret_string, censored_string)
+        self.assertTrue(censored_string.startswith(secret_string[0]))
+        self.assertEqual(
+            censored_string, f"{secret_string[0]}{(len(secret_string)-1)*'*'}"
+        )
+
+        # With different censor character
+        secret_string = faker.word()
+        censored_string = censor(secret_string, "&")
+        self.assertNotEqual(secret_string, censored_string)
+        self.assertTrue(censored_string.startswith(secret_string[0]))
+        self.assertEqual(
+            censored_string, f"{secret_string[0]}{(len(secret_string)-1)*'&'}"
+        )
+
+        # With empty string
+        censored_string = censor("")
+        self.assertEqual(censored_string, "")