def test_generate_preference(self, mock_get_image, mock_list_images, mock_fit): """ Test that a prefered placeholder image is chosen and generated correctly. """ mock_list_images.return_value = ['source/02.jpg', ] mock_fit.return_value = MagicMock() form = PlaceholderForm({'height': '400', 'width': '400'}) form.is_valid() form.generate(preference='02') # Called with the right "random" image? mock_get_image.assert_called_with('source/02.jpg') # Called with the right dimensions to fit? self.assertTrue((400, 400) in mock_fit.call_args[0])
def test_generate_random(self, mock_get_image, mock_list_images, mock_fit, mock_choice): """ Test that a random choice placeholder image is chosen and generated correctly. """ mock_list_images.return_value = ['source/01.jpg', ] mock_choice.return_value = 'source/01.jpg' mock_fit.return_value = MagicMock() form = PlaceholderForm({'height': '300', 'width': '200'}) form.is_valid() form.generate() # Called with the right "random" image? mock_get_image.assert_called_with('source/01.jpg') # Called with the right dimensions to fit? self.assertTrue((300, 200) in mock_fit.call_args[0])