Update references to mock in Organization tests

sigmavirus24 · sigmavirus24 · commit 1886cf3af73c · 2014-05-04T13:57:26.000-05:00
diff --git a/tests/test_orgs.py b/tests/test_orgs.py
@@ -1,9 +1,5 @@
 import github3
-try:
-   from unittest.mock import patch, Mock
-except ImportError:
-   from mock import patch, Mock
-from tests.utils import BaseCase, load
+from tests.utils import BaseCase, load, mock
 
 
 class TestTeam(BaseCase):
@@ -149,10 +145,10 @@ def test_add_member(self):
         self.assertRaises(github3.GitHubError, self.org.add_member, None, None)
 
         self.login()
-        with patch.object(github3.orgs.Organization, 'iter_teams') as it:
+        with mock.patch.object(github3.orgs.Organization, 'iter_teams') as it:
             it.return_value = iter([])
             assert self.org.add_member('foo', 'bar') is False
-            team = Mock()
+            team = mock.Mock()
             team.name = 'bar'
             team.add_member.return_value = True
             it.return_value = iter([team])
@@ -163,10 +159,10 @@ def test_add_repo(self):
         self.assertRaises(github3.GitHubError, self.org.add_repo, None, None)
 
         self.login()
-        with patch.object(github3.orgs.Organization, 'iter_teams') as it:
+        with mock.patch.object(github3.orgs.Organization, 'iter_teams') as it:
             it.return_value = iter([])
             assert self.org.add_repo('foo', 'bar') is False
-            team = Mock()
+            team = mock.Mock()
             team.name = 'bar'
             team.add_repo.return_value = True
             it.return_value = iter([team])
@@ -343,10 +339,10 @@ def test_remove_repo(self):
                           None, None)
 
         self.login()
-        with patch.object(github3.orgs.Organization, 'iter_teams') as it:
+        with mock.patch.object(github3.orgs.Organization, 'iter_teams') as it:
             it.return_value = iter([])
             assert self.org.remove_repo('foo', 'bar') is False
-            team = Mock()
+            team = mock.Mock()
             team.name = 'bar'
             team.remove_repo.return_value = True
             it.return_value = iter([team])
