class UpsellViewSet(CreateModelMixin, DestroyModelMixin, RetrieveModelMixin, UpdateModelMixin, GenericViewSet): permission_classes = ( switch('allow-b2g-paid-submission'), UpsellPermission, ) queryset = AddonUpsell.objects.filter() serializer_class = UpsellSerializer def pre_save(self, obj): if not UpsellPermission().check(self.request, obj.free, obj.premium): raise PermissionDenied('Not allowed to alter that object')
def test_not_switch_flag(self): ok_(not switch('foo')().has_permission(self.request, ''))
def test_waffle_switch(self): self.create_switch('foo') ok_(switch('foo')().has_permission(self.request, ''))
def test_not_switch_flag(self): ok_(not switch("foo")().has_permission(self.request, ""))
def test_waffle_switch(self): self.create_switch("foo") ok_(switch("foo")().has_permission(self.request, ""))