Add additional starred_at attribute for repository

itsmemattchung · itsmemattchung · commit de9853356ffd · 2015-11-29T10:12:23.000Z
This is only retrieved on github3.users.User.starred_repositories call
diff --git a/github3/repos/repo.py b/github3/repos/repo.py
@@ -59,6 +59,10 @@ class Repository(GitHubCore):
 
     """
 
+    STAR_HEADERS = {
+        'Accept': 'application/vnd.github.v3.star+json'
+    }
+
     def _update_attributes(self, repo):
         #: URL used to clone via HTTPS.
         self.clone_url = repo.get('clone_url', '')
@@ -142,6 +146,9 @@ def _update_attributes(self, repo):
         #: Number of users who starred the repository
         self.stargazers_count = repo.get('stargazers_count', 0)
 
+        #: ``datetime`` object representing when the repository was starred
+        self.starred_at = self._strptime(repo.get('starred_at'))
+
         # SSH url e.g. git@github.com/sigmavirus24/github3.py
         #: URL to clone the repository via SSH.
         self.ssh_url = repo.get('ssh_url', '')
diff --git a/github3/users.py b/github3/users.py
@@ -387,7 +387,8 @@ def starred_repositories(self, sort=None, direction=None, number=-1,
         params = {'sort': sort, 'direction': direction}
         self._remove_none(params)
         url = self.starred_urlt.expand(owner=None, repo=None)
-        return self._iter(int(number), url, Repository, params, etag)
+        return self._iter(int(number), url, Repository, params, etag,
+                          headers=Repository.STAR_HEADERS)
 
     def subscriptions(self, number=-1, etag=None):
         """Iterate over repositories subscribed to by this user.
diff --git a/tests/cassettes/User_starred_repositories.json b/tests/cassettes/User_starred_repositories.json
diff --git a/tests/integration/test_users.py b/tests/integration/test_users.py
@@ -1,5 +1,6 @@
 """Integration tests for the User class."""
 import github3
+import datetime
 
 from .helper import IntegrationHelper
 
@@ -93,6 +94,7 @@ def test_starred_repositories(self):
         assert len(repos) > 0
         for starred in repos:
             assert isinstance(starred, github3.repos.Repository)
+            assert isinstance(starred.starred_at, datetime.datetime)
 
     def test_subscriptions(self):
         """Show that a user can retrieve the repos subscribed to by a user."""
