Repos.forks service done

copitux · copitux · commit 36381d426f5d · 2012-02-22T01:16:32.000+01:00
diff --git a/pygithub3/requests/repos/forks.py b/pygithub3/requests/repos/forks.py
@@ -0,0 +1,18 @@
+#!/usr/bin/env python
+# -*- encoding: utf-8 -*-
+
+from . import Request
+
+from pygithub3.resources.repos import Repo
+
+
+class List(Request):
+
+    uri = 'repos/{user}/{repo}/forks'
+    resource = Repo
+
+
+class Create(Request):
+
+    uri = '/repos/{user}/{repo}/forks'
+    resource = Repo
diff --git a/pygithub3/resources/repos.py b/pygithub3/resources/repos.py
@@ -106,6 +106,9 @@ def __str__(self):
 
 class Download(Resource):
 
+    def __str__(self):
+        return '<Download (%s)>' % getattr(self, 'name', '')
+
     def ball_to_upload(self):
         return OrderedDict({
             'key': self.path, 'acl': self.acl, 'success_action_status': '201',
diff --git a/pygithub3/services/repos.py b/pygithub3/services/repos.py
@@ -6,6 +6,18 @@
 from .base import Service, MimeTypeMixin
 
 
+class Forks(Service):
+
+    def list(self, user=None, repo=None, sort='newest'):
+        request = self.make_request('repos.forks.list', user=user, repo=repo)
+        return self._get_result(request, sort=sort)
+
+    def create(self, user=None, repo=None, org=None):
+        request = self.make_request('repos.forks.create', user=user, repo=repo)
+        org = {'org': org} if org else {}
+        return self._post(request, **org)
+
+
 class Downloads(Service):
 
     def list(self, user=None, repo=None):
@@ -111,6 +123,7 @@ def __init__(self, **config):
         self.collaborators = Collaborator(**config)
         self.commits = Commits(**config)
         self.downloads = Downloads(**config)
+        self.forks = Forks(**config)
         super(Repo, self).__init__(**config)
 
     def list(self, user=None, type='all'):
diff --git a/pygithub3/tests/services/test_repos.py b/pygithub3/tests/services/test_repos.py
@@ -6,7 +6,8 @@
 import requests
 from mock import patch, Mock
 
-from pygithub3.services.repos import Repo, Collaborator, Commits, Downloads
+from pygithub3.services.repos import (Repo, Collaborator, Commits, Downloads,
+                                      Forks)
 from pygithub3.resources.base import json
 from pygithub3.tests.utils.base import (mock_response, mock_response_result,
                                         mock_json)
@@ -268,3 +269,22 @@ def test_CREATE(self, request_method):
         self.assertEqual(request_method.call_args[0],
                          ('post', _('repos/oct/re_oct/downloads')))
         self.assertTrue(hasattr(download, 'upload'))
+
+
+@patch.object(requests.sessions.Session, 'request')
+class TestForksService(TestCase):
+
+    def setUp(self):
+        self.fs = Forks(user='oct', repo='re_oct')
+
+    def test_LIST(self, request_method):
+        request_method.return_value = mock_response_result()
+        self.fs.list().all()
+        self.assertEqual(request_method.call_args[0],
+                         ('get', _('repos/oct/re_oct/forks')))
+
+    def test_CREATE(self, request_method):
+        request_method.return_value = mock_response('post')
+        self.fs.create()
+        self.assertEqual(request_method.call_args[0],
+                         ('post', _('repos/oct/re_oct/forks')))
