Create unit tests for locking and unlocking an issue

adrianmoisey · adrianmoisey · commit ac76823c27f8 · 2016-02-16T11:02:57.000+02:00
diff --git a/tests/unit/test_issues_issue.py b/tests/unit/test_issues_issue.py
@@ -66,6 +66,11 @@ def test_edit_comment(self):
         with pytest.raises(github3.AuthenticationFailed):
             self.instance.edit()
 
+    def test_lock(self):
+        """Verify that locking an issue requires authentication."""
+        with pytest.raises(github3.AuthenticationFailed):
+            self.instance.lock()
+
     def test_remove_all_labels(self):
         """Verify that removing all labels requires authentication."""
         with pytest.raises(github3.AuthenticationFailed):
@@ -81,6 +86,11 @@ def test_reopen(self):
         with pytest.raises(github3.AuthenticationFailed):
             self.instance.reopen()
 
+    def test_unlock(self):
+        """Verify that unlocking an issue requires authentication."""
+        with pytest.raises(github3.AuthenticationFailed):
+            self.instance.unlock()
+
 
 class TestIssue(helper.UnitHelper):
 
@@ -156,6 +166,15 @@ def test_create_comment_required_body(self):
         self.instance.create_comment(body='')
         assert self.session.post.called is False
 
+    def test_create_lock(self):
+        """Verify the request for removing a lock from an issue."""
+        self.instance.lock()
+
+        self.session.put.assert_called_once_with(
+            url_for('lock'),
+            headers={'Accept': 'application/vnd.github.the-key-preview+json'}
+        )
+
     def test_comment_positive_id(self):
         """Verify the request for retrieving an issue comment."""
         self.instance.comment(-1)
@@ -267,6 +286,15 @@ def test_remove_label(self):
             url_for('labels/enhancement')
         )
 
+    def test_remove_lock(self):
+        """Verify the request for removing a lock from an issue."""
+        self.instance.unlock()
+
+        self.session.delete.assert_called_once_with(
+            url_for('lock'),
+            headers={'Accept': 'application/vnd.github.the-key-preview+json'}
+        )
+
     def test_reopen(self):
         """Test the request for reopening an issue."""
         labels = [str(label) for label in self.instance.original_labels]
