1 files changed, 102 insertions, 0 deletions
diff --git a/tests/gist_tests.py b/tests/gist_tests.py
new file mode 100644
index 0000000..7e31066
--- /dev/null
+++ b/tests/gist_tests.py
@@ -0,0 +1,102 @@
+#!/usr/bin/env python
+# -*- encoding: utf-8 -*-
+#
+# author: Antti Kaihola
+
+from datetime import datetime
+import json
+from mock import MagicMock, Mock, patch
+import unittest
+
+import github3.api
+import github3.handlers.gists
+import github3.handlers.user
+import github3.models
+
+
+GIST_RESPONSE = '{"user":{"gravatar_id":"123","url":"https://api.github.com/users/testuser","avatar_url":"https://secure.gravatar.com/avatar/123?d=https://a248.e.akamai.net/assets.github.com%2Fimages%2Fgravatars%2Fgravatar-1.png","login":"testuser","id":12345},"url":"https://api.github.com/gists/791920","history":[{"version":"deadbeefdeadbeefdeadbeefdeadbeefdeadbeef","url":"https://api.github.com/gists/791920/deadbeefdeadbeefdeadbeefdeadbeefdeadbeef","user":{"gravatar_id":"123","url":"https://api.github.com/users/testuser","avatar_url":"https://secure.gravatar.com/avatar/123?d=https://a248.e.akamai.net/assets.github.com%2Fimages%2Fgravatars%2Fgravatar-1.png","login":"testuser","id":12345},"committed_at":"2011-11-09T08:50:53Z","change_status":{"deletions":0,"additions":1,"total":1}}],"description":"description","created_at":"2011-11-09T08:50:53Z","public":true,"comments":0,"updated_at":"2011-11-09T08:50:53Z","git_pull_url":"git://gist.github.com/791920.git","forks":[],"git_push_url":"git@gist.github.com:791920.git","html_url":"https://gist.github.com/791920","id":"791920","files":{"filename.ext":{"raw_url":"https://gist.github.com/raw/791920/badafadacadafadabadacadafadabadabadacada/filename.ext","type":"text/plain","content":"content","size":7,"language":null,"filename":"filename.ext"}}}'
+
+
+class GistsTestCase(unittest.TestCase):
+    def test_create_gist(self):
+        """The HTTP request for creating a gist is correct"""
+        g = github3.api.Github()
+        g.session.auth = ('testuser', 'password')
+        u = github3.handlers.user.AuthUser(g)
+        gists = github3.handlers.gists.Gist(g)
+        OpenerDirector = MagicMock(name='OpenerDirector')
+        opener = OpenerDirector.return_value
+        response = opener.open.return_value
+        response.read.return_value = GIST_RESPONSE
+        response.code = 201
+        
+        with patch('urllib2.OpenerDirector', OpenerDirector):
+
+            gist = gists.create_gist(
+                'description',
+                files={'filename.ext': {'content': 'content'}})
+
+        request = opener.open.call_args[0][0]
+        self.assertEqual(request.method, 'POST')
+        self.assertEqual(request.get_full_url(),
+                         'https://api.github.com/gists?per_page=100')
+        self.assertEqual(request.headers['Authorization'],
+                         'Basic dGVzdHVzZXI6cGFzc3dvcmQ=')
+        self.assertEqual(json.loads(request.data),
+                         {u'description': u'description',
+                          u'files': {u'filename.ext': {u'content': u'content'}},
+                          u'public': True})
+
+
+class GistHandlerTestCase(unittest.TestCase):
+    def test_response_conversion(self):
+        """A gist response is decoded correctly to a Gist object"""
+        g = github3.api.Github()
+        handler = github3.handlers.gists.Gist(g)
+        converter = handler._get_converter()
+        converter.inject(github3.models.Gist)
+
+        gist = converter.loads(json.loads(GIST_RESPONSE))
+
+        self.assertEqual(
+            {filename: value.__dict__
+             for filename, value in gist.files.iteritems()},
+            {u'filename.ext': {
+                'content': u'content',
+                'filename': u'filename.ext',
+                'raw_url': (u'https://gist.github.com/'
+                            u'raw/791920/'
+                            u'badafadacadafadabadacadafadabadabadacada/'
+                            u'filename.ext'),
+                'size': 7,
+                'type': u'text/plain'}})
+        self.assertEqual(gist.description, u'description')
+        self.assertEqual(gist.url, u'https://api.github.com/gists/791920')
+        self.assertEqual(gist.created_at, datetime(2011, 11, 9, 8, 50, 53))
+        self.assertEqual(gist.html_url, u'https://gist.github.com/791920')
+        self.assertEqual(gist.public, True)
+        self.assertEqual(
+            gist.user.__dict__,
+            {'avatar_url': (u'https://secure.gravatar.com/avatar/123'
+                            u'?d=https://a248.e.akamai.net/'
+                            u'assets.github.com%2Fimages%2Fgravatars'
+                            u'%2Fgravatar-1.png'),
+             'id': 12345,
+             'login': u'testuser',
+             'url': u'https://api.github.com/users/testuser'})
+        self.assertEqual(gist.git_pull_url, u'git://gist.github.com/791920.git')
+        self.assertEqual(gist.git_push_url, u'git@gist.github.com:791920.git')
+        self.assertEqual(gist.id, u'791920')
+        self.assertEqual(len(gist.history), 1)
+        h = gist.history[0]
+        self.assertEqual(h.change_status.__dict__, {'additions': 1, 'total': 1})
+        self.assertEqual(h.committed_at, datetime(2011, 11, 9, 8, 50, 53))
+        self.assertEqual(h.url,
+                         u'https://api.github.com/gists/791920/'
+                         u'deadbeefdeadbeefdeadbeefdeadbeefdeadbeef')
+        self.assertEqual(h.user.__dict__, gist.user.__dict__)
+        self.assertEqual(h.version, u'deadbeefdeadbeefdeadbeefdeadbeefdeadbeef')
+
+
+if __name__ == '__main__':
+    unittest.main()