Update references to mock in Issues tests

sigmavirus24 · sigmavirus24 · commit d381ed80cefa · 2014-05-04T13:57:11.000-05:00
diff --git a/tests/test_issues.py b/tests/test_issues.py
@@ -5,11 +5,7 @@
 from github3.issues.milestone import Milestone
 from github3.issues import Issue
 import datetime
-from tests.utils import BaseCase, load
-try:
-   from unittest.mock import patch
-except ImportError:
-   from mock import patch
+from tests.utils import BaseCase, load, mock
 
 
 class TestLabel(BaseCase):
@@ -171,7 +167,7 @@ def test_assign(self):
 
         self.login()
 
-        with patch.object(Issue, 'edit') as ed:
+        with mock.patch.object(Issue, 'edit') as ed:
             ed.return_value = True
             assert self.i.assign(None) is False
             self.not_called()
@@ -189,7 +185,7 @@ def test_close(self):
         self.not_called()
         self.login()
 
-        with patch.object(Issue, 'edit') as ed:
+        with mock.patch.object(Issue, 'edit') as ed:
             ed.return_value = True
             assert self.i.close()
             u = self.i.assignee.login if self.i.assignee else ''
@@ -277,7 +273,7 @@ def test_remove_all_labels(self):
 
         self.login()
 
-        with patch.object(Issue, 'replace_labels') as rl:
+        with mock.patch.object(Issue, 'replace_labels') as rl:
             rl.return_value = []
             assert self.i.remove_all_labels() == []
             rl.assert_called_once_with([])
@@ -303,7 +299,7 @@ def test_reopen(self):
         n = self.i.milestone.number if self.i.milestone else None
         u = self.i.assignee.login if self.i.assignee else None
 
-        with patch.object(Issue, 'edit') as ed:
+        with mock.patch.object(Issue, 'edit') as ed:
             ed.return_value = True
             assert self.i.reopen()
             labels = [str(l) for l in self.i.labels]
