def test_download_sample_raise_exception(self, mock_sample,
mock_serve_download):
"""Test function download_sample to raise SampleNotFoundException."""
from mod_sample.controllers import (SampleNotFoundException,
download_sample)
mock_sample.query.filter.return_value.first.return_value = None
with self.assertRaises(SampleNotFoundException):
download_sample(1)
mock_sample.query.filter.assert_called_once_with(mock_sample.id == 1)
mock_serve_download.assert_not_called()
def test_download_sample(self, mock_sample, mock_serve_download):
"""Test function download_sample."""
from mod_sample.controllers import download_sample
response = download_sample(1)
self.assertEqual(response, mock_serve_download())
mock_sample.query.filter.assert_called_once_with(mock_sample.id == 1)
