Also define __iter__ for BaseAPIJsonObject

Author: Taejun Lee

boxsdk/object/base_api_json_object.py

@@ -106,6 +106,12 @@ def __contains__(self, item):
         """
         return item in self._response_object
 
+    def __iter__(self):
+        """
+        Get all of the keys of the API response object.
+        """
+        return iter(self._response_object)
+
     def __repr__(self):
         """Base class override. Return a human-readable representation using the Box ID or name of the object."""
         extra_description = ' - {0}'.format(self._description) if self._description else ''

test/unit/object/test_base_api_json_object.py

@@ -33,6 +33,13 @@ def test_contains(base_api_json_object):
     assert 'some_key_that_doesnt_exist' not in test_object
 
 
+def test_iter(base_api_json_object):
+    dictionary_response, test_object = base_api_json_object
+    all_test_object_keys = [key for key in test_object]
+    all_dictionary_response_keys = [key for key in dictionary_response]
+    assert set(all_test_object_keys) == set(all_dictionary_response_keys)
+
+
 def test_meta_registers_new_item_type_in_default_translator(default_translator, original_default_translator):
     item_type = u'ƒøø'