Added converters test

copitux · copitux · commit b2890225a70d · 2011-11-12T13:13:28.000+01:00
diff --git a/github3/core.py b/github3/core.py
@@ -29,7 +29,6 @@ def _last_page(self, link):
 
         return self.last
 
-    # TODO: reset iterators... multiple?
     def __iter__(self):
         return self
 
diff --git a/github3/handlers/base.py b/github3/handlers/base.py
@@ -53,9 +53,8 @@ def _get_resources(self, resource, model=None, limit=None, **kwargs):
         """ Hander request to multiple resources """
 
         resource = self._prefix_resource(resource)
-        page_resources = Paginate(resource, self._gh.get, **kwargs)
         counter = 1
-        for page in page_resources:
+        for page in Paginate(resource, self._gh.get, **kwargs):
             for raw_resource in page:
                 if limit and counter > limit: break
                 counter += 1
diff --git a/github3/tests/gist_tests.py b/github3/tests/gist_tests.py
diff --git a/github3/tests/test_converters.py b/github3/tests/test_converters.py
@@ -0,0 +1,91 @@
+#!/usr/bin/env python
+# -*- encoding: utf-8 -*-
+
+from github3.converters import *
+from github3.models.base import BaseResource
+from unittest import TestCase
+from datetime import datetime
+
+API_STUB = {
+    'test_str': 'string', 'test_int': 1,
+    'test_date': '2008-01-14T04:33:35Z', 'test_bool': True,
+    'map': {'test_str': 'string'},
+    'dict_map': {
+        'map1': {
+            'test_str': 'string',
+            'test_int': 1
+        },
+        'map2': {
+            'test_str': 'string',
+            'test_int': 2
+        },
+    },
+    'list_map': [
+        {'test_str': 'string', 'test_int': 1},
+        {'test_str': 'string', 'test_int': 2},
+    ]
+}
+
+
+class Model(BaseResource):
+
+    @classmethod
+    def idl(self):
+        return {
+            'strs': ['test_str'],
+            'ints': ['test_int'],
+            'dates': ['test_date'],
+            'bools': ['test_bool'],
+            'maps': {'map': Model},
+            'collection_maps': {
+                'dict_map': Model,
+                'list_map': Model,
+            },
+        }
+
+
+class TestModelizer(TestCase):
+
+    def setUp(self):
+        model = Model
+        self.modelizer = Modelizer()
+        self.modelizer.inject(model)
+
+    def test_loads(self):
+        parsed_model = self.modelizer.loads(API_STUB)
+        self.assertEquals(parsed_model.test_str, 'string')
+        self.assertEquals(parsed_model.test_int, 1)
+        self.assertEquals(
+            parsed_model.test_date,
+            datetime(2008, 1, 14, 4, 33, 35))
+        self.assertTrue(parsed_model.test_bool)
+        self.assertTrue(isinstance(parsed_model.map, Model))
+        self.assertEquals(parsed_model.map.test_str, 'string')
+        self.assertTrue(isinstance(parsed_model.dict_map, dict))
+        map1 = parsed_model.dict_map['map1']
+        map2 = parsed_model.dict_map['map2']
+        self.assertTrue(isinstance(map1, Model))
+        self.assertTrue(isinstance(map2, Model))
+        self.assertEquals(map1.test_str, 'string')
+        self.assertEquals(map1.test_int, 1)
+        self.assertEquals(map2.test_str, 'string')
+        self.assertEquals(map2.test_int, 2)
+
+        list_map = parsed_model.list_map
+        self.assertTrue(isinstance(list_map, list))
+        self.assertEquals(list_map[0].test_str, 'string')
+        self.assertEquals(list_map[0].test_int, 1)
+        self.assertEquals(list_map[1].test_str, 'string')
+        self.assertEquals(list_map[1].test_int, 2)
+
+
+class TestRawlizer(TestCase):
+
+    def setUp(self):
+        model = Model
+        self.rawlizer = Rawlizer()
+
+    #  Trivial, I know it
+    def test_loads(self):
+        raw = self.rawlizer.loads(API_STUB)
+        self.assertEquals(raw, API_STUB)
