def test_user_follow_clean_up(client):
user = UserFactory()
# create a forum that the user follows
f = ForumFactory(type=Forum.FORUM_POST)
follow(user, f)
assert Follow.objects.count() == 1
# delete user and check that follow is deleted as well
user.delete()
assert Follow.objects.count() == 0
def test_create_basket_wrong_user(self):
"""Only user with permissions should be able to create a basket."""
client = APIClient()
request = client.post(self.API_PATH, self.basket_data, format="json")
dummy_user = UserFactory(username="******")
client.force_authenticate(user=dummy_user)
request = client.post(self.API_PATH, self.basket_data, format="json")
self.assertEqual(403, request.status_code)
dummy_user.delete()
def test_get_basket_variable_GET_data_as_superuser(self):
"""Can we get basket variable data."""
superuser = UserFactory(username="******")
superuser.is_superuser = True
superuser.save()
self.client.force_authenticate(user=superuser)
response = self.client.get(self.API_PATH)
results = json.loads(response.content)["results"]
basket = results[0]
self.assertEqual(self.basket_variable.basket_id, basket["basket_id"])
self.assertEqual(str(self.basket_variable.variable_id),
basket["variable_id"])
superuser.delete()
def test_create_basket_superuser(self):
"""Only user with permissions should be able to create a basket."""
client = APIClient()
dummy_user = UserFactory(username="******")
dummy_user.is_superuser = True
dummy_user.save()
client.force_authenticate(user=dummy_user)
request = client.get(self.API_PATH)
self.assertEqual(200, request.status_code)
dummy_user.delete()
