Unit tests added to run tests after GitHub action success

Author: Tarun-Arora

tests/test_ci/TestControllers.py

@@ -840,6 +840,59 @@ def test_webhook_with_invalid_ci_signature(self, mock_github, mock_warning):
                 data=json.dumps({}), headers=self.generate_header({}, 'workflow_run', "1"))
         mock_warning.assert_called_once()
 
+    @mock.patch('mod_ci.controllers.BlockedUsers')
+    @mock.patch('mod_ci.controllers.queue_test')
+    @mock.patch('requests.get', side_effect=mock_api_request_github)
+    def test_webhook_workflow_run_event_completed_action_successful_pull_request(self, mock_request,
+                                                                                 mock_queue_test, mock_blocked):
+        """Test webhook triggered with workflow run event with action completed and status success for pull request."""
+        data = {'action': 'completed',
+                'workflow_run': {'event': 'pull_request',
+                                 'name': 'Build CCExtractor on Linux', 'head_sha': '1',
+                                 'head_branch': 'master'}, 'sender': {'login': 'test_owner'}}
+        fakedata = {'workflow_runs': [
+            {'head_sha': '1', 'status': 'completed',
+             'conclusion': 'success', 'name': Workflow_builds.LINUX},
+            {'head_sha': '1', 'status': 'completed',
+             'conclusion': 'success', 'name': Workflow_builds.WINDOWS}
+        ]}
+        pull_requests = [{'head': {'sha': '1'}, 'user': {'id': 1}, 'number': '1'}]
+
+        class MockedRepository:
+            def statuses(self, *args):
+                class gh_status:
+                    def post(*args, **kwargs):
+                        return None
+                return gh_status
+
+            class actions:
+                class runs:
+                    def get(*args, **kwargs):
+                        return fakedata
+
+            class pulls:
+                def get(*args, **kwargs):
+                    return pull_requests
+
+        class MockedGitHub:
+            def repos(self, *args):
+                return MockedRepository
+        with self.app.test_client() as c:
+            from github import GitHub
+            GitHub.repos = Mock(return_value=MockedGitHub.repos)
+            mock_blocked.query.filter.return_value.first.return_value = None
+            response = c.post(
+                '/start-ci', environ_overrides=WSGI_ENVIRONMENT,
+                data=json.dumps(data), headers=self.generate_header(data, 'workflow_run'))
+            mock_queue_test.assert_called_once()
+            mock_queue_test.reset_mock()
+            mock_blocked.query.filter.return_value.first.return_value = 1
+            response = c.post(
+                '/start-ci', environ_overrides=WSGI_ENVIRONMENT,
+                data=json.dumps(data), headers=self.generate_header(data, 'workflow_run'))
+            mock_queue_test.assert_not_called()
+            self.assertEqual(response.data, b'ERROR')
+
     def test_start_ci_with_a_get_request(self):
         """Test start_ci function with a request method other than post."""
         from mod_ci.controllers import start_ci
@@ -903,6 +956,18 @@ def test_webhook_issue_opened(self, mock_issue, mock_requests, mock_mailing):
         mock_issue.query.filter(mock_issue.issue_id == '1234')
         mock_mailing.assert_called_once_with(mock.ANY, '1234', 'testTitle', 'testAuthor', 'testing')
 
+    @mock.patch('run.log.critical')
+    def test_github_api_error(self, mock_critical):
+        """Test."""
+        from github import GitHub
+
+        from mod_ci.controllers import deschedule_test, schedule_test
+        schedule_test(GitHub('1').repos('1')('1').statuses('1'), 1, None)
+        mock_critical.assert_called()
+        mock_critical.reset_mock()
+        deschedule_test(GitHub('1').repos('1')('1').statuses('1'), 1, None)
+        mock_critical.assert_called()
+
     @mock.patch('mod_ci.controllers.is_main_repo')
     @mock.patch('mod_ci.controllers.shutil')
     def test_update_build_badge(self, mock_shutil, mock_check_repo):