예제 #1
0
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'
예제 #2
0
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'