Support to map self in resources.

copitux · copitux · commit 869e2abf99e5 · 2012-02-12T12:44:06.000+01:00
Very nested resources like Repo
diff --git a/pygithub3/resources/base.py b/pygithub3/resources/base.py
@@ -48,6 +48,8 @@ def parse_date(string_date):
             return date
 
         def parse_map(resource, raw_resource):
+            if resource == 'self':
+                return self.__load(raw_resource)
             if hasattr(raw_resource, 'items'):
                 return resource.__load(raw_resource)
 
diff --git a/pygithub3/tests/resources/test_core.py b/pygithub3/tests/resources/test_core.py
@@ -4,7 +4,7 @@
 from unittest import TestCase
 from datetime import datetime
 
-from pygithub3.resources.base import Resource, Raw, json
+from pygithub3.resources.base import Raw
 from pygithub3.tests.utils.resources import Nested, Simple, HasSimple
 
 simple_resource = dict(type='simple')
@@ -16,7 +16,9 @@
     simple=simple_resource,
     list_collection=[has_simple] * 2,
     items_collections=dict(arg1=has_simple, arg2=has_simple)
-    )
+)
+github_return_nested = github_return.copy()
+github_return.update({'self_nested': github_return_nested})
 
 
 class TestResourceMapping(TestCase):
@@ -45,6 +47,11 @@ def test_DICT_collection_map(self):
         self.assertEqual(arg1_has_simple.type, 'has_simple')
         self.assertEqual(arg1_has_simple.simple.type, 'simple')
 
+    def test_SELF_nested(self):
+        self.assertIsInstance(self.r.self_nested, Nested)
+        self.assertIsInstance(self.r.self_nested.simple, Simple)
+        self.assertIsInstance(self.r.list_collection[0], HasSimple)
+        self.assertIsInstance(self.r.items_collections['arg1'], HasSimple)
 
 class TestRawResource(TestCase):
     """ Litle obvious :P """
diff --git a/pygithub3/tests/utils/resources.py b/pygithub3/tests/utils/resources.py
@@ -14,7 +14,7 @@ class HasSimple(Resource):
 
 class Nested(Resource):
     _dates = ('date', )
-    _maps = {'simple': Simple}
+    _maps = {'simple': Simple, 'self_nested': 'self'}
     _collection_maps = {
         'list_collection': HasSimple,
         'items_collections': HasSimple
