Add some more tests to convert

sigmavirus24 · sigmavirus24 · commit 724b4856be58 · 2015-04-11T16:30:00.000-05:00
diff --git a/tests/integration/test_repos_repo.py b/tests/integration/test_repos_repo.py
@@ -11,6 +11,16 @@ class TestRepository(helper.IntegrationHelper):
 
     """Integration tests for the Repository object."""
 
+    @pytest.skip('Need internet to record the cassette')
+    def test_add_collaborator(self):
+        """Test the ability to add a collaborator to a repository."""
+        self.basic_login()
+        cassette_name = self.cassette_name('add_collaborator')
+        with self.recorder.use_cassette(cassette_name):
+            repository = self.gh.repository('github3py', 'fork_this')
+            assert repository
+            assert repository.add_collaborator('esacteksab')
+
     def test_assignees(self):
         """Test the ability to retrieve assignees of issues on a repo."""
         cassette_name = self.cassette_name('assignees')
diff --git a/tests/unit/test_repos_repo.py b/tests/unit/test_repos_repo.py
@@ -3,6 +3,7 @@
 import pytest
 
 from github3 import GitHubError
+from github3.null import NullObject
 from github3.repos.repo import Repository
 
 from .helper import (UnitHelper, UnitIteratorHelper, create_url_helper)
@@ -136,6 +137,21 @@ class TestRepository(UnitHelper):
     described_class = Repository
     example_data = repo_example_data
 
+    def test_add_collaborator(self):
+        """Verify the request to add a collaborator to a repository."""
+        self.instance.add_collaborator('sigmavirus24')
+
+        self.session.put.assert_called_once_with(
+            url_for('collaborators/sigmavirus24')
+        )
+
+    def test_add_null_collaborator(self):
+        """Verify no request is made when adding `None` as a collaborator."""
+        self.instance.add_collaborator(None)
+        self.instance.add_collaborator(NullObject())
+
+        assert self.session.put.called is False
+
     def test_asset(self):
         """Test retrieving an asset uses the right headers.
 
@@ -687,6 +703,11 @@ def after_setup(self):
         """Set-up the session to not be authenticated."""
         self.session.has_auth.return_value = False
 
+    def test_add_collaborator(self):
+        """Verify that adding a collaborator requires authentication."""
+        with pytest.raises(GitHubError):
+            self.instance.add_collaborator('foo')
+
     def test_hooks(self):
         """Show that a user must be authenticated to list hooks."""
         with pytest.raises(GitHubError):
