feat(auth): added property decorators for token and headers

src/nbiatoolkit/auth.py

@@ -146,3 +146,27 @@ def getToken(self) -> Union[dict, int]:
             self.scope = token_data.get('scope')
 
             return self.api_headers
+
+    @property
+    def token(self):
+        """
+        Returns the access token.
+
+        Returns
+        -------
+        access_token : str or None
+            The access token retrieved from the API.
+        """
+        return self.access_token
+
+    @property
+    def headers(self):
+        """
+        Returns the API headers.
+
+        Returns
+        -------
+        api_headers : dict or None
+            The authentication headers containing the access token.
+        """
+        return self.api_headers

tests/test_auth.py

@@ -21,6 +21,7 @@ def failed_oauth2():
 
 def test_getToken(oauth2):
     assert oauth2.access_token is not None
+    assert oauth2.token is not None
 
 def test_expiry(oauth2):
     # expiry should be in the form of :'Tue Jun 29 13:58:57 2077'
@@ -34,8 +35,9 @@ def test_failed_oauth(failed_oauth2):
         failed_oauth2.getToken()
         assert failed_oauth2.getToken() == 401
     assert failed_oauth2.access_token == -1
+    assert failed_oauth2.token == -1
     assert failed_oauth2.getToken() == 401
-    # self.api_headers
+    assert failed_oauth2.headers is None
     assert failed_oauth2.api_headers is None
     assert failed_oauth2.expiry_time is None
     assert failed_oauth2.refresh_token is None
@@ -47,7 +49,10 @@ def test_getToken_valid_token(oauth2):
     # Test if the access token is valid and not expired
     assert oauth2.getToken() == oauth2.access_token
     assert oauth2.getToken() != 401
+    assert oauth2.access_token != -1
+    assert oauth2.token != -1
     assert oauth2.api_headers is not None
+    assert oauth2.headers is not None
     assert oauth2.expiry_time is not None
     assert oauth2.refresh_token is not None
     assert oauth2.refresh_expiry is not None