def test_extending_shipment_with_extra_fields(rf, admin_user):
shop = get_default_shop()
supplier = get_default_supplier()
product = create_product(sku="test-sku",
shop=shop,
supplier=supplier,
default_price=3.33)
quantity = 1
order = create_order_with_product(product,
supplier,
quantity=quantity,
taxless_base_unit_price=1,
shop=shop)
extend_form_class = "shuup_tests.admin.test_shipment_creator.ShipmentFormModifierTest"
with override_provides(FORM_MODIFIER_PROVIDER_KEY, [extend_form_class]):
request = apply_request_middleware(rf.get("/"), user=admin_user)
view = OrderCreateShipmentView.as_view()
response = view(request, pk=order.pk).render()
assert response.status_code == 200
# Should contain supplier input, input for product and input for phone
soup = BeautifulSoup(response.content)
assert soup.find("input", {"id": "id_q_%s" % product.pk})
assert soup.find("select", {"id": "id_supplier"})
assert soup.find("input", {"id": "id_phone"})
def test_extending_shipment_form_valid_hook(rf, admin_user):
shop = get_default_shop()
supplier = get_default_supplier()
product = create_product(sku="test-sku", shop=shop, supplier=supplier, default_price=3.33)
quantity = 1
order = create_order_with_product(
product, supplier, quantity=quantity, taxless_base_unit_price=1, shop=shop)
extend_form_class = "shuup_tests.admin.test_shipment_creator.ShipmentFormModifierTest"
with override_provides(ShipmentForm.form_modifier_provide_key, [extend_form_class]):
phone_number = "+358911"
data = {
"q_%s" % product.pk: 1,
"phone": phone_number
}
request = apply_request_middleware(rf.post("/", data=data), user=admin_user)
view = OrderCreateShipmentView.as_view()
response = view(request, pk=order.pk, supplier_pk=supplier.pk)
assert response.status_code == 302
# Order should now have shipment, but let's re fetch it first
order = Order.objects.get(pk=order.pk)
assert order.shipments.count() == 1
shipment = order.shipments.first()
assert order.shipping_data.get(shipment.identifier).get("phone") == phone_number
assert shipment.supplier_id == supplier.id
assert shipment.products.count() == 1
assert shipment.products.first().product_id == product.id
def test_extending_shipment_clean_hook(rf, admin_user):
shop = get_default_shop()
supplier = get_default_supplier()
product = create_product(sku="test-sku",
shop=shop,
supplier=supplier,
default_price=3.33)
quantity = 1
order = create_order_with_product(product,
supplier,
quantity=quantity,
taxless_base_unit_price=1,
shop=shop)
extend_form_class = "shuup_tests.admin.test_shipment_creator.ShipmentFormModifierTest"
with override_provides(FORM_MODIFIER_PROVIDER_KEY, [extend_form_class]):
data = {
"q_%s" % product.pk: 1,
"supplier": supplier.pk,
"phone": "911"
}
request = apply_request_middleware(rf.post("/", data=data),
user=admin_user)
view = OrderCreateShipmentView.as_view()
response = view(request, pk=order.pk).render()
assert response.status_code == 200
soup = BeautifulSoup(response.content)
assert soup.body.findAll(
text=re.compile("Phone number should start with country code!"))
def test_shipment_creating_view_post(rf, admin_user):
shop = get_default_shop()
supplier = get_default_supplier()
product = create_product(sku="test-sku",
shop=shop,
supplier=supplier,
default_price=3.33)
order = create_order_with_product(product,
supplier,
quantity=1,
taxless_base_unit_price=1,
shop=shop)
data = {"q_%s" % product.pk: 1, "supplier": supplier.pk}
request = apply_request_middleware(rf.post("/", data=data),
user=admin_user)
view = OrderCreateShipmentView.as_view()
response = view(request, pk=order.pk)
assert response.status_code == 302
# Order should have shipment
assert order.shipments.count() == 1
shipment = order.shipments.first()
assert shipment.supplier_id == supplier.id
assert shipment.products.count() == 1
assert shipment.products.first().product_id == product.id
def test_extending_shipment_form_valid_hook(rf, admin_user):
shop = get_default_shop()
supplier = get_default_supplier()
product = create_product(sku="test-sku", shop=shop, supplier=supplier, default_price=3.33)
quantity = 1
order = create_order_with_product(
product, supplier, quantity=quantity, taxless_base_unit_price=1, shop=shop)
extend_form_class = "shuup_tests.admin.test_shipment_creator.ShipmentFormModifierTest"
with override_provides(ShipmentForm.form_modifier_provide_key, [extend_form_class]):
phone_number = "+358911"
data = {
"q_%s" % product.pk: 1,
"supplier": supplier.pk,
"phone": phone_number
}
request = apply_request_middleware(rf.post("/", data=data), user=admin_user)
view = OrderCreateShipmentView.as_view()
response = view(request, pk=order.pk)
assert response.status_code == 302
# Order should now have shipment, but let's re fetch it first
order = Order.objects.get(pk=order.pk)
assert order.shipments.count() == 1
shipment = order.shipments.first()
assert order.shipping_data.get(shipment.identifier).get("phone") == phone_number
assert shipment.supplier_id == supplier.id
assert shipment.products.count() == 1
assert shipment.products.first().product_id == product.id
def test_shipment_creating_view_get(rf, admin_user):
shop = get_default_shop()
supplier = get_default_supplier()
product = create_product(sku="test-sku", shop=shop, supplier=supplier, default_price=3.33)
quantity = 1
order = create_order_with_product(
product, supplier, quantity=quantity, taxless_base_unit_price=1, shop=shop)
request = apply_request_middleware(rf.get("/"), user=admin_user)
view = OrderCreateShipmentView.as_view()
response = view(request, pk=order.pk, supplier_pk=supplier.pk).render()
assert response.status_code == 200
# Should contain supplier input and input for product
soup = BeautifulSoup(response.content)
assert soup.find("input", {"id": "id_q_%s" % product.pk})
def test_shipment_creating_view_get(rf, admin_user):
shop = get_default_shop()
supplier = get_default_supplier()
product = create_product(sku="test-sku", shop=shop, supplier=supplier, default_price=3.33)
quantity = 1
order = create_order_with_product(
product, supplier, quantity=quantity, taxless_base_unit_price=1, shop=shop)
request = apply_request_middleware(rf.get("/"), user=admin_user)
view = OrderCreateShipmentView.as_view()
response = view(request, pk=order.pk, supplier_pk=supplier.pk).render()
assert response.status_code == 200
# Should contain supplier input and input for product
soup = BeautifulSoup(response.content)
assert soup.find("input", {"id": "id_q_%s" % product.pk})
def test_extending_shipment_with_extra_fields(rf, admin_user):
shop = get_default_shop()
supplier = get_default_supplier()
product = create_product(sku="test-sku", shop=shop, supplier=supplier, default_price=3.33)
quantity = 1
order = create_order_with_product(
product, supplier, quantity=quantity, taxless_base_unit_price=1, shop=shop)
extend_form_class = "shuup_tests.admin.test_shipment_creator.ShipmentFormModifierTest"
with override_provides(ShipmentForm.form_modifier_provide_key, [extend_form_class]):
request = apply_request_middleware(rf.get("/"), user=admin_user)
view = OrderCreateShipmentView.as_view()
response = view(request, pk=order.pk, supplier_pk=supplier.pk).render()
assert response.status_code == 200
# Should contain supplier input, input for product and input for phone
soup = BeautifulSoup(response.content)
assert soup.find("input", {"id": "id_q_%s" % product.pk})
assert soup.find("input", {"id": "id_phone"})
def test_extending_shipment_clean_hook(rf, admin_user):
shop = get_default_shop()
supplier = get_default_supplier()
product = create_product(sku="test-sku", shop=shop, supplier=supplier, default_price=3.33)
quantity = 1
order = create_order_with_product(
product, supplier, quantity=quantity, taxless_base_unit_price=1, shop=shop)
extend_form_class = "shuup_tests.admin.test_shipment_creator.ShipmentFormModifierTest"
with override_provides(ShipmentForm.form_modifier_provide_key, [extend_form_class]):
data = {
"q_%s" % product.pk: 1,
"phone": "911"
}
request = apply_request_middleware(rf.post("/", data=data), user=admin_user)
view = OrderCreateShipmentView.as_view()
response = view(request, pk=order.pk, supplier_pk=supplier.pk).render()
assert response.status_code == 200
soup = BeautifulSoup(response.content)
assert soup.body.findAll(text=re.compile("Phone number should start with country code!"))
def test_shipment_creating_view_post(rf, admin_user):
shop = get_default_shop()
supplier = get_default_supplier()
product = create_product(sku="test-sku", shop=shop, supplier=supplier, default_price=3.33)
order = create_order_with_product(product, supplier, quantity=1, taxless_base_unit_price=1, shop=shop)
data = {
"q_%s" % product.pk: 1
}
request = apply_request_middleware(rf.post("/", data=data), user=admin_user)
view = OrderCreateShipmentView.as_view()
response = view(request, pk=order.pk, supplier_pk=supplier.pk)
assert response.status_code == 302
# Order should have shipment
assert order.shipments.count() == 1
shipment = order.shipments.first()
assert shipment.supplier_id == supplier.id
assert shipment.products.count() == 1
assert shipment.products.first().product_id == product.id