102 lines
3.3 KiB
Python
102 lines
3.3 KiB
Python
|
import time
|
||
|
|
from datetime import datetime, timezone
|
||
|
|
|
||
|
|
import requests
|
||
|
|
|
||
|
|
from tests.conftest import BASE_URL
|
||
|
|
from devplacepy.database import get_table
|
||
|
|
from devplacepy.utils import generate_uid
|
||
|
|
|
||
|
|
_counter = [0]
|
||
|
|
AJAX = {"X-Requested-With": "fetch"}
|
||
|
|
|
||
|
|
|
||
|
|
def _session():
|
||
|
|
_counter[0] += 1
|
||
|
|
name = f"vot{int(time.time() * 1000)}{_counter[0]}"
|
||
|
|
s = requests.Session()
|
||
|
|
s.post(f"{BASE_URL}/auth/signup", data={
|
||
|
|
"username": name, "email": f"{name}@t.dev",
|
||
|
|
"password": "secret123", "confirm_password": "secret123",
|
||
|
|
}, allow_redirects=True)
|
||
|
|
return s, name
|
||
|
|
|
||
|
|
|
||
|
|
def _uid(username):
|
||
|
|
return get_table("users").find_one(username=username)["uid"]
|
||
|
|
|
||
|
|
|
||
|
|
def _make_post(owner_uid):
|
||
|
|
uid = generate_uid()
|
||
|
|
get_table("posts").insert({
|
||
|
|
"uid": uid, "user_uid": owner_uid, "slug": f"{uid[:8]}-vote-post",
|
||
|
|
"title": None, "content": "vote target content", "topic": "random",
|
||
|
|
"project_uid": None, "image": None, "stars": 0,
|
||
|
|
"created_at": datetime.now(timezone.utc).isoformat(),
|
||
|
|
})
|
||
|
|
return uid
|
||
|
|
|
||
|
|
|
||
|
|
def test_upvote_returns_ajax_payload(app_server):
|
||
|
|
s_a, a_name = _session()
|
||
|
|
s_b, _ = _session()
|
||
|
|
post_uid = _make_post(_uid(a_name))
|
||
|
|
r = s_b.post(f"{BASE_URL}/votes/post/{post_uid}", data={"value": "1"}, headers=AJAX)
|
||
|
|
assert r.json() == {"net": 1, "up": 1, "down": 0, "value": 1}
|
||
|
|
|
||
|
|
|
||
|
|
def test_repeated_vote_toggles_off(app_server):
|
||
|
|
s_a, a_name = _session()
|
||
|
|
s_b, _ = _session()
|
||
|
|
post_uid = _make_post(_uid(a_name))
|
||
|
|
s_b.post(f"{BASE_URL}/votes/post/{post_uid}", data={"value": "1"}, headers=AJAX)
|
||
|
|
r = s_b.post(f"{BASE_URL}/votes/post/{post_uid}", data={"value": "1"}, headers=AJAX)
|
||
|
|
assert r.json()["net"] == 0
|
||
|
|
assert r.json()["value"] == 0
|
||
|
|
|
||
|
|
|
||
|
|
def test_switch_upvote_to_downvote(app_server):
|
||
|
|
s_a, a_name = _session()
|
||
|
|
s_b, _ = _session()
|
||
|
|
post_uid = _make_post(_uid(a_name))
|
||
|
|
s_b.post(f"{BASE_URL}/votes/post/{post_uid}", data={"value": "1"}, headers=AJAX)
|
||
|
|
r = s_b.post(f"{BASE_URL}/votes/post/{post_uid}", data={"value": "-1"}, headers=AJAX)
|
||
|
|
assert r.json()["net"] == -1
|
||
|
|
assert r.json()["value"] == -1
|
||
|
|
|
||
|
|
|
||
|
|
def test_upvote_notifies_owner_and_awards_xp(app_server):
|
||
|
|
s_a, a_name = _session()
|
||
|
|
s_b, _ = _session()
|
||
|
|
a_uid = _uid(a_name)
|
||
|
|
post_uid = _make_post(a_uid)
|
||
|
|
before = get_table("users").find_one(uid=a_uid).get("xp", 0) or 0
|
||
|
|
s_b.post(f"{BASE_URL}/votes/post/{post_uid}", data={"value": "1"}, headers=AJAX)
|
||
|
|
assert get_table("notifications").count(user_uid=a_uid, type="vote") == 1
|
||
|
|
after = get_table("users").find_one(uid=a_uid).get("xp", 0) or 0
|
||
|
|
assert after - before == 5
|
||
|
|
|
||
|
|
|
||
|
|
def test_self_vote_does_not_notify(app_server):
|
||
|
|
s_a, a_name = _session()
|
||
|
|
a_uid = _uid(a_name)
|
||
|
|
post_uid = _make_post(a_uid)
|
||
|
|
before = get_table("notifications").count(user_uid=a_uid, type="vote")
|
||
|
|
s_a.post(f"{BASE_URL}/votes/post/{post_uid}", data={"value": "1"}, headers=AJAX)
|
||
|
|
after = get_table("notifications").count(user_uid=a_uid, type="vote")
|
||
|
|
assert after == before
|
||
|
|
|
||
|
|
|
||
|
|
def test_non_ajax_vote_redirects_to_referer(app_server):
|
||
|
|
s_a, a_name = _session()
|
||
|
|
s_b, _ = _session()
|
||
|
|
post_uid = _make_post(_uid(a_name))
|
||
|
|
r = s_b.post(
|
||
|
|
f"{BASE_URL}/votes/post/{post_uid}",
|
||
|
|
data={"value": "1"},
|
||
|
|
headers={"Referer": f"{BASE_URL}/feed"},
|
||
|
|
allow_redirects=False,
|
||
|
|
)
|
||
|
|
assert r.status_code == 302
|
||
|
|
assert r.headers["location"] == f"{BASE_URL}/feed"
|