def test_check_product_availability_and_warn(monkeypatch, checkout, product): variant = product.variants.get() add_variant_to_checkout(checkout, variant) monkeypatch.setattr('django.contrib.messages.warning', Mock(warning=Mock())) monkeypatch.setattr('saleor.checkout.utils.contains_unavailable_variants', Mock(return_value=False)) utils.check_product_availability_and_warn(MagicMock(), checkout) assert len(checkout) == 1 monkeypatch.setattr('saleor.checkout.utils.contains_unavailable_variants', Mock(return_value=True)) monkeypatch.setattr( 'saleor.checkout.utils.remove_unavailable_variants', lambda c: add_variant_to_checkout(checkout, variant, 0, replace=True)) utils.check_product_availability_and_warn(MagicMock(), checkout) assert len(checkout) == 0
def test_check_product_availability_and_warn(monkeypatch, checkout, product): variant = product.variants.get() add_variant_to_checkout(checkout, variant) monkeypatch.setattr("django.contrib.messages.warning", Mock(warning=Mock())) monkeypatch.setattr( "saleor.checkout.utils.contains_unavailable_variants", Mock(return_value=False) ) utils.check_product_availability_and_warn(MagicMock(), checkout) assert len(checkout) == 1 monkeypatch.setattr( "saleor.checkout.utils.contains_unavailable_variants", Mock(return_value=True) ) monkeypatch.setattr( "saleor.checkout.utils.remove_unavailable_variants", lambda c: add_variant_to_checkout(checkout, variant, 0, replace=True), ) utils.check_product_availability_and_warn(MagicMock(), checkout) assert len(checkout) == 0