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)