2 files changed, 30 insertions, 5 deletions
diff --git a/github3/tests/base_test.py b/github3/tests/base_test.py
new file mode 100644
index 0000000..59d474a
--- /dev/null
+++ b/github3/tests/base_test.py
@@ -0,0 +1,29 @@
+#!/usr/bin/env python
+# -*- encoding: utf-8 -*-
+
+from unittest import TestCase
+import github3
+from github3 import api
+from github3 import handlers
+
+
+class TestGetHandlers(TestCase):
+
+    def setUp(self):
+        self.anom_gh = api.Github()
+        self.auth_gh = api.Github('test', 'password')
+
+    def test_get_user(self):
+        anom_user = self.anom_gh.users
+        auth_user = self.auth_gh.users
+
+        self.assertEquals(isinstance(anom_user, handlers.users.User), True)
+        self.assertEquals(isinstance(auth_user, handlers.users.AuthUser), True)
+
+    def test_get_gists(self):
+        anom_gists = self.anom_gh.gists
+        auth_gists = self.auth_gh.gists
+
+        self.assertEquals(isinstance(anom_gists, handlers.gists.Gist), True)
+        self.assertEquals(
+            isinstance(auth_gists, handlers.gists.AuthGist), True)
diff --git a/tests/gist_tests.py b/github3/tests/gist_tests.py
index 7e31066..1218772 100644
--- a/tests/gist_tests.py
+++ b/github3/tests/gist_tests.py
@@ -23,7 +23,7 @@ class GistsTestCase(unittest.TestCase):
         g = github3.api.Github()
         g.session.auth = ('testuser', 'password')
         u = github3.handlers.user.AuthUser(g)
-        gists = github3.handlers.gists.Gist(g)
+        gists = github3.handlers.gists.AuthGist(g)
         OpenerDirector = MagicMock(name='OpenerDirector')
         opener = OpenerDirector.return_value
         response = opener.open.return_value
@@ -96,7 +96,3 @@ class GistHandlerTestCase(unittest.TestCase):
                          u'deadbeefdeadbeefdeadbeefdeadbeefdeadbeef')
         self.assertEqual(h.user.__dict__, gist.user.__dict__)
         self.assertEqual(h.version, u'deadbeefdeadbeefdeadbeefdeadbeefdeadbeef')
-
-
-if __name__ == '__main__':
-    unittest.main()