starting to fill in issues service; starting with GET for issues and milestones.

mpdaugherty · mpdaugherty · commit 044010426dcd · 2012-05-17T08:03:01.000+08:00
diff --git a/pygithub3/requests/issues/__init__.py b/pygithub3/requests/issues/__init__.py
@@ -0,0 +1,9 @@
+# -*- encoding: utf-8 -*-
+
+from pygithub3.requests.base import Request, ValidationError
+from pygithub3.resources.issues import Issue
+
+class List(Request):
+
+    uri = 'issues'
+    resource = Issue
diff --git a/pygithub3/requests/issues/milestones.py b/pygithub3/requests/issues/milestones.py
@@ -0,0 +1,15 @@
+# -*- encoding: utf-8 -*-
+
+from pygithub3.requests.base import Request
+from pygithub3.resources.issues import Milestone
+
+class List(Request):
+
+    uri = 'issues/{user}/{repo}/milestones'
+    resource = Milestone
+
+
+class Get(Request):
+
+    uri = 'issues/{user}/{repo}/milestones/{number}'
+    resource = Milestone
diff --git a/pygithub3/resources/issues.py b/pygithub3/resources/issues.py
@@ -0,0 +1,23 @@
+#!/usr/bin/env python
+# -*- encoding: utf-8 -*-
+
+from .base import Resource
+from .users import User
+
+
+class Issue(Resource):
+
+    _dates = ('created_at', 'updated_at', )
+    _maps = {'pull_request': ..., 'milestone': Milestone, 'assignee': User, 'user': User}
+    _collection_maps = {'labels': ..., }
+
+    def __str__(self):
+        return '<Issue (%s)>' % getattr(self, 'title', '')
+
+class Milestone(Resource):
+
+    _dates = ('created_at', )
+    _maps = {'creator': User}
+
+    def __str__(self):
+        return '<Milestone (%s)>' % getattr(self, 'description', '')
diff --git a/pygithub3/services/issues/__init__.py b/pygithub3/services/issues/__init__.py
@@ -0,0 +1,28 @@
+#!/usr/bin/env python
+# -*- encoding: utf-8 -*-
+
+from pygithub3.services.base import Service
+from milestones import Milestones
+
+
+class Issue(Service):
+    """ Consume `Issues API <http://developer.github.com/v3/issues>`_ """
+
+    def __init__(self, **config):
+        self.milestones = Milestones(**config)
+        super(Issue, self).__init__(**config)
+
+    def list(self):
+        """ Get authenticated user's issues
+
+        :returns: A :doc:`result`
+
+        If you call it and you are authenticated, get the
+        authenticated user's gists.
+
+        ::
+
+            gist_service.list()
+        """
+        request = self.request_builder('issues.list')
+        return self._get_result(request)
diff --git a/pygithub3/services/issues/milestones.py b/pygithub3/services/issues/milestones.py
@@ -0,0 +1,30 @@
+#!/usr/bin/env python
+# -*- encoding: utf-8 -*-
+
+from pygithub3.services.base import Service
+
+
+class Milestones(Service):
+    """ Consume `Milestones API
+    <http://developer.github.com/v3/issues/milestones>`_
+    """
+
+    def list(self, user=None, repo=None):
+        """ Get a repository's issues
+
+        :param str user: Username
+        :param str repo: Repo Name
+        :returns: A :doc:`result`
+        """
+        request = self.request_builder('issues.milestones.list', user=user, repo=repo)
+        return self._get_result(request, **self._get_mimetype_as_header())
+
+    def get(self, number, user=None, repo=None):
+        """ Get a single milestone
+
+        :param str user: Username
+        :param str repo: Repo Name
+        :param int number: Milestone number
+        """
+        request = self.request_builder('issues.milestones.get', user=user, repo=repo, number=number)
+        return self._get(request)
diff --git a/pygithub3/tests/services/test_issues.py b/pygithub3/tests/services/test_issues.py
@@ -0,0 +1,49 @@
+#!/usr/bin/env python
+# -*- encoding: utf-8 -*-
+
+import requests
+from mock import patch, Mock
+
+from pygithub3.tests.utils.core import TestCase
+from pygithub3.resources.base import json
+from pygithub3.services.issues import Issue, Milestones
+from pygithub3.tests.utils.base import (mock_response, mock_response_result,
+                                        mock_json)
+from pygithub3.tests.utils.services import _
+
+json.dumps = Mock(side_effect=mock_json)
+json.loads = Mock(side_effect=mock_json)
+
+@patch.object(requests.sessions.Session, 'request')
+class TestIssueService(TestCase):
+
+    def setUp(self):
+        self.issues = Issue()
+
+    def test_LIST_without_user(self, request_method):
+        request_method.return_value = mock_response_result()
+        self.issues.list().all()
+        self.assertEqual(request_method.call_args[0],
+            ('get', _('issues')))
+
+@patch.object(requests.sessions.Session, 'request')
+class TestMilestonesService(TestCase):
+    def setUp(self):
+        self.ms = Milestones()
+
+    def test_LIST_with_user_and_repo(self, request_method):
+        request_method.return_value = mock_response_result()
+        self.ms.list(user='octocat', repo='Hello-World').all()
+        self.assertEqual(request_method.call_args[0],
+            ('get', _('repos/octocat/Hello-World/milestones')))
+
+    def test_LIST_with_state(self, request_method):
+        # Fail for now; I don't know how to do this test yet
+        self.ms.abc()
+
+    def test_GET(self, request_method):
+
+        request_method.return_value = mock_response_result()
+        self.ms.get(user='octocat', repo='Hello-World', number=1)
+        self.assertEqual(request_method.call_args[0],
+            ('get', _('repos/octocat/Hello-World/milestones/1')))
