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'