fix: child user API key test

Author: Justintime50

Makefile

@@ -27,7 +27,7 @@ clean:
 
 ## coverage - Test the project and generate an HTML coverage report
 coverage:
-	$(VIRTUAL_BIN)/pytest --cov=$(PROJECT_NAME) --cov-branch --cov-report=html --cov-report=term-missing --cov-fail-under=87
+	$(VIRTUAL_BIN)/pytest --cov=$(PROJECT_NAME) --cov-branch --cov-report=html --cov-report=term-missing --cov-fail-under=88
 
 ## docs - Generates docs for the library
 docs:

easypost/user.py

@@ -1,4 +1,6 @@
 from typing import (
+    Any,
+    Dict,
     List,
     Optional,
 )
@@ -48,21 +50,24 @@ def retrieve_me(cls, api_key: Optional[str] = None, **params) -> "User":
         return convert_to_easypost_object(response=response, api_key=api_key)
 
     @classmethod
-    def all_api_keys(cls, api_key: Optional[str] = None) -> "User":
-        """Get all API keys including child user keys."""
+    def all_api_keys(cls, api_key: Optional[str] = None) -> Dict[str, Any]:
+        """Retrieve a list of all API keys."""
         requestor = Requestor(local_api_key=api_key)
         url = "/api_keys"
         response, api_key = requestor.request(method=RequestMethod.GET, url=url)
         return convert_to_easypost_object(response=response, api_key=api_key)
 
-    def api_keys(self) -> List[str]:
-        """Get the authenticated user's API keys."""
+    def api_keys(self) -> List[Dict[str, Any]]:
+        """Retrieve a list of API keys (works for the authenticated user or a child user)."""
         api_keys = self.all_api_keys()
 
-        if api_keys.id == self.id:
-            return api_keys.keys
+        if api_keys["id"] == self.id:
+            # This function was called on the authenticated user
+            return api_keys["keys"]
         else:
-            for child in api_keys.children:
+            # This function was called on a child user (authenticated as parent, only return
+            # this child user's details).
+            for child in api_keys["children"]:
                 if child.id == self.id:
                     return child.keys
 

tests/cassettes/test_authenticated_user_api_keys.yaml

@@ -66,20 +66,33 @@ def test_user_delete(prod_api_key):
 
 @pytest.mark.vcr()
 def test_user_all_api_keys(prod_api_key):
+    """Tests that we can retrieve all API keys."""
     user = easypost.User.retrieve_me()
     api_keys = user.all_api_keys()
 
     assert api_keys["keys"] is not None
 
 
 @pytest.mark.vcr()
-def test_user_api_keys(prod_api_key):
+def test_authenticated_user_api_keys(prod_api_key):
+    """Tests that we can retrieve the authenticated user's API keys."""
     user = easypost.User.retrieve_me()
     api_keys = user.api_keys()
 
     assert api_keys is not None
 
 
+@pytest.mark.vcr()
+def test_child_user_api_keys(prod_api_key):
+    """Tests that we can retrieve a child user's API keys as the parent."""
+    user = easypost.User.create(name="Test User")
+    child_user = easypost.User.retrieve(user.id)
+
+    api_keys = child_user.api_keys()
+
+    assert api_keys is not None
+
+
 @pytest.mark.vcr()
 def test_user_update_brand(prod_api_key):
     user = easypost.User.retrieve_me()