def test_follow_user(auth_client):
url = reverse('auth:user-list').replace('users', 'user')
current_user_data = auth_client.get(url).data
user = UserFactory(first_name=fake.first_name(), last_name=fake.last_name())
url = reverse('auth:user-follow', kwargs={'pk': user.id})
response = auth_client.post(url)
assert response.status_code == status.HTTP_200_OK
assert current_user_data.get('username') in user.followers.all().values_list('username', flat=True)
def test_create_vote(auth_client):
post_data = factory.build(dict, FACTORY_CLASS=PostFactory)
post_data['options'] = ",".join(fake.sentence(nb_words=3) for i in range(random.randint(0, 5)))
url = reverse('blog:posts-list')
response = auth_client.post(url, data=post_data)
assert response.status_code == status.HTTP_201_CREATED
assert response.data.get('vote')
def test_create_post(auth_client):
url = reverse('auth:user-list').replace('users', 'user')
current_user_data = auth_client.get(url).data
url = reverse('blog:posts-list')
post_data = factory.build(dict, FACTORY_CLASS=PostFactory)
response = auth_client.post(url, data=post_data)
assert response.status_code == status.HTTP_201_CREATED
assert response.data.get('user') == current_user_data.get('id')
def test_add_comment(auth_client):
url = reverse('auth:user-list').replace('users', 'user')
current_user_data = auth_client.get(url).data
post = PostFactory(user_id=current_user_data.get('id'))
url = reverse('blog:posts-comments', kwargs={'pk': post.id})
comment_data = factory.build(dict, FACTORY_CLASS=PostFactory)
response = auth_client.post(url, data=comment_data)
assert response.status_code == status.HTTP_201_CREATED
assert response.data.get('user').get('id') == current_user_data.get('id')
assert response.data.get('post') == post.id
def test_vote(auth_client):
url = reverse('auth:user-list').replace('users', 'user')
current_user_data = auth_client.get(url).data
vote = VoteFactory()
PostFactory(user_id=current_user_data.get('id'), vote=vote)
options = []
for i in range(random.randint(1, 5)):
option = OptionFactory(vote=vote)
options.append(option)
url = reverse('blog:votes-vote', kwargs={'pk': vote.id})
response = auth_client.post(url, data={'option': options[0].id})
assert response.status_code == status.HTTP_200_OK
assert current_user_data.get('id') in response.data.get('vote').get('users')
assert current_user_data.get('id') in response.data.get('vote').get('options')[0].get('users')