def test_is_removable_by_user(self):
"""Can current user remove provider."""
# Create Customer
customer = None
serializer = CustomerSerializer(data=self.customer_data[0])
if serializer.is_valid(raise_exception=True):
customer = serializer.save()
# Create Provider
provider = Provider.objects.create(name='providername', created_by=customer.owner, customer=customer)
provider_uuid = provider.uuid
# Create another user for negative tests
# Add another user
group = Group.objects.get(name=customer.name)
new_user_dict = self.gen_user_data()
user_serializer = UserSerializer(data=new_user_dict)
new_user = None
if user_serializer.is_valid(raise_exception=True):
new_user = user_serializer.save()
group.user_set.add(new_user)
manager = ProviderManager(provider_uuid)
self.assertTrue(manager.is_removable_by_user(customer.owner))
self.assertFalse(manager.is_removable_by_user(new_user))
superuser = User.objects.filter(is_superuser=True).first()
self.assertFalse(manager.is_removable_by_user(superuser))
def test_is_removable_by_user(self):
"""Can current user remove provider."""
# Create Provider
provider = Provider.objects.create(name="providername", created_by=self.user, customer=self.customer)
provider_uuid = provider.uuid
user_data = self._create_user_data()
request_context = self._create_request_context(self.create_mock_customer_data(), user_data)
new_user = None
serializer = UserSerializer(data=user_data, context=request_context)
if serializer.is_valid(raise_exception=True):
new_user = serializer.save()
manager = ProviderManager(provider_uuid)
self.assertTrue(manager.is_removable_by_user(self.user))
self.assertFalse(manager.is_removable_by_user(new_user))