def test_add_and_remove_and_clear():
product = create_product('fractionable', fractional=True)
complete_product(product)
supplier = get_default_supplier()
request = get_request_with_basket()
basket = request.basket
with pytest.raises(ValidationError):
# Ordering antimatter is not supported
basket_commands.handle_add(request, basket, product_id=product.pk, quantity=-3)
# These will get merged into one line...
basket_commands.handle_add(request, basket, **{"product_id": product.pk, "quantity": 1, "supplier_id": supplier.pk})
basket_commands.handle_add(request, basket, **{"product_id": product.pk, "quantity": 2})
# Fractions should also be supported
basket_commands.handle_add(request, basket, **{"product_id": product.pk, "quantity": 0.75})
# ... so there will be 3 products but one line
assert basket.product_count == 3.75
lines = basket.get_lines()
assert len(lines) == 1
# ... and deleting that line will clear the basket...
basket_commands.handle_del(request, basket, lines[0].line_id)
assert basket.product_count == 0
# ... and adding another product will create a new line...
basket_commands.handle_add(request, basket, product_id=product.pk, quantity=1)
assert basket.product_count == 1
# ... that can be cleared.
basket_commands.handle_clear(request, basket)
assert basket.product_count == 0
def test_basket_partial_quantity_update():
request = get_request_with_basket()
basket = request.basket
product = get_default_product()
sales_unit = SalesUnit.objects.create(identifier="test-sales-partial", decimals=2, name="Partial unit")
product.sales_unit = sales_unit # Set the sales unit for the product
product.save()
basket_commands.handle_add(request, basket, product_id=product.pk, quantity=1.5)
assert basket.product_count == 1.5
line_id = basket.get_lines()[0].line_id
basket_commands.handle_update(request, basket, **{"q_%s" % line_id: "1.5"})
assert basket.product_count == 1.5
basket_commands.handle_update(request, basket, **{"q_%s" % line_id: "3.5"})
assert basket.product_count == 3.5
basket_commands.handle_update(request, basket, **{"q_%s" % line_id: "3.0"})
assert basket.product_count == 3.0
basket_commands.handle_update(request, basket, **{"q_%s" % line_id: "4"})
assert basket.product_count == 4
basket_commands.handle_update(request, basket, **{"delete_%s" % line_id: "1"})
assert basket.product_count == 0
def test_parallel_baskets(rf):
request = get_request_with_basket()
shop = get_default_shop()
customer = create_random_person()
request = rf.get("/")
request.shop = shop
apply_request_middleware(request)
request.customer = customer
basket_one = get_basket(request, basket_name="basket_one")
basket_two = get_basket(request, basket_name="basket_two")
product_one = get_default_product()
product_two = get_default_product()
product_two.sku = "derpy-hooves"
sales_unit = SalesUnit.objects.create(identifier="test-sales-partial", decimals=2, name="Partial unit")
product_two.sales_unit = sales_unit # Set the sales unit for the product
product_two.save()
basket_commands.handle_add(request, basket_one, product_id=product_one.pk, quantity=1)
basket_commands.handle_add(request, basket_two, product_id=product_two.pk, quantity=3.5)
assert basket_one.product_count == 1
assert basket_two.product_count == 3.5
def test_complex_variation():
request = get_request_with_basket()
basket = request.basket
shop = get_default_shop()
supplier = get_default_supplier()
parent = create_product("SuperComplexVarParent", shop=shop, supplier=supplier)
color_var = ProductVariationVariable.objects.create(product=parent, identifier="color")
size_var = ProductVariationVariable.objects.create(product=parent, identifier="size")
ProductVariationVariableValue.objects.create(variable=color_var, identifier="yellow")
ProductVariationVariableValue.objects.create(variable=size_var, identifier="small")
combinations = list(parent.get_all_available_combinations())
for combo in combinations:
child = create_product("xyz-%s" % combo["sku_part"], shop=shop, supplier=supplier)
child.link_to_parent(parent, combo["variable_to_value"])
# Elided product should not yield a result
yellow_color_value = ProductVariationVariableValue.objects.get(variable=color_var, identifier="yellow")
small_size_value = ProductVariationVariableValue.objects.get(variable=size_var, identifier="small")
# add to basket yellow + small
kwargs = {"var_%d" % color_var.pk: yellow_color_value.pk, "var_%d" % size_var.pk: small_size_value.pk}
basket_commands.handle_add_var(request, basket, parent.id, **kwargs)
assert basket.get_product_ids_and_quantities()[child.pk] == 1
with pytest.raises(ValidationError):
kwargs = {"var_%d" % color_var.pk: yellow_color_value.pk, "var_%d" % size_var.pk: small_size_value.pk + 1}
basket_commands.handle_add_var(request, basket, parent.id, **kwargs)
def test_ajax():
product = get_default_product()
commands = get_basket_command_dispatcher(get_request_with_basket())
commands.ajax = True
rv = commands.handle("add", kwargs=dict(product_id=product.pk, quantity=-3))
assert isinstance(rv, JsonResponse)
assert commands.basket.product_count == 0
def test_add_invalid_product():
shop = get_default_shop()
supplier = get_default_supplier()
request = get_request_with_basket()
basket = request.basket
# cannot add simple/variable variation parent to the basket
parent = create_product("parent", shop=shop, supplier=supplier)
child = create_product("child", shop=shop, supplier=supplier)
child.link_to_parent(parent)
parent.refresh_from_db()
assert parent.mode == ProductMode.SIMPLE_VARIATION_PARENT
with pytest.raises(ValidationError) as excinfo:
basket_commands.handle_add(request, basket, product_id=parent.pk, quantity=2)
assert excinfo.value.code == 'invalid_product'
child.unlink_from_parent()
child.link_to_parent(parent, variables={"size": "XXL"})
parent.refresh_from_db()
assert parent.mode == ProductMode.VARIABLE_VARIATION_PARENT
with pytest.raises(ValidationError) as excinfo:
basket_commands.handle_add(request, basket, product_id=parent.pk, quantity=3)
assert excinfo.value.code == 'invalid_product'
def test_basket_update():
request = get_request_with_basket()
basket = request.basket
product = create_product('fractionable', fractional=True)
complete_product(product)
basket_commands.handle_add(request, basket, product_id=product.pk, quantity=1.75)
assert basket.product_count == 1.75
line_id = basket.get_lines()[0].line_id
basket_commands.handle_update(request, basket, **{"q_%s" % line_id: "2"})
assert basket.product_count == 2
basket_commands.handle_update(request, basket, **{"delete_%s" % line_id: "1"})
assert basket.product_count == 0
def test_add_and_invalid_product():
shop = get_default_shop()
product = create_product('fractionable', fractional=True)
complete_product(product)
supplier = get_default_supplier()
request = get_request_with_basket()
basket = request.basket
# remove the shop product
product.get_shop_instance(shop).delete()
with pytest.raises(ValidationError) as exc:
basket_commands.handle_add(request, basket, **{
"product_id": product.pk, "quantity": 1, "supplier_id": supplier.pk
})
assert "Product not available in this shop" in exc.value.message
def test_custom_basket_command():
ok = []
def noop(**kwargs):
ok.append(kwargs)
def get_custom_command(command, **kwargs):
if command == "test_custom_basket_command":
return noop
old_n_receivers = len(get_basket_command_handler.receivers)
try:
get_basket_command_handler.connect(get_custom_command, dispatch_uid="test_custom_basket_command")
commands = get_basket_command_dispatcher(request=get_request_with_basket())
commands.handle("test_custom_basket_command")
assert ok # heh.
finally:
get_basket_command_handler.disconnect(dispatch_uid="test_custom_basket_command")
assert old_n_receivers == len(get_basket_command_handler.receivers)
def test_variation():
request = get_request_with_basket()
basket = request.basket
shop = get_default_shop()
supplier = get_default_supplier()
parent = create_product("BuVarParent", shop=shop, supplier=supplier)
child = create_product("BuVarChild", shop=shop, supplier=supplier)
child.link_to_parent(parent, variables={"test": "very"})
attr = parent.variation_variables.get(identifier="test")
val = attr.values.get(identifier="very")
basket_commands.handle_add_var(request, basket, parent.id, **{"var_%s" % attr.id: val.id})
assert basket.get_product_ids_and_quantities()[child.pk] == 1
with pytest.raises(ValidationError):
basket_commands.handle_add_var(request, basket, parent.id, **{"var_%s" % attr.id: (val.id + 1)})
def test_dne():
commands = get_basket_command_dispatcher(get_request_with_basket())
with pytest.raises(Exception):
commands.handle("_doesnotexist_")
def test_basket_update_with_package_product():
if "E-Commerce.simple_supplier" not in settings.INSTALLED_APPS:
pytest.skip("Need E-Commerce.simple_supplier in INSTALLED_APPS")
from E-Commerce_tests.simple_supplier.utils import get_simple_supplier
request = get_request_with_basket()
basket = request.basket
shop = get_default_shop()
supplier = get_simple_supplier()
parent, child = get_unstocked_package_product_and_stocked_child(shop, supplier, child_logical_quantity=2)
# There should be enough stock for 1 parent and 1 extra child, each of quantity 1
basket_commands.handle_add(request, basket, product_id=parent.pk, quantity=1)
assert basket.product_count == 1
basket_commands.handle_add(request, basket, product_id=child.pk, quantity=1)
assert basket.product_count == 2
assert not messages.get_messages(request)
basket_lines = {line.product.id: line for line in basket.get_lines()}
package_line = basket_lines[parent.id]
extra_child_line = basket_lines[child.id]
# Trying to increase package product line quantity should fail, with error message
basket_commands.handle_update(request, basket, **{"q_%s" % package_line.line_id: "2"})
assert basket.product_count == 2
assert len(messages.get_messages(request)) == 1
# So should increasing the extra child line quantity
basket_commands.handle_update(request, basket, **{"q_%s" % extra_child_line.line_id: "2"})
assert basket.product_count == 2
assert len(messages.get_messages(request)) == 2
# However, if we delete the parent line, we can increase the extra child
basket_commands.handle_update(request, basket, **{"delete_%s" % package_line.line_id: "1"})
assert basket.product_count == 1
basket_commands.handle_update(request, basket, **{"q_%s" % extra_child_line.line_id: "2"})
assert basket.product_count == 2
# Resetting to original basket contents
basket_commands.handle_update(request, basket, **{"q_%s" % extra_child_line.line_id: "1"})
basket_commands.handle_add(request, basket, product_id=parent.pk, quantity=1)
basket_lines = {line.product.id: line for line in basket.get_lines()}
package_line = basket_lines[parent.id] # Package line will have a new ID
assert basket.product_count == 2
# Like above, delete the child line and we can now increase the parent
basket_commands.handle_update(request, basket, **{"delete_%s" % extra_child_line.line_id: "1"})
assert basket.product_count == 1
basket_commands.handle_update(request, basket, **{"q_%s" % package_line.line_id: "2"})
assert basket.product_count == 2
# Clear basket
basket_commands.handle_clear(request, basket)
assert basket.product_count == 0
# Remove the Shop Product from the child
child.get_shop_instance(shop).delete()
# Child not available for this shop
with pytest.raises(ProductNotOrderableProblem):
basket_commands.handle_add(request, basket, product_id=parent.pk, quantity=1)
# use the update methods object to check orderability errors
update_methods = BasketUpdateMethods(request, basket)
errors = update_methods._get_orderability_errors(parent, supplier, 1)
assert len(errors) == 2
assert any(["product_not_available_in_shop" in error.code for error in errors])
def test_basket_partial_quantity_update_all_product_counts():
shop = get_default_shop()
supplier = get_default_supplier()
request = get_request_with_basket()
basket = request.basket
pieces = SalesUnit.objects.create(
identifier="pieces", decimals=0, name="Pieces", symbol='pc.')
kilograms = SalesUnit.objects.create(
identifier="kilograms", decimals=3, name="Kilograms", symbol='kg')
cup = create_product(
sku="COFFEE-CUP-123", sales_unit=pieces, shop=shop, supplier=supplier)
beans = create_product(
sku="COFFEEBEANS3", sales_unit=kilograms, shop=shop, supplier=supplier)
beans_shop_product = beans.get_shop_instance(shop)
beans_shop_product.minimum_purchase_quantity = Decimal('0.1')
beans_shop_product.save()
pears = create_product(
sku="PEARS-27", sales_unit=kilograms, shop=shop, supplier=supplier)
add = basket_commands.handle_add
update = basket_commands.handle_update
# Empty basket
assert basket.product_count == 0
assert basket.smart_product_count == 0
assert basket.product_line_count == 0
# 1 cup
add(request, basket, product_id=cup.pk, quantity=1)
assert basket.product_count == 1
assert basket.smart_product_count == 1
assert basket.product_line_count == 1
# Basket update operations work by prefixing line id with operation
qty_update_cup = 'q_' + basket.get_lines()[0].line_id
delete_cup = 'delete_' + basket.get_lines()[0].line_id
# 3 cups
update(request, basket, **{qty_update_cup: "3"})
assert basket.product_count == 3
assert basket.smart_product_count == 3
assert basket.product_line_count == 1
# 3 cups + 0.5 kg beans
add(request, basket, product_id=beans.pk, quantity='0.5')
assert basket.product_count == Decimal('3.5')
assert basket.smart_product_count == 4
assert basket.product_line_count == 2
qty_update_beans = 'q_' + basket.get_lines()[1].line_id
delete_beans1 = 'delete_' + basket.get_lines()[1].line_id
# 1 cup + 2.520 kg beans
update(request, basket, **{qty_update_cup: "1.0"})
update(request, basket, **{qty_update_beans: "2.520"})
assert basket.product_count == Decimal('3.520')
assert basket.smart_product_count == 2
assert basket.product_line_count == 2
# 42 cups + 2.520 kg beans
update(request, basket, **{qty_update_cup: "42"})
assert basket.product_count == Decimal('44.520')
assert basket.smart_product_count == 43
assert basket.product_line_count == 2
# 42 cups + 2.520 kg beans + 3.5 kg pears
add(request, basket, product_id=pears.pk, quantity='3.5')
assert basket.product_count == Decimal('48.020')
assert basket.smart_product_count == 44
assert basket.product_line_count == 3
# 42 cups + 3.5 kg pears
update(request, basket, **{delete_beans1: "1"})
assert basket.product_count == Decimal('45.5')
assert basket.smart_product_count == 43
assert basket.product_line_count == 2
# 3.5 kg pears
update(request, basket, **{delete_cup: "1"})
assert basket.product_count == Decimal('3.5')
assert basket.smart_product_count == 1
assert basket.product_line_count == 1
def test_redirect():
commands = ReturnUrlBasketCommandDispatcher(request=get_request_with_basket())
commands.ajax = False
assert isinstance(commands.handle("clear"), HttpResponseRedirect)
def test_nonajax():
product = get_default_product()
commands = get_basket_command_dispatcher(get_request_with_basket())
commands.ajax = False
with pytest.raises(Exception):
commands.handle("add", kwargs=dict(product_id=product.pk, quantity=-3))