def test_product_first_image(): mock_product_image = Mock() mock_product_image.image = Mock() mock_product_image.image.crop = {'10x10': Mock(url='crop.jpg')} mock_queryset = Mock() mock_queryset.all.return_value = [mock_product_image] mock_product = Mock(images=mock_queryset) out = product_first_image(mock_product, '10x10', method='crop') assert out == 'crop.jpg'