Adding optional parameter for token-info endpoint (#256)

Allow token_info parameter to pass in an account_id so the call to GET /accounts doesn't throw 401. This is how the actual endpoint behaves.

---------

Co-authored-by: Kiran Raju <[email protected]@Kirans-MacBook-Pro-2.local>
Co-authored-by: Mostafa Rashed <[email protected]>

Author: 3 people

nylas/client/client.py

@@ -289,10 +289,16 @@ def ip_addresses(self):
         _validate(resp).json()
         return resp.json()
 
-    def token_info(self):
-        token_info_url = self.token_info_url.format(
-            client_id=self.client_id, account_id=self.account.id
-        )
+    def token_info(self, account_id=None):
+        token_info_url = ""
+        if account_id is not None:
+            token_info_url = self.token_info_url.format(
+                client_id=self.client_id, account_id=account_id
+            )
+        else:
+            token_info_url = self.token_info_url.format(
+                client_id=self.client_id, account_id=self.account.id
+            )
         headers = {"Content-Type": "application/json"}
         headers.update(self.admin_session.headers)
         resp = self.admin_session.post(

tests/conftest.py

@@ -1955,9 +1955,7 @@ def mock_ip_addresses(mocked_responses, api_url, client_id):
 
 @pytest.fixture
 def mock_token_info(mocked_responses, api_url, account_id, client_id):
-    token_info_url = "{base}/a/{client_id}/accounts/{id}/token-info".format(
-        base=api_url, id=account_id, client_id=client_id
-    )
+    token_info_url = re.compile(api_url + "/a/.*/accounts/.*/token-info")
     mocked_responses.add(
         responses.POST,
         token_info_url,

tests/test_accounts.py

@@ -38,6 +38,16 @@ def test_token_info(api_client_with_client_id):
     assert "scopes" in result
 
 
+@pytest.mark.usefixtures("mock_token_info", "mock_account")
+def test_token_info_with_account_id(api_client_with_client_id):
+    result = api_client_with_client_id.token_info(
+        account_id="anvkhwelkfdoehdflhdjkfhe1"
+    )
+    assert isinstance(result, dict)
+    assert "updated_at" in result
+    assert "scopes" in result
+
+
 @pytest.mark.usefixtures("mock_account")
 def test_account_datetime(api_client):
     account = api_client.account