def test_download_result_file_not_found(self, mock_regression_output):
"""Test that non-existent result file gives 404."""
from mod_regression.controllers import test_result_file
mock_regression_output.query.filter.return_value.first.return_value = None
with self.assertRaises(NotFound):
test_result_file(1)
mock_regression_output.query.filter.assert_called_once_with(
mock_regression_output.id == 1)
def test_download_result_file(self, mock_regression_output, mock_serve):
"""Test that correct result file triggers serve download."""
from mod_regression.controllers import test_result_file
response = test_result_file(1)
mock_regression_output.query.filter.assert_called_once_with(
mock_regression_output.id == 1)
mock_serve.assert_called_once()