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)
