mirror of
https://gitlab.archlinux.org/archlinux/aurweb.git
synced 2025-02-03 10:43:03 +01:00
d2d47254b4
TU_VoteInfo -> VoteInfo TU_Votes -> Votes TU_VoteInfo.ActiveTUs -> VoteInfo.ActiveUsers script: tuvotereminder -> votereminder Signed-off-by: moson <moson@archlinux.org>
114 lines
3.1 KiB
Python
114 lines
3.1 KiB
Python
from typing import Tuple
|
|
|
|
import pytest
|
|
|
|
from aurweb import config, db, time
|
|
from aurweb.models import User, Vote, VoteInfo
|
|
from aurweb.models.account_type import PACKAGE_MAINTAINER_ID
|
|
from aurweb.scripts import votereminder as reminder
|
|
from aurweb.testing.email import Email
|
|
|
|
aur_location = config.get("options", "aur_location")
|
|
|
|
|
|
def create_vote(user: User, voteinfo: VoteInfo) -> Vote:
|
|
with db.begin():
|
|
vote = db.create(Vote, User=user, VoteID=voteinfo.ID)
|
|
return vote
|
|
|
|
|
|
def create_user(username: str, type_id: int):
|
|
with db.begin():
|
|
user = db.create(
|
|
User,
|
|
AccountTypeID=type_id,
|
|
Username=username,
|
|
Email=f"{username}@example.org",
|
|
Passwd=str(),
|
|
)
|
|
return user
|
|
|
|
|
|
def email_pieces(voteinfo: VoteInfo) -> Tuple[str, str]:
|
|
"""
|
|
Return a (subject, content) tuple based on voteinfo.ID
|
|
|
|
:param voteinfo: VoteInfo instance
|
|
:return: tuple(subject, content)
|
|
"""
|
|
subject = f"Package Maintainer Vote Reminder: Proposal {voteinfo.ID}"
|
|
content = (
|
|
f"Please remember to cast your vote on proposal {voteinfo.ID} "
|
|
f"[1]. The voting period\nends in less than 48 hours.\n\n"
|
|
f"[1] {aur_location}/package-maintainer/?id={voteinfo.ID}"
|
|
)
|
|
return subject, content
|
|
|
|
|
|
@pytest.fixture
|
|
def user(db_test) -> User:
|
|
yield create_user("test", PACKAGE_MAINTAINER_ID)
|
|
|
|
|
|
@pytest.fixture
|
|
def user2() -> User:
|
|
yield create_user("test2", PACKAGE_MAINTAINER_ID)
|
|
|
|
|
|
@pytest.fixture
|
|
def user3() -> User:
|
|
yield create_user("test3", PACKAGE_MAINTAINER_ID)
|
|
|
|
|
|
@pytest.fixture
|
|
def voteinfo(user: User) -> VoteInfo:
|
|
now = time.utcnow()
|
|
start = config.getint("votereminder", "range_start")
|
|
with db.begin():
|
|
voteinfo = db.create(
|
|
VoteInfo,
|
|
Agenda="Lorem ipsum.",
|
|
User=user.Username,
|
|
End=(now + start + 1),
|
|
Quorum=0.00,
|
|
Submitter=user,
|
|
Submitted=0,
|
|
)
|
|
yield voteinfo
|
|
|
|
|
|
def test_vote_reminders(user: User, user2: User, user3: User, voteinfo: VoteInfo):
|
|
reminder.main()
|
|
assert Email.count() == 3
|
|
|
|
emails = [Email(i).parse() for i in range(1, 4)]
|
|
subject, content = email_pieces(voteinfo)
|
|
expectations = [
|
|
# (to, content)
|
|
(user.Email, subject, content),
|
|
(user2.Email, subject, content),
|
|
(user3.Email, subject, content),
|
|
]
|
|
for i, element in enumerate(expectations):
|
|
email, subject, content = element
|
|
assert emails[i].headers.get("To") == email
|
|
assert emails[i].headers.get("Subject") == subject
|
|
assert emails[i].body == content
|
|
|
|
|
|
def test_vote_reminders_only_unvoted(
|
|
user: User, user2: User, user3: User, voteinfo: VoteInfo
|
|
):
|
|
# Vote with user2 and user3; leaving only user to be notified.
|
|
create_vote(user2, voteinfo)
|
|
create_vote(user3, voteinfo)
|
|
|
|
reminder.main()
|
|
assert Email.count() == 1
|
|
|
|
email = Email(1).parse()
|
|
assert email.headers.get("To") == user.Email
|
|
|
|
subject, content = email_pieces(voteinfo)
|
|
assert email.headers.get("Subject") == subject
|
|
assert email.body == content
|