|
import time
|
|
|
|
import requests
|
|
|
|
from tests.conftest import BASE_URL
|
|
from devplacepy.database import get_table
|
|
|
|
_counter = [0]
|
|
AJAX = {"X-Requested-With": "fetch"}
|
|
|
|
|
|
def _session():
|
|
_counter[0] += 1
|
|
name = f"pol{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 _create_post(session, title, question, options):
|
|
fields = [
|
|
("content", "Poll host post content for tests."),
|
|
("title", title),
|
|
("topic", "question"),
|
|
("poll_question", question),
|
|
]
|
|
fields.extend(("poll_options", option) for option in options)
|
|
r = session.post(f"{BASE_URL}/posts/create", data=fields, allow_redirects=False)
|
|
slug = r.headers["location"].split("/posts/")[-1]
|
|
return get_table("posts").find_one(slug=slug)["uid"]
|
|
|
|
|
|
def _poll_for(post_uid):
|
|
poll = get_table("polls").find_one(post_uid=post_uid)
|
|
options = list(get_table("poll_options").find(poll_uid=poll["uid"], order_by=["position"]))
|
|
return poll, options
|
|
|
|
|
|
def test_valid_poll_persists(app_server):
|
|
s, _ = _session()
|
|
title = f"valid-poll-{int(time.time() * 1000)}"
|
|
post_uid = _create_post(s, title, "Tabs or spaces?", ["Tabs", "Spaces", "Both"])
|
|
poll, options = _poll_for(post_uid)
|
|
assert poll is not None
|
|
assert len(options) == 3
|
|
|
|
|
|
def test_poll_dropped_with_one_option(app_server):
|
|
s, _ = _session()
|
|
title = f"one-option-{int(time.time() * 1000)}"
|
|
post_uid = _create_post(s, title, "Only one?", ["Solo"])
|
|
assert get_table("polls").find_one(post_uid=post_uid) is None
|
|
|
|
|
|
def test_poll_dropped_without_question(app_server):
|
|
s, _ = _session()
|
|
title = f"no-question-{int(time.time() * 1000)}"
|
|
post_uid = _create_post(s, title, "", ["Tabs", "Spaces"])
|
|
assert get_table("polls").find_one(post_uid=post_uid) is None
|
|
|
|
|
|
def test_vote_records_choice(app_server):
|
|
s, _ = _session()
|
|
title = f"vote-{int(time.time() * 1000)}"
|
|
post_uid = _create_post(s, title, "Pick one", ["A", "B"])
|
|
poll, options = _poll_for(post_uid)
|
|
r = s.post(f"{BASE_URL}/polls/{poll['uid']}/vote", data={"option_uid": options[0]["uid"]}, headers=AJAX)
|
|
payload = r.json()
|
|
assert payload["total"] == 1
|
|
assert payload["my_choice"] == options[0]["uid"]
|
|
assert get_table("poll_votes").count(poll_uid=poll["uid"]) == 1
|
|
|
|
|
|
def test_switch_vote_moves_choice(app_server):
|
|
s, _ = _session()
|
|
title = f"switch-{int(time.time() * 1000)}"
|
|
post_uid = _create_post(s, title, "Pick one", ["A", "B"])
|
|
poll, options = _poll_for(post_uid)
|
|
s.post(f"{BASE_URL}/polls/{poll['uid']}/vote", data={"option_uid": options[0]["uid"]}, headers=AJAX)
|
|
r = s.post(f"{BASE_URL}/polls/{poll['uid']}/vote", data={"option_uid": options[1]["uid"]}, headers=AJAX)
|
|
payload = r.json()
|
|
assert payload["my_choice"] == options[1]["uid"]
|
|
assert payload["total"] == 1
|
|
|
|
|
|
def test_repeat_same_option_retracts(app_server):
|
|
s, _ = _session()
|
|
title = f"retract-{int(time.time() * 1000)}"
|
|
post_uid = _create_post(s, title, "Pick one", ["A", "B"])
|
|
poll, options = _poll_for(post_uid)
|
|
s.post(f"{BASE_URL}/polls/{poll['uid']}/vote", data={"option_uid": options[0]["uid"]}, headers=AJAX)
|
|
r = s.post(f"{BASE_URL}/polls/{poll['uid']}/vote", data={"option_uid": options[0]["uid"]}, headers=AJAX)
|
|
payload = r.json()
|
|
assert payload["my_choice"] is None
|
|
assert payload["total"] == 0
|
|
assert get_table("poll_votes").count(poll_uid=poll["uid"]) == 0
|
|
|
|
|
|
def test_vote_invalid_option_returns_400(app_server):
|
|
s, _ = _session()
|
|
title = f"badopt-{int(time.time() * 1000)}"
|
|
post_uid = _create_post(s, title, "Pick one", ["A", "B"])
|
|
poll, _ = _poll_for(post_uid)
|
|
r = s.post(f"{BASE_URL}/polls/{poll['uid']}/vote", data={"option_uid": "not-a-real-option"}, headers=AJAX)
|
|
assert r.status_code == 400
|
|
|
|
|
|
def test_vote_requires_login(app_server):
|
|
s, _ = _session()
|
|
title = f"authvote-{int(time.time() * 1000)}"
|
|
post_uid = _create_post(s, title, "Pick one", ["A", "B"])
|
|
poll, options = _poll_for(post_uid)
|
|
anon = requests.Session()
|
|
r = anon.post(f"{BASE_URL}/polls/{poll['uid']}/vote", data={"option_uid": options[0]["uid"]}, headers=AJAX, allow_redirects=False)
|
|
assert r.status_code == 303
|
|
assert get_table("poll_votes").count(poll_uid=poll["uid"]) == 0
|
|
|
|
|
|
def test_feed_shows_poll_results_without_voting(app_server):
|
|
s, _ = _session()
|
|
title = f"feedpoll-{int(time.time() * 1000)}"
|
|
_create_post(s, title, "Visible results question?", ["Yes", "No"])
|
|
html = s.get(f"{BASE_URL}/feed").text
|
|
assert "Visible results question?" in html
|
|
assert "poll-option-bar" in html
|