def test_multishop_edit_view(rf, admin_user): get_default_shop() request = apply_request_middleware(rf.get("/"), user=admin_user) view = ShopEditView(request=request, kwargs={"pk": None}) with override_settings(SHUUP_ENABLE_MULTIPLE_SHOPS=False): with pytest.raises(Problem): view.get_object() # Now view object should throw Problem with override_settings(SHUUP_ENABLE_MULTIPLE_SHOPS=True): new_shop = view.get_object() assert new_shop.pk is None
def test_shop_edit_has_custom_toolbar_button(rf, admin_user): shop = get_default_shop() request = apply_request_middleware(rf.get("/"), user=admin_user) view_func = ShopEditView.as_view() response = view_func(request, pk=shop.pk) content = force_text(response.render().content) assert "#mocktoolbarbuttonforshop" in content, 'custom toolbar button not found on edit page'