mirror of
https://github.com/EDeev/deev.space.git
synced 2026-10-07 20:49:59 +03:00
Egor Deev
794fceece5
33 теста на pytest-django в боевом режиме (DEBUG=False, настройки dspace/settings_test.py): лайки и дизлайки анонимов, лимит оценок с одного IP, очистка HTML в комментариях и форме обратной связи, капча, экранирование бейджей и постов в Telegram, проверка адресов превью, уникальные просмотры и фильтр ботов, cookie посетителя.
71 lines
2 KiB
Python
71 lines
2 KiB
Python
import json
|
|
|
|
import pytest
|
|
from django.urls import reverse
|
|
|
|
from main.models import ArticleLike
|
|
|
|
PUBLIC_IP = '93.184.215.14'
|
|
UA = 'Mozilla/5.0 (X11; Linux x86_64) Firefox/131.0'
|
|
|
|
|
|
def vote(client, article, is_like=True, ip=PUBLIC_IP):
|
|
return client.post(
|
|
reverse('toggle_article_like', args=[article.id]),
|
|
data=json.dumps({'is_like': is_like}),
|
|
content_type='application/json',
|
|
REMOTE_ADDR=ip,
|
|
HTTP_USER_AGENT=UA,
|
|
)
|
|
|
|
|
|
@pytest.mark.django_db
|
|
def test_anonymous_like_is_counted(client, article):
|
|
response = vote(client, article)
|
|
assert response.status_code == 200
|
|
assert response.json() == {'success': True, 'likes': 1, 'dislikes': 0, 'user_vote': True}
|
|
assert ArticleLike.objects.get().visitor_id
|
|
|
|
|
|
@pytest.mark.django_db
|
|
def test_second_click_removes_vote(client, article):
|
|
vote(client, article)
|
|
response = vote(client, article)
|
|
assert response.json()['user_vote'] is None
|
|
assert ArticleLike.objects.count() == 0
|
|
|
|
|
|
@pytest.mark.django_db
|
|
def test_like_switches_to_dislike(client, article):
|
|
vote(client, article, is_like=True)
|
|
response = vote(client, article, is_like=False)
|
|
assert response.json()['likes'] == 0
|
|
assert response.json()['dislikes'] == 1
|
|
assert ArticleLike.objects.count() == 1
|
|
|
|
|
|
@pytest.mark.django_db
|
|
def test_votes_per_ip_are_limited(client_factory, article):
|
|
statuses = [vote(client_factory(), article).status_code for _ in range(3)]
|
|
assert statuses == [200, 200, 429]
|
|
|
|
|
|
@pytest.mark.django_db
|
|
def test_private_ip_is_not_limited(client_factory, article):
|
|
statuses = [vote(client_factory(), article, ip='10.0.0.5').status_code for _ in range(3)]
|
|
assert statuses == [200, 200, 200]
|
|
|
|
|
|
@pytest.mark.django_db
|
|
def test_invalid_json_returns_400(client, article):
|
|
response = client.post(
|
|
reverse('toggle_article_like', args=[article.id]), data='not json', content_type='application/json'
|
|
)
|
|
assert response.status_code == 400
|
|
|
|
|
|
@pytest.fixture
|
|
def client_factory():
|
|
from django.test import Client
|
|
|
|
return Client
|