def test_progress_reporter_no_test(self, mock_test, mock_request): """ Test progress_reporter with no test found. """ from mod_ci.controllers import progress_reporter mock_test.query.filter.return_value.first.return_value = None expected_ret = "FAIL" ret_val = progress_reporter(1, "token") self.assertEqual(expected_ret, ret_val) mock_test.query.filter.assert_called_once() mock_request.assert_not_called()
def test_progress_reporter_finish_type(self, mock_finish_type, mock_test, mock_request): """ Test progress_reporter with request type finish. """ from mod_ci.controllers import progress_reporter mock_test_obj = MagicMock() mock_test_obj.token = "token" mock_test.query.filter.return_value.first.return_value = mock_test_obj mock_request.form = {'type': 'finish'} mock_finish_type.return_value = "OK" expected_ret = "OK" ret_val = progress_reporter(1, "token") self.assertEqual(expected_ret, ret_val) mock_test.query.filter.assert_called_once() mock_request.assert_not_called() mock_finish_type.assert_called_once_with(mock.ANY, 1, mock.ANY, mock.ANY)
def test_progress_reporter_upload_type_empty(self, mock_upload_type, mock_test, mock_request): """ Test progress_reporter with request type upload with returning 'EMPTY'. """ from mod_ci.controllers import progress_reporter mock_test_obj = MagicMock() mock_test_obj.token = "token" mock_test.query.filter.return_value.first.return_value = mock_test_obj mock_request.form = {'type': 'upload'} mock_upload_type.return_value = False expected_ret = "EMPTY" ret_val = progress_reporter(1, "token") self.assertEqual(expected_ret, ret_val) mock_test.query.filter.assert_called_once() mock_request.assert_not_called() mock_upload_type.assert_called_once_with(mock.ANY, 1, mock.ANY, mock.ANY, mock.ANY)
def test_progress_reporter_progress_type_fail(self, mock_progress_type, mock_test, mock_request): """ Test progress_reporter with failing of request type progress. """ from mod_ci.controllers import progress_reporter mock_test_obj = MagicMock() mock_test_obj.token = "token" mock_test.query.filter.return_value.first.return_value = mock_test_obj mock_request.form = {'type': 'progress'} mock_progress_type.return_value = False expected_ret = "FAIL" ret_val = progress_reporter(1, "token") self.assertEqual(expected_ret, ret_val) mock_test.query.filter.assert_called_once() mock_request.assert_not_called() mock_progress_type.assert_called_once_with(mock.ANY, mock.ANY, 1, mock.ANY)