def test_order_creator_account_manager():
company = create_random_company()
shop = get_shop(identifier="random-shop", enabled=True)
source = seed_source(create_random_user(), shop)
source.customer = company
source.add_line(
type=OrderLineType.PRODUCT,
product=get_default_product(),
supplier=get_default_supplier(),
quantity=1,
base_unit_price=source.create_price(10),
)
creator = OrderCreator()
order = creator.create_order(source)
assert order.account_manager is None # Company contact doesn't have account manager field
person = create_random_person()
person.account_manager = create_random_person()
person.save()
source = seed_source(create_random_user(), shop)
source.customer = person
source.add_line(
type=OrderLineType.PRODUCT,
product=get_default_product(),
supplier=get_default_supplier(),
quantity=1,
base_unit_price=source.create_price(10),
)
creator = OrderCreator()
order = creator.create_order(source)
assert order.account_manager is not None
assert order.account_manager == person.account_manager
with pytest.raises(ProtectedError):
person.account_manager.delete()
def test_get_shop(rf, get_shop_fn):
with override_settings(WSHOP_ENABLE_MULTIPLE_SHOPS=True):
activate("en")
shop1 = Shop.objects.create(identifier="shop1",
status=ShopStatus.ENABLED)
shop2 = Shop.objects.create(identifier="shop2",
status=ShopStatus.ENABLED)
normal_user = factories.create_random_user()
staff_user = factories.create_random_user(is_staff=True)
request = apply_request_middleware(rf.post("/"), user=normal_user)
# user not staff
assert get_shop_fn(request) is None
# staff user now
request = apply_request_middleware(rf.post("/"), user=staff_user)
# no shop set
assert get_shop_fn(request) is None
# adds the user to a shop
shop1.staff_members.add(staff_user)
assert get_shop_fn(request) == shop1
# adds the user to another shop
shop2.staff_members.add(staff_user)
# still the first shop as we do not set any
assert get_shop_fn(request) == shop1
def test_add_product_with_extra_parent_line(rf):
"""
Add product to basket with extra info and parent line
"""
with override_settings(**CORE_BASKET_SETTINGS):
shop = factories.get_default_shop()
user = factories.create_random_user()
product = factories.create_product("product", shop, factories.get_default_supplier(), 10)
request = apply_request_middleware(rf.get("/"), user=user)
basket = get_basket(request, "basket")
basket.customer = get_person_contact(user)
cmd_response = basket_commands.handle_add(request, basket, product.id, 1, extra={"more": "stuff"})
line_id1 = cmd_response["line_id"]
assert cmd_response["ok"]
line1 = basket.get_basket_line(line_id1)
assert line1._data["more"] == "stuff"
cmd_response = basket_commands.handle_add(
request, basket, product.id, 1, parent_line=line1, force_new_line=True)
line_id2 = cmd_response["line_id"]
assert cmd_response["ok"]
line2 = basket.get_basket_line(line_id2)
assert not line2._data
assert line_id1 != line_id2
assert line2.parent_line.line_id == line_id1
def test_contact_module_search_multishop(rf):
with override_settings(WSHOP_MANAGE_CONTACTS_PER_SHOP=True,
WSHOP_ENABLE_MULTIPLE_SHOPS=True):
staff_user = create_random_user(is_staff=True)
shop1 = get_shop(identifier="shop-1", enabled=True)
shop2 = get_shop(identifier="shop-2", enabled=True)
shop1.staff_members.add(staff_user)
shop2.staff_members.add(staff_user)
cm = ContactModule()
contact = create_random_person(locale="en_US", minimum_name_comp_len=5)
contact.shops.add(shop2)
request = apply_request_middleware(rf.get("/"),
user=staff_user,
shop=shop2)
# find the shop
assert not empty_iterable(
cm.get_search_results(request, query=contact.email))
assert not empty_iterable(
cm.get_search_results(request, query=contact.first_name))
# no shop found
request = apply_request_middleware(rf.get("/"),
user=staff_user,
shop=shop1)
assert empty_iterable(
cm.get_search_results(request, query=contact.email))
def test_contact_company_list_multishop(rf):
with override_settings(WSHOP_MANAGE_CONTACTS_PER_SHOP=True, WSHOP_ENABLE_MULTIPLE_SHOPS=True):
staff_user = create_random_user(is_staff=True)
shop1 = get_shop(identifier="shop-1", enabled=True)
shop2 = get_shop(identifier="shop-2", enabled=True)
shop1.staff_members.add(staff_user)
shop2.staff_members.add(staff_user)
# only available in shop2
contact = create_random_person(locale="en_US", minimum_name_comp_len=5)
contact.shops.add(shop2)
# only available in shop1
company = create_random_company()
company.shops.add(shop1)
view = ContactListView()
request = apply_request_middleware(rf.get("/"), user=staff_user, shop=shop1)
view.request = request
assert company in view.get_queryset()
assert contact not in view.get_queryset()
request = apply_request_middleware(rf.get("/"), user=staff_user, shop=shop2)
view.request = request
assert contact in view.get_queryset()
assert company not in view.get_queryset()
def test_set_non_shop_member_customer(rf):
"""
Set some customer to the basket that is not member of the shop
"""
with override_settings(**CORE_BASKET_SETTINGS):
shop = factories.get_shop(False)
assert shop != factories.get_default_shop()
user = factories.create_random_user()
request = apply_request_middleware(rf.get("/"), user=user)
basket = get_basket(request, "basket")
basket.customer = get_person_contact(user)
assert basket.shop == factories.get_default_shop()
person = factories.create_random_person()
person.shops.add(shop)
company = factories.create_random_company()
company.shops.add(shop)
for customer in [person, company]:
with pytest.raises(ValidationError) as exc:
basket_commands.handle_set_customer(request, basket, customer)
assert exc.value.code == "invalid_customer_shop"
assert basket.customer == get_person_contact(user)
def test_set_company_customer(rf):
"""
Set a company as the basket customer
"""
with override_settings(**CORE_BASKET_SETTINGS):
user = factories.create_random_user()
request = apply_request_middleware(rf.get("/"), user=user)
basket = get_basket(request, "basket")
basket.customer = get_person_contact(user)
person = factories.create_random_person()
company = factories.create_random_company()
# no orderer provided
with pytest.raises(ValidationError) as exc:
basket_commands.handle_set_customer(request, basket, company)
assert exc.value.code == "invalid_orderer"
assert basket.customer == get_person_contact(user)
# orderer provided but not member of the company
with pytest.raises(ValidationError) as exc:
basket_commands.handle_set_customer(request, basket, company, person)
assert exc.value.code == "orderer_not_company_member"
assert basket.customer == get_person_contact(user)
# orderer provided but user not member of the company
company.members.add(person)
with pytest.raises(ValidationError) as exc:
basket_commands.handle_set_customer(request, basket, company, person)
assert exc.value.code == "not_company_member"
assert basket.customer == get_person_contact(user)
# staff and admin can add any the company and orderer without being member of the company
superuser = factories.create_random_user(is_superuser=True)
staff = factories.create_random_user(is_staff=True)
basket.shop.staff_members.add(staff)
for user in [superuser, staff]:
basket.customer = None
basket.orderer = None
request = apply_request_middleware(rf.get("/"), user=user)
assert basket_commands.handle_set_customer(request, basket, company, person)["ok"] is True
assert basket.customer == company
assert basket.orderer == person
def test_set_different_customer(rf):
"""
Set some customer to the basket that is not the request one
"""
with override_settings(**CORE_BASKET_SETTINGS):
user = factories.create_random_user()
request = apply_request_middleware(rf.get("/"), user=user)
basket = get_basket(request, "basket")
basket.customer = get_person_contact(user)
person1 = factories.create_random_person()
with pytest.raises(ValidationError) as exc:
basket_commands.handle_set_customer(request, basket, person1)
assert exc.value.code == "no_permission"
assert basket.customer == get_person_contact(user)
# with superuser
person2 = factories.create_random_person()
superuser = factories.create_random_user(is_superuser=True)
request = apply_request_middleware(rf.get("/"), user=superuser)
assert basket_commands.handle_set_customer(request, basket, person2)["ok"] is True
assert basket.customer == person2
assert basket.orderer == person2
# with staff user not member of the shop
person3 = factories.create_random_person()
staff = factories.create_random_user(is_staff=True)
request = apply_request_middleware(rf.get("/"), user=staff)
with pytest.raises(ValidationError) as exc:
basket_commands.handle_set_customer(request, basket, person3)
assert exc.value.code == "no_permission"
assert basket.customer == person2
assert basket.orderer == person2
# with staff user member of the shop
person4 = factories.create_random_person()
staff_member = factories.create_random_user(is_staff=True)
basket.shop.staff_members.add(staff_member)
request = apply_request_middleware(rf.get("/"), user=staff_member)
assert basket_commands.handle_set_customer(request, basket, person4)["ok"] is True
assert basket.customer == person4
assert basket.orderer == person4
def test_set_shop(rf, set_shop_fn, get_shop_fn):
with override_settings(WSHOP_ENABLE_MULTIPLE_SHOPS=True):
activate("en")
factories.get_default_shop()
shop1 = Shop.objects.create(identifier="shop1",
status=ShopStatus.ENABLED)
shop2 = Shop.objects.create(identifier="shop2",
status=ShopStatus.ENABLED)
normal_user = factories.create_random_user()
staff_user = factories.create_random_user(is_staff=True)
request = apply_request_middleware(rf.post("/"), user=normal_user)
# user not staff
with pytest.raises(PermissionDenied):
set_shop_fn(request, shop1)
# staff user now
request = apply_request_middleware(rf.post("/"), user=staff_user)
# the user is not member of the shop staff
with pytest.raises(PermissionDenied):
set_shop_fn(request, shop1)
assert get_shop_fn(request) is None
# user is member of the shop staff
shop1.staff_members.add(staff_user)
set_shop_fn(request, shop1)
assert get_shop_fn(request) == shop1
# can't set a shop which user is not member
with pytest.raises(PermissionDenied):
set_shop_fn(request, shop2)
assert get_shop_fn(request) == shop1
# adds the user to another shop
shop2.staff_members.add(staff_user)
set_shop_fn(request, shop2)
assert get_shop_fn(request) == shop2
def test_set_from_customer_to_anonymous(rf):
"""
Set anonymous to the basket customer
"""
with override_settings(**CORE_BASKET_SETTINGS):
user = factories.create_random_user()
request = apply_request_middleware(rf.get("/"), user=user)
basket = get_basket(request, "basket")
basket.customer = get_person_contact(user)
assert basket_commands.handle_set_customer(request, basket, AnonymousContact())["ok"] is True
assert basket.customer == AnonymousContact()
assert basket.orderer == AnonymousContact()
assert basket.creator == user
def test_add_product(rf):
"""
Add product to basket
"""
with override_settings(**CORE_BASKET_SETTINGS):
shop = factories.get_default_shop()
user = factories.create_random_user()
product = factories.create_product("product", shop, factories.get_default_supplier(), 10)
request = apply_request_middleware(rf.get("/"), user=user)
basket = get_basket(request, "basket")
basket.customer = get_person_contact(user)
cmd_response = basket_commands.handle_add(request, basket, product.id, 1)
assert cmd_response["ok"]
assert cmd_response["line_id"]
assert cmd_response["added"] == 1
assert len(basket.get_lines()) == 1
def test_anonymous_set_company_customer(rf):
"""
Set a company as the basket customer
"""
with override_settings(**CORE_BASKET_SETTINGS):
user = factories.create_random_user()
request = apply_request_middleware(rf.get("/"), user=user)
basket = get_basket(request, "basket")
basket.customer = AnonymousContact()
person = factories.create_random_person()
company = factories.create_random_company()
company.members.add(person)
with pytest.raises(ValidationError) as exc:
basket_commands.handle_set_customer(request, basket, company, person)
assert exc.value.code == "not_company_member"
assert basket.customer == AnonymousContact()
def test_set_from_anonymous_to_customer_auth(rf):
"""
Set some random customer to the basket when authenticated
"""
with override_settings(**CORE_BASKET_SETTINGS):
user = factories.create_random_user()
request = apply_request_middleware(rf.get("/"), user=user)
basket = get_basket(request, "basket")
basket.customer = AnonymousContact()
# can not set the customer for something different as the request customer
with pytest.raises(ValidationError) as exc:
basket_commands.handle_set_customer(request, basket, factories.create_random_person())
assert exc.value.code == "no_permission"
assert basket.customer == AnonymousContact()
assert basket_commands.handle_set_customer(request, basket, get_person_contact(user))["ok"] is True
assert basket.customer == get_person_contact(user)
def test_order_creator_company_multishop():
with override_settings(WSHOP_MANAGE_CONTACTS_PER_SHOP=True,
WSHOP_ENABLE_MULTIPLE_SHOPS=True):
company = create_random_company()
shop = get_shop(identifier="random-shop", enabled=True)
source = seed_source(create_random_user(), shop)
source.customer = company
source.add_line(
type=OrderLineType.PRODUCT,
product=get_default_product(),
supplier=get_default_supplier(),
quantity=1,
base_unit_price=source.create_price(10),
)
creator = OrderCreator()
creator.create_order(source)
company.refresh_from_db()
assert shop in company.shops.all()
def test_unset_shop(rf, set_shop_fn, get_shop_fn, unset_shop_fn):
activate("en")
factories.get_default_shop()
staff_user = factories.create_random_user(is_staff=True)
shop1 = Shop.objects.create(identifier="shop1", status=ShopStatus.ENABLED)
shop2 = Shop.objects.create(identifier="shop2", status=ShopStatus.ENABLED)
shop1.staff_members.add(staff_user)
shop2.staff_members.add(staff_user)
request = apply_request_middleware(rf.post("/"), user=staff_user)
with override_settings(WSHOP_ENABLE_MULTIPLE_SHOPS=True):
set_shop_fn(request, shop2)
assert get_shop_fn(request) == shop2
unset_shop_fn(request)
# returns the first available shop
assert get_shop_fn(request) == shop1
def test_set_customer_with_custom_basket_lines(rf):
"""
Set anonymous to the basket customer
"""
with override_settings(**CORE_BASKET_SETTINGS):
factories.get_default_shop()
user = factories.create_random_user()
request = apply_request_middleware(rf.get("/"), user=user)
basket = get_basket(request, "basket")
base_unit_price = basket.shop.create_price("10.99")
basket.add_line(text="Custom Line",
type=OrderLineType.OTHER,
line_id="random-you-know",
shop=basket.shop,
quantity=1,
base_unit_price=base_unit_price)
basket.customer = get_person_contact(user)
basket.refresh_lines()
basket.save()
assert basket.customer == get_person_contact(user)
def test_set_not_active_customer(rf):
"""
Set a not active customer to the basket
"""
with override_settings(**CORE_BASKET_SETTINGS):
user = factories.create_random_user()
request = apply_request_middleware(rf.get("/"), user=user)
basket = get_basket(request, "basket")
basket.customer = get_person_contact(user)
person = factories.create_random_person()
person.is_active = False
person.save()
company = factories.create_random_company()
company.is_active = False
company.save()
for customer in [person, company]:
with pytest.raises(ValidationError) as exc:
basket_commands.handle_set_customer(request, basket, customer)
assert exc.value.code == "invalid_customer"
assert basket.customer == get_person_contact(user)
def test_contact_company_edit_multishop(rf):
with override_settings(WSHOP_MANAGE_CONTACTS_PER_SHOP=True, WSHOP_ENABLE_MULTIPLE_SHOPS=True):
staff_user = create_random_user(is_staff=True)
shop1 = get_shop(identifier="shop-1", enabled=True)
shop2 = get_shop(identifier="shop-2", enabled=True)
shop1.staff_members.add(staff_user)
shop2.staff_members.add(staff_user)
# only available in shop2
contact = create_random_person(locale="en_US", minimum_name_comp_len=5)
contact.shops.add(shop2)
# only available in shop1
company = create_random_company()
company.shops.add(shop1)
view = ContactEditView.as_view()
# permission denied for contact and shop1
request = apply_request_middleware(rf.get("/"), user=staff_user, shop=shop1)
with pytest.raises(PermissionDenied):
response = view(request, pk=contact.id)
# permission granted for contact and shop2
request = apply_request_middleware(rf.get("/"), user=staff_user, shop=shop2)
response = view(request, pk=contact.id)
assert response.status_code == 200
# permission denied for company and shop2
request = apply_request_middleware(rf.get("/"), user=staff_user, shop=shop2)
with pytest.raises(PermissionDenied):
response = view(request, pk=company.id)
# permission granted for company and shop1
request = apply_request_middleware(rf.get("/"), user=staff_user, shop=shop1)
response = view(request, pk=company.id)
assert response.status_code == 200
def test_company_contact_detail_multishop(rf):
with override_settings(WSHOP_MANAGE_CONTACTS_PER_SHOP=True, WSHOP_ENABLE_MULTIPLE_SHOPS=True):
staff_user = create_random_user(is_staff=True)
shop1 = get_shop(identifier="shop-1", enabled=True)
shop2 = get_shop(identifier="shop-2", enabled=True)
shop1.staff_members.add(staff_user)
shop2.staff_members.add(staff_user)
company = create_random_company()
# only available in shop1
company.shops.add(shop1)
view = ContactDetailView.as_view()
# company not found for this shop
request = apply_request_middleware(rf.get("/"), user=staff_user, shop=shop2)
with pytest.raises(PermissionDenied):
response = view(request, pk=company.id)
request = apply_request_middleware(rf.get("/"), user=staff_user, shop=shop1)
response = view(request, pk=company.id)
assert response.status_code == 200
def _create_random_staff(shop):
user = factories.create_random_user(is_staff=True)
shop.staff_members.add(user)
return user