Fix-up Issue Event usage of Users

sigmavirus24 · sigmavirus24 · commit 361824d9b89e · 2017-02-01T06:46:13.000-06:00
diff --git a/github3/issues/event.py b/github3/issues/event.py
@@ -1,8 +1,8 @@
 # -*- coding: utf-8 -*-
 from __future__ import unicode_literals
 
+from .. import users
 from ..models import GitHubCore
-from ..users import User
 
 
 class IssueEvent(GitHubCore):
@@ -37,7 +37,9 @@ def _update_attributes(self, event):
         self.issue = self._class_attribute(event, 'issue', Issue, self)
 
         #: :class:`User <github3.users.User>` who caused this event.
-        self.actor = self._class_attribute(event, 'actor', User, self)
+        self.actor = self._class_attribute(
+            event, 'actor', users.ShortUser, self,
+        )
 
         #: Number of comments
         self.comments = self._get_attribute(event, 'comments')
@@ -55,7 +57,9 @@ def _update_attributes(self, event):
         self.id = self._get_attribute(event, 'id')
 
         #: :class:`User <github3.users.User>` that is assigned
-        self.assignee = self._class_attribute(event, 'assignee', User, self)
+        self.assignee = self._class_attribute(
+            event, 'assignee', users.ShortUser, self,
+        )
 
         #: Dictionary containing milestone details
         self.milestone = self._get_attribute(event, 'milestone', {})
diff --git a/tests/integration/test_issue.py b/tests/integration/test_issue.py
@@ -146,7 +146,7 @@ def test_events(self):
             issue = repository.issue(218)
             for event in issue.events():
                 assert isinstance(event, github3.issues.event.IssueEvent)
-                assert isinstance(event.actor, github3.users.User)
+                assert isinstance(event.actor, github3.users.ShortUser)
 
     def test_labels(self):
         """Test the ability to iterate over issue labels."""
