mirror of
https://gitlab.archlinux.org/archlinux/aurweb.git
synced 2025-02-03 10:43:03 +01:00
change(notify): converted to use aurweb.db ORM
- Removed notify sharness test Signed-off-by: Kevin Morris <kevr@0cost.org>
This commit is contained in:
Kevin Morris
parent
9fb1fbe32c
commit
d8e3ca1abb
7 changed files with 934 additions and 627 deletions
|
|
@ -1,431 +0,0 @@
|
|||
#!/bin/sh
|
||||
|
||||
test_description='notify tests'
|
||||
|
||||
. "$(dirname "$0")/setup.sh"
|
||||
|
||||
test_expect_success 'Test out-of-date notifications.' '
|
||||
cat <<-EOD | sqlite3 aur.db &&
|
||||
/* Use package base IDs which can be distinguished from user IDs. */
|
||||
INSERT INTO PackageBases (ID, Name, MaintainerUID, SubmittedTS, ModifiedTS, FlaggerComment) VALUES (1001, "foobar", 1, 0, 0, "This is a test OOD comment.");
|
||||
INSERT INTO PackageBases (ID, Name, MaintainerUID, SubmittedTS, ModifiedTS, FlaggerComment) VALUES (1002, "foobar2", 2, 0, 0, "");
|
||||
INSERT INTO PackageBases (ID, Name, MaintainerUID, SubmittedTS, ModifiedTS, FlaggerComment) VALUES (1003, "foobar3", NULL, 0, 0, "");
|
||||
INSERT INTO PackageBases (ID, Name, MaintainerUID, SubmittedTS, ModifiedTS, FlaggerComment) VALUES (1004, "foobar4", 1, 0, 0, "");
|
||||
INSERT INTO PackageComaintainers (PackageBaseID, UsersID, Priority) VALUES (1001, 2, 1);
|
||||
INSERT INTO PackageComaintainers (PackageBaseID, UsersID, Priority) VALUES (1001, 4, 2);
|
||||
INSERT INTO PackageComaintainers (PackageBaseID, UsersID, Priority) VALUES (1002, 3, 1);
|
||||
INSERT INTO PackageComaintainers (PackageBaseID, UsersID, Priority) VALUES (1002, 5, 2);
|
||||
INSERT INTO PackageComaintainers (PackageBaseID, UsersID, Priority) VALUES (1003, 4, 1);
|
||||
EOD
|
||||
>sendmail.out &&
|
||||
cover "$NOTIFY" flag 1 1001 &&
|
||||
cat <<-EOD >expected &&
|
||||
Subject: AUR Out-of-date Notification for foobar
|
||||
To: tu@localhost
|
||||
Subject: AUR Out-of-date Notification for foobar
|
||||
To: user2@localhost
|
||||
Subject: AUR Out-of-date Notification for foobar
|
||||
To: user@localhost
|
||||
EOD
|
||||
grep "^\(Subject\|To\)" sendmail.out >sendmail.parts &&
|
||||
test_cmp sendmail.parts expected &&
|
||||
cat <<-EOD | sqlite3 aur.db
|
||||
DELETE FROM PackageComaintainers;
|
||||
EOD
|
||||
'
|
||||
|
||||
test_expect_success 'Test subject and body of reset key notifications.' '
|
||||
cat <<-EOD | sqlite3 aur.db &&
|
||||
UPDATE Users SET ResetKey = "12345678901234567890123456789012" WHERE ID = 1;
|
||||
EOD
|
||||
>sendmail.out &&
|
||||
cover "$NOTIFY" send-resetkey 1 &&
|
||||
grep ^Subject: sendmail.out >actual &&
|
||||
cat <<-EOD >expected &&
|
||||
Subject: AUR Password Reset
|
||||
EOD
|
||||
test_cmp actual expected &&
|
||||
sed -n "/^\$/,\$p" sendmail.out | base64 -d >actual &&
|
||||
echo >>actual &&
|
||||
cat <<-EOD >expected &&
|
||||
A password reset request was submitted for the account user associated
|
||||
with your email address. If you wish to reset your password follow the
|
||||
link [1] below, otherwise ignore this message and nothing will happen.
|
||||
|
||||
[1] https://aur.archlinux.org/passreset/?resetkey=12345678901234567890123456789012
|
||||
EOD
|
||||
test_cmp actual expected
|
||||
'
|
||||
|
||||
test_expect_success 'Test subject and body of welcome notifications.' '
|
||||
cat <<-EOD | sqlite3 aur.db &&
|
||||
UPDATE Users SET ResetKey = "12345678901234567890123456789012" WHERE ID = 1;
|
||||
EOD
|
||||
>sendmail.out &&
|
||||
cover "$NOTIFY" welcome 1 &&
|
||||
grep ^Subject: sendmail.out >actual &&
|
||||
cat <<-EOD >expected &&
|
||||
Subject: Welcome to the Arch User Repository
|
||||
EOD
|
||||
test_cmp actual expected &&
|
||||
sed -n "/^\$/,\$p" sendmail.out | base64 -d >actual &&
|
||||
echo >>actual &&
|
||||
cat <<-EOD >expected &&
|
||||
Welcome to the Arch User Repository! In order to set an initial
|
||||
password for your new account, please click the link [1] below. If the
|
||||
link does not work, try copying and pasting it into your browser.
|
||||
|
||||
[1] https://aur.archlinux.org/passreset/?resetkey=12345678901234567890123456789012
|
||||
EOD
|
||||
test_cmp actual expected
|
||||
'
|
||||
|
||||
test_expect_success 'Test subject and body of comment notifications.' '
|
||||
cat <<-EOD | sqlite3 aur.db &&
|
||||
/* Use package comments IDs which can be distinguished from other IDs. */
|
||||
INSERT INTO PackageComments (ID, PackageBaseID, UsersID, Comments, RenderedComment) VALUES (2001, 1001, 1, "This is a test comment.", "This is a test comment.");
|
||||
INSERT INTO PackageNotifications (PackageBaseID, UserID) VALUES (1001, 2);
|
||||
UPDATE Users SET CommentNotify = 1 WHERE ID = 2;
|
||||
EOD
|
||||
>sendmail.out &&
|
||||
cover "$NOTIFY" comment 1 1001 2001 &&
|
||||
grep ^Subject: sendmail.out >actual &&
|
||||
cat <<-EOD >expected &&
|
||||
Subject: AUR Comment for foobar
|
||||
EOD
|
||||
test_cmp actual expected &&
|
||||
sed -n "/^\$/,\$p" sendmail.out | base64 -d >actual &&
|
||||
echo >>actual &&
|
||||
cat <<-EOD >expected &&
|
||||
user [1] added the following comment to foobar [2]:
|
||||
|
||||
This is a test comment.
|
||||
|
||||
--
|
||||
If you no longer wish to receive notifications about this package,
|
||||
please go to the package page [2] and select "Disable notifications".
|
||||
|
||||
[1] https://aur.archlinux.org/account/user/
|
||||
[2] https://aur.archlinux.org/pkgbase/foobar/
|
||||
EOD
|
||||
test_cmp actual expected
|
||||
'
|
||||
|
||||
test_expect_success 'Test subject and body of update notifications.' '
|
||||
cat <<-EOD | sqlite3 aur.db &&
|
||||
UPDATE Users SET UpdateNotify = 1 WHERE ID = 2;
|
||||
EOD
|
||||
>sendmail.out &&
|
||||
cover "$NOTIFY" update 1 1001 &&
|
||||
grep ^Subject: sendmail.out >actual &&
|
||||
cat <<-EOD >expected &&
|
||||
Subject: AUR Package Update: foobar
|
||||
EOD
|
||||
test_cmp actual expected &&
|
||||
sed -n "/^\$/,\$p" sendmail.out | base64 -d >actual &&
|
||||
echo >>actual &&
|
||||
cat <<-EOD >expected &&
|
||||
user [1] pushed a new commit to foobar [2].
|
||||
|
||||
--
|
||||
If you no longer wish to receive notifications about this package,
|
||||
please go to the package page [2] and select "Disable notifications".
|
||||
|
||||
[1] https://aur.archlinux.org/account/user/
|
||||
[2] https://aur.archlinux.org/pkgbase/foobar/
|
||||
EOD
|
||||
test_cmp actual expected
|
||||
'
|
||||
|
||||
test_expect_success 'Test subject and body of out-of-date notifications.' '
|
||||
>sendmail.out &&
|
||||
cover "$NOTIFY" flag 1 1001 &&
|
||||
grep ^Subject: sendmail.out >actual &&
|
||||
cat <<-EOD >expected &&
|
||||
Subject: AUR Out-of-date Notification for foobar
|
||||
EOD
|
||||
test_cmp actual expected &&
|
||||
sed -n "/^\$/,\$p" sendmail.out | base64 -d >actual &&
|
||||
echo >>actual &&
|
||||
cat <<-EOD >expected &&
|
||||
Your package foobar [1] has been flagged out-of-date by user [2]:
|
||||
|
||||
This is a test OOD comment.
|
||||
|
||||
[1] https://aur.archlinux.org/pkgbase/foobar/
|
||||
[2] https://aur.archlinux.org/account/user/
|
||||
EOD
|
||||
test_cmp actual expected
|
||||
'
|
||||
|
||||
test_expect_success 'Test subject and body of adopt notifications.' '
|
||||
>sendmail.out &&
|
||||
cover "$NOTIFY" adopt 1 1001 &&
|
||||
grep ^Subject: sendmail.out >actual &&
|
||||
cat <<-EOD >expected &&
|
||||
Subject: AUR Ownership Notification for foobar
|
||||
EOD
|
||||
test_cmp actual expected &&
|
||||
sed -n "/^\$/,\$p" sendmail.out | base64 -d >actual &&
|
||||
echo >>actual &&
|
||||
cat <<-EOD >expected &&
|
||||
The package foobar [1] was adopted by user [2].
|
||||
|
||||
[1] https://aur.archlinux.org/pkgbase/foobar/
|
||||
[2] https://aur.archlinux.org/account/user/
|
||||
EOD
|
||||
test_cmp actual expected
|
||||
'
|
||||
|
||||
test_expect_success 'Test subject and body of disown notifications.' '
|
||||
>sendmail.out &&
|
||||
cover "$NOTIFY" disown 1 1001 &&
|
||||
grep ^Subject: sendmail.out >actual &&
|
||||
cat <<-EOD >expected &&
|
||||
Subject: AUR Ownership Notification for foobar
|
||||
EOD
|
||||
test_cmp actual expected &&
|
||||
sed -n "/^\$/,\$p" sendmail.out | base64 -d >actual &&
|
||||
echo >>actual &&
|
||||
cat <<-EOD >expected &&
|
||||
The package foobar [1] was disowned by user [2].
|
||||
|
||||
[1] https://aur.archlinux.org/pkgbase/foobar/
|
||||
[2] https://aur.archlinux.org/account/user/
|
||||
EOD
|
||||
test_cmp actual expected
|
||||
'
|
||||
|
||||
test_expect_success 'Test subject and body of co-maintainer addition notifications.' '
|
||||
>sendmail.out &&
|
||||
cover "$NOTIFY" comaintainer-add 1 1001 &&
|
||||
grep ^Subject: sendmail.out >actual &&
|
||||
cat <<-EOD >expected &&
|
||||
Subject: AUR Co-Maintainer Notification for foobar
|
||||
EOD
|
||||
test_cmp actual expected &&
|
||||
sed -n "/^\$/,\$p" sendmail.out | base64 -d >actual &&
|
||||
echo >>actual &&
|
||||
cat <<-EOD >expected &&
|
||||
You were added to the co-maintainer list of foobar [1].
|
||||
|
||||
[1] https://aur.archlinux.org/pkgbase/foobar/
|
||||
EOD
|
||||
test_cmp actual expected
|
||||
'
|
||||
|
||||
test_expect_success 'Test subject and body of co-maintainer removal notifications.' '
|
||||
>sendmail.out &&
|
||||
cover "$NOTIFY" comaintainer-remove 1 1001 &&
|
||||
grep ^Subject: sendmail.out >actual &&
|
||||
cat <<-EOD >expected &&
|
||||
Subject: AUR Co-Maintainer Notification for foobar
|
||||
EOD
|
||||
test_cmp actual expected &&
|
||||
sed -n "/^\$/,\$p" sendmail.out | base64 -d >actual &&
|
||||
echo >>actual &&
|
||||
cat <<-EOD >expected &&
|
||||
You were removed from the co-maintainer list of foobar [1].
|
||||
|
||||
[1] https://aur.archlinux.org/pkgbase/foobar/
|
||||
EOD
|
||||
test_cmp actual expected
|
||||
'
|
||||
|
||||
test_expect_success 'Test subject and body of delete notifications.' '
|
||||
>sendmail.out &&
|
||||
cover "$NOTIFY" delete 1 1001 &&
|
||||
grep ^Subject: sendmail.out >actual &&
|
||||
cat <<-EOD >expected &&
|
||||
Subject: AUR Package deleted: foobar
|
||||
EOD
|
||||
test_cmp actual expected &&
|
||||
sed -n "/^\$/,\$p" sendmail.out | base64 -d >actual &&
|
||||
echo >>actual &&
|
||||
cat <<-EOD >expected &&
|
||||
user [1] deleted foobar [2].
|
||||
|
||||
You will no longer receive notifications about this package.
|
||||
|
||||
[1] https://aur.archlinux.org/account/user/
|
||||
[2] https://aur.archlinux.org/pkgbase/foobar/
|
||||
EOD
|
||||
test_cmp actual expected
|
||||
'
|
||||
|
||||
test_expect_success 'Test subject and body of merge notifications.' '
|
||||
>sendmail.out &&
|
||||
cover "$NOTIFY" delete 1 1001 1002 &&
|
||||
grep ^Subject: sendmail.out >actual &&
|
||||
cat <<-EOD >expected &&
|
||||
Subject: AUR Package deleted: foobar
|
||||
EOD
|
||||
test_cmp actual expected &&
|
||||
sed -n "/^\$/,\$p" sendmail.out | base64 -d >actual &&
|
||||
echo >>actual &&
|
||||
cat <<-EOD >expected &&
|
||||
user [1] merged foobar [2] into foobar2 [3].
|
||||
|
||||
--
|
||||
If you no longer wish receive notifications about the new package,
|
||||
please go to [3] and click "Disable notifications".
|
||||
|
||||
[1] https://aur.archlinux.org/account/user/
|
||||
[2] https://aur.archlinux.org/pkgbase/foobar/
|
||||
[3] https://aur.archlinux.org/pkgbase/foobar2/
|
||||
EOD
|
||||
test_cmp actual expected
|
||||
'
|
||||
|
||||
test_expect_success 'Test Cc, subject and body of request open notifications.' '
|
||||
cat <<-EOD | sqlite3 aur.db &&
|
||||
/* Use package request IDs which can be distinguished from other IDs. */
|
||||
INSERT INTO PackageRequests (ID, PackageBaseID, PackageBaseName, UsersID, ReqTypeID, Comments, ClosureComment) VALUES (3001, 1001, "foobar", 2, 1, "This is a request test comment.", "");
|
||||
EOD
|
||||
>sendmail.out &&
|
||||
cover "$NOTIFY" request-open 1 3001 orphan 1001 &&
|
||||
grep ^Cc: sendmail.out >actual &&
|
||||
cat <<-EOD >expected &&
|
||||
Cc: user@localhost, tu@localhost
|
||||
EOD
|
||||
test_cmp actual expected &&
|
||||
grep ^Subject: sendmail.out >actual &&
|
||||
cat <<-EOD >expected &&
|
||||
Subject: [PRQ#3001] Orphan Request for foobar
|
||||
EOD
|
||||
test_cmp actual expected &&
|
||||
sed -n "/^\$/,\$p" sendmail.out | base64 -d >actual &&
|
||||
echo >>actual &&
|
||||
cat <<-EOD >expected &&
|
||||
user [1] filed an orphan request for foobar [2]:
|
||||
|
||||
This is a request test comment.
|
||||
|
||||
[1] https://aur.archlinux.org/account/user/
|
||||
[2] https://aur.archlinux.org/pkgbase/foobar/
|
||||
EOD
|
||||
test_cmp actual expected
|
||||
'
|
||||
|
||||
test_expect_success 'Test subject and body of request open notifications for merge requests.' '
|
||||
>sendmail.out &&
|
||||
cover "$NOTIFY" request-open 1 3001 merge 1001 foobar2 &&
|
||||
grep ^Subject: sendmail.out >actual &&
|
||||
cat <<-EOD >expected &&
|
||||
Subject: [PRQ#3001] Merge Request for foobar
|
||||
EOD
|
||||
test_cmp actual expected &&
|
||||
sed -n "/^\$/,\$p" sendmail.out | base64 -d >actual &&
|
||||
echo >>actual &&
|
||||
cat <<-EOD >expected &&
|
||||
user [1] filed a request to merge foobar [2] into foobar2 [3]:
|
||||
|
||||
This is a request test comment.
|
||||
|
||||
[1] https://aur.archlinux.org/account/user/
|
||||
[2] https://aur.archlinux.org/pkgbase/foobar/
|
||||
[3] https://aur.archlinux.org/pkgbase/foobar2/
|
||||
EOD
|
||||
test_cmp actual expected
|
||||
'
|
||||
|
||||
test_expect_success 'Test Cc, subject and body of request close notifications.' '
|
||||
>sendmail.out &&
|
||||
cover "$NOTIFY" request-close 1 3001 accepted &&
|
||||
grep ^Cc: sendmail.out >actual &&
|
||||
cat <<-EOD >expected &&
|
||||
Cc: user@localhost, tu@localhost
|
||||
EOD
|
||||
test_cmp actual expected &&
|
||||
grep ^Subject: sendmail.out >actual &&
|
||||
cat <<-EOD >expected &&
|
||||
Subject: [PRQ#3001] Deletion Request for foobar Accepted
|
||||
EOD
|
||||
test_cmp actual expected &&
|
||||
sed -n "/^\$/,\$p" sendmail.out | base64 -d >actual &&
|
||||
echo >>actual &&
|
||||
cat <<-EOD >expected &&
|
||||
Request #3001 has been accepted by user [1].
|
||||
|
||||
[1] https://aur.archlinux.org/account/user/
|
||||
EOD
|
||||
test_cmp actual expected
|
||||
'
|
||||
|
||||
test_expect_success 'Test subject and body of request close notifications (auto-accept).' '
|
||||
>sendmail.out &&
|
||||
cover "$NOTIFY" request-close 0 3001 accepted &&
|
||||
grep ^Subject: sendmail.out >actual &&
|
||||
cat <<-EOD >expected &&
|
||||
Subject: [PRQ#3001] Deletion Request for foobar Accepted
|
||||
EOD
|
||||
test_cmp actual expected &&
|
||||
sed -n "/^\$/,\$p" sendmail.out | base64 -d >actual &&
|
||||
echo >>actual &&
|
||||
cat <<-EOD >expected &&
|
||||
Request #3001 has been accepted automatically by the Arch User
|
||||
Repository package request system.
|
||||
EOD
|
||||
test_cmp actual expected
|
||||
'
|
||||
|
||||
test_expect_success 'Test Cc of request close notification with co-maintainer.' '
|
||||
cat <<-EOD | sqlite3 aur.db &&
|
||||
/* Use package base IDs which can be distinguished from user IDs. */
|
||||
INSERT INTO PackageComaintainers (PackageBaseID, UsersID, Priority) VALUES (1001, 3, 1);
|
||||
EOD
|
||||
>sendmail.out &&
|
||||
"$NOTIFY" request-close 0 3001 accepted &&
|
||||
grep ^Cc: sendmail.out >actual &&
|
||||
cat <<-EOD >expected &&
|
||||
Cc: user@localhost, tu@localhost, dev@localhost
|
||||
EOD
|
||||
test_cmp actual expected &&
|
||||
cat <<-EOD | sqlite3 aur.db
|
||||
DELETE FROM PackageComaintainers;
|
||||
EOD
|
||||
'
|
||||
|
||||
test_expect_success 'Test subject and body of request close notifications with closure comment.' '
|
||||
cat <<-EOD | sqlite3 aur.db &&
|
||||
UPDATE PackageRequests SET ClosureComment = "This is a test closure comment." WHERE ID = 3001;
|
||||
EOD
|
||||
>sendmail.out &&
|
||||
cover "$NOTIFY" request-close 1 3001 accepted &&
|
||||
grep ^Subject: sendmail.out >actual &&
|
||||
cat <<-EOD >expected &&
|
||||
Subject: [PRQ#3001] Deletion Request for foobar Accepted
|
||||
EOD
|
||||
test_cmp actual expected &&
|
||||
sed -n "/^\$/,\$p" sendmail.out | base64 -d >actual &&
|
||||
echo >>actual &&
|
||||
cat <<-EOD >expected &&
|
||||
Request #3001 has been accepted by user [1]:
|
||||
|
||||
This is a test closure comment.
|
||||
|
||||
[1] https://aur.archlinux.org/account/user/
|
||||
EOD
|
||||
test_cmp actual expected
|
||||
'
|
||||
|
||||
test_expect_success 'Test subject and body of TU vote reminders.' '
|
||||
>sendmail.out &&
|
||||
cover "$NOTIFY" tu-vote-reminder 1 &&
|
||||
grep ^Subject: sendmail.out | head -1 >actual &&
|
||||
cat <<-EOD >expected &&
|
||||
Subject: TU Vote Reminder: Proposal 1
|
||||
EOD
|
||||
test_cmp actual expected &&
|
||||
sed -n "/^\$/,\$p" sendmail.out | head -4 | base64 -d >actual &&
|
||||
echo >>actual &&
|
||||
cat <<-EOD >expected &&
|
||||
Please remember to cast your vote on proposal 1 [1]. The voting period
|
||||
ends in less than 48 hours.
|
||||
|
||||
[1] https://aur.archlinux.org/tu/?id=1
|
||||
EOD
|
||||
test_cmp actual expected
|
||||
'
|
||||
|
||||
test_done
|
||||
643
test/test_notify.py
Normal file
643
test/test_notify.py
Normal file
|
|
@ -0,0 +1,643 @@
|
|||
from datetime import datetime
|
||||
from typing import List
|
||||
from unittest import mock
|
||||
|
||||
import pytest
|
||||
|
||||
from aurweb import config, db, models
|
||||
from aurweb.models import Package, PackageBase, PackageRequest, User
|
||||
from aurweb.models.account_type import TRUSTED_USER_ID, USER_ID
|
||||
from aurweb.models.request_type import ORPHAN_ID
|
||||
from aurweb.scripts import notify, rendercomment
|
||||
from aurweb.testing.email import Email
|
||||
from aurweb.testing.smtp import FakeSMTP, FakeSMTP_SSL
|
||||
|
||||
aur_location = config.get("options", "aur_location")
|
||||
aur_request_ml = config.get("options", "aur_request_ml")
|
||||
|
||||
|
||||
@pytest.fixture(autouse=True)
|
||||
def setup(db_test):
|
||||
return
|
||||
|
||||
|
||||
@pytest.fixture
|
||||
def user() -> User:
|
||||
with db.begin():
|
||||
user = db.create(User, Username="test", Email="test@example.org",
|
||||
Passwd=str(), AccountTypeID=USER_ID)
|
||||
yield user
|
||||
|
||||
|
||||
@pytest.fixture
|
||||
def user1() -> User:
|
||||
with db.begin():
|
||||
user1 = db.create(User, Username="user1", Email="user1@example.org",
|
||||
Passwd=str(), AccountTypeID=USER_ID)
|
||||
yield user1
|
||||
|
||||
|
||||
@pytest.fixture
|
||||
def user2() -> User:
|
||||
with db.begin():
|
||||
user2 = db.create(User, Username="user2", Email="user2@example.org",
|
||||
Passwd=str(), AccountTypeID=USER_ID)
|
||||
yield user2
|
||||
|
||||
|
||||
@pytest.fixture
|
||||
def pkgbases(user: User) -> List[PackageBase]:
|
||||
now = int(datetime.utcnow().timestamp())
|
||||
|
||||
output = []
|
||||
with db.begin():
|
||||
for i in range(5):
|
||||
output.append(
|
||||
db.create(PackageBase, Name=f"pkgbase_{i}",
|
||||
Maintainer=user, SubmittedTS=now,
|
||||
ModifiedTS=now))
|
||||
db.create(models.PackageNotification, PackageBase=output[-1],
|
||||
User=user)
|
||||
yield output
|
||||
|
||||
|
||||
@pytest.fixture
|
||||
def pkgreq(user2: User, pkgbases: List[PackageBase]):
|
||||
pkgbase = pkgbases[0]
|
||||
with db.begin():
|
||||
pkgreq_ = db.create(PackageRequest, PackageBase=pkgbase,
|
||||
PackageBaseName=pkgbase.Name, User=user2,
|
||||
ReqTypeID=ORPHAN_ID,
|
||||
Comments="This is a request test comment.",
|
||||
ClosureComment=str())
|
||||
yield pkgreq_
|
||||
|
||||
|
||||
@pytest.fixture
|
||||
def packages(pkgbases: List[PackageBase]) -> List[Package]:
|
||||
output = []
|
||||
with db.begin():
|
||||
for i, pkgbase in enumerate(pkgbases):
|
||||
output.append(
|
||||
db.create(Package, PackageBase=pkgbase,
|
||||
Name=f"pkg_{i}", Version=f"{i}.0"))
|
||||
yield output
|
||||
|
||||
|
||||
def test_out_of_date(user: User, user1: User, user2: User,
|
||||
pkgbases: List[PackageBase]):
|
||||
pkgbase = pkgbases[0]
|
||||
# Create two comaintainers. We'll pass the maintainer uid to
|
||||
# FlagNotification, so we should expect to get two emails.
|
||||
with db.begin():
|
||||
db.create(models.PackageComaintainer,
|
||||
PackageBase=pkgbase, User=user1, Priority=1)
|
||||
db.create(models.PackageComaintainer,
|
||||
PackageBase=pkgbase, User=user2, Priority=2)
|
||||
|
||||
# Send the notification for pkgbases[0].
|
||||
notif = notify.FlagNotification(user.ID, pkgbases[0].ID)
|
||||
notif.send()
|
||||
|
||||
# Should've gotten three emails: maintainer + the two comaintainers.
|
||||
assert Email.count() == 3
|
||||
|
||||
# Comaintainer 1.
|
||||
first = Email(1).parse()
|
||||
assert first.headers.get("To") == user1.Email
|
||||
|
||||
expected = f"AUR Out-of-date Notification for {pkgbase.Name}"
|
||||
assert first.headers.get("Subject") == expected
|
||||
|
||||
# Comaintainer 2.
|
||||
second = Email(2).parse()
|
||||
assert second.headers.get("To") == user2.Email
|
||||
|
||||
# Maintainer.
|
||||
third = Email(3).parse()
|
||||
assert third.headers.get("To") == user.Email
|
||||
|
||||
|
||||
def test_reset(user: User):
|
||||
with db.begin():
|
||||
user.ResetKey = "12345678901234567890123456789012"
|
||||
|
||||
notif = notify.ResetKeyNotification(user.ID)
|
||||
notif.send()
|
||||
assert Email.count() == 1
|
||||
|
||||
email = Email(1).parse()
|
||||
expected = "AUR Password Reset"
|
||||
assert email.headers.get("Subject") == expected
|
||||
|
||||
expected = f"""\
|
||||
A password reset request was submitted for the account test associated
|
||||
with your email address. If you wish to reset your password follow the
|
||||
link [1] below, otherwise ignore this message and nothing will happen.
|
||||
|
||||
[1] {aur_location}/passreset/?resetkey=12345678901234567890123456789012\
|
||||
"""
|
||||
assert email.body == expected
|
||||
|
||||
|
||||
def test_welcome(user: User):
|
||||
with db.begin():
|
||||
user.ResetKey = "12345678901234567890123456789012"
|
||||
|
||||
notif = notify.WelcomeNotification(user.ID)
|
||||
notif.send()
|
||||
assert Email.count() == 1
|
||||
|
||||
email = Email(1).parse()
|
||||
expected = "Welcome to the Arch User Repository"
|
||||
assert email.headers.get("Subject") == expected
|
||||
|
||||
expected = f"""\
|
||||
Welcome to the Arch User Repository! In order to set an initial
|
||||
password for your new account, please click the link [1] below. If the
|
||||
link does not work, try copying and pasting it into your browser.
|
||||
|
||||
[1] {aur_location}/passreset/?resetkey=12345678901234567890123456789012\
|
||||
"""
|
||||
assert email.body == expected
|
||||
|
||||
|
||||
def test_comment(user: User, user2: User, pkgbases: List[PackageBase]):
|
||||
pkgbase = pkgbases[0]
|
||||
|
||||
with db.begin():
|
||||
comment = db.create(models.PackageComment, PackageBase=pkgbase,
|
||||
User=user2, Comments="This is a test comment.")
|
||||
rendercomment.update_comment_render_fastapi(comment)
|
||||
|
||||
notif = notify.CommentNotification(user2.ID, pkgbase.ID, comment.ID)
|
||||
notif.send()
|
||||
assert Email.count() == 1
|
||||
|
||||
email = Email(1).parse()
|
||||
assert email.headers.get("To") == user.Email
|
||||
expected = f"AUR Comment for {pkgbase.Name}"
|
||||
assert email.headers.get("Subject") == expected
|
||||
|
||||
expected = f"""\
|
||||
{user2.Username} [1] added the following comment to {pkgbase.Name} [2]:
|
||||
|
||||
This is a test comment.
|
||||
|
||||
--
|
||||
If you no longer wish to receive notifications about this package,
|
||||
please go to the package page [2] and select "Disable notifications".
|
||||
|
||||
[1] {aur_location}/account/{user2.Username}/
|
||||
[2] {aur_location}/pkgbase/{pkgbase.Name}/\
|
||||
"""
|
||||
assert expected == email.body
|
||||
|
||||
|
||||
def test_update(user: User, user2: User, pkgbases: List[PackageBase]):
|
||||
pkgbase = pkgbases[0]
|
||||
with db.begin():
|
||||
user.UpdateNotify = 1
|
||||
|
||||
notif = notify.UpdateNotification(user2.ID, pkgbase.ID)
|
||||
notif.send()
|
||||
assert Email.count() == 1
|
||||
|
||||
email = Email(1).parse()
|
||||
assert email.headers.get("To") == user.Email
|
||||
expected = f"AUR Package Update: {pkgbase.Name}"
|
||||
assert email.headers.get("Subject") == expected
|
||||
|
||||
expected = f"""\
|
||||
{user2.Username} [1] pushed a new commit to {pkgbase.Name} [2].
|
||||
|
||||
--
|
||||
If you no longer wish to receive notifications about this package,
|
||||
please go to the package page [2] and select "Disable notifications".
|
||||
|
||||
[1] {aur_location}/account/{user2.Username}/
|
||||
[2] {aur_location}/pkgbase/{pkgbase.Name}/\
|
||||
"""
|
||||
assert expected == email.body
|
||||
|
||||
|
||||
def test_adopt(user: User, user2: User, pkgbases: List[PackageBase]):
|
||||
pkgbase = pkgbases[0]
|
||||
notif = notify.AdoptNotification(user2.ID, pkgbase.ID)
|
||||
notif.send()
|
||||
assert Email.count() == 1
|
||||
|
||||
email = Email(1).parse()
|
||||
assert email.headers.get("To") == user.Email
|
||||
expected = f"AUR Ownership Notification for {pkgbase.Name}"
|
||||
assert email.headers.get("Subject") == expected
|
||||
|
||||
expected = f"""\
|
||||
The package {pkgbase.Name} [1] was adopted by {user2.Username} [2].
|
||||
|
||||
[1] {aur_location}/pkgbase/{pkgbase.Name}/
|
||||
[2] {aur_location}/account/{user2.Username}/\
|
||||
"""
|
||||
assert email.body == expected
|
||||
|
||||
|
||||
def test_disown(user: User, user2: User, pkgbases: List[PackageBase]):
|
||||
pkgbase = pkgbases[0]
|
||||
notif = notify.DisownNotification(user2.ID, pkgbase.ID)
|
||||
notif.send()
|
||||
assert Email.count() == 1
|
||||
|
||||
email = Email(1).parse()
|
||||
assert email.headers.get("To") == user.Email
|
||||
expected = f"AUR Ownership Notification for {pkgbase.Name}"
|
||||
assert email.headers.get("Subject") == expected
|
||||
|
||||
expected = f"""\
|
||||
The package {pkgbase.Name} [1] was disowned by {user2.Username} [2].
|
||||
|
||||
[1] {aur_location}/pkgbase/{pkgbase.Name}/
|
||||
[2] {aur_location}/account/{user2.Username}/\
|
||||
"""
|
||||
assert email.body == expected
|
||||
|
||||
|
||||
def test_comaintainer_addition(user: User, pkgbases: List[PackageBase]):
|
||||
# TODO: Add this in fastapi code!
|
||||
pkgbase = pkgbases[0]
|
||||
notif = notify.ComaintainerAddNotification(user.ID, pkgbase.ID)
|
||||
notif.send()
|
||||
assert Email.count() == 1
|
||||
|
||||
email = Email(1).parse()
|
||||
assert email.headers.get("To") == user.Email
|
||||
expected = f"AUR Co-Maintainer Notification for {pkgbase.Name}"
|
||||
assert email.headers.get("Subject") == expected
|
||||
|
||||
expected = f"""\
|
||||
You were added to the co-maintainer list of {pkgbase.Name} [1].
|
||||
|
||||
[1] {aur_location}/pkgbase/{pkgbase.Name}/\
|
||||
"""
|
||||
assert email.body == expected
|
||||
|
||||
|
||||
def test_comaintainer_removal(user: User, pkgbases: List[PackageBase]):
|
||||
# TODO: Add this in fastapi code!
|
||||
pkgbase = pkgbases[0]
|
||||
notif = notify.ComaintainerRemoveNotification(user.ID, pkgbase.ID)
|
||||
notif.send()
|
||||
assert Email.count() == 1
|
||||
|
||||
email = Email(1).parse()
|
||||
assert email.headers.get("To") == user.Email
|
||||
expected = f"AUR Co-Maintainer Notification for {pkgbase.Name}"
|
||||
assert email.headers.get("Subject") == expected
|
||||
|
||||
expected = f"""\
|
||||
You were removed from the co-maintainer list of {pkgbase.Name} [1].
|
||||
|
||||
[1] {aur_location}/pkgbase/{pkgbase.Name}/\
|
||||
"""
|
||||
assert email.body == expected
|
||||
|
||||
|
||||
def test_delete(user: User, user2: User, pkgbases: List[PackageBase]):
|
||||
pkgbase = pkgbases[0]
|
||||
notif = notify.DeleteNotification(user2.ID, pkgbase.ID)
|
||||
notif.send()
|
||||
assert Email.count() == 1
|
||||
|
||||
email = Email(1).parse()
|
||||
assert email.headers.get("To") == user.Email
|
||||
expected = f"AUR Package deleted: {pkgbase.Name}"
|
||||
assert email.headers.get("Subject") == expected
|
||||
|
||||
expected = f"""\
|
||||
{user2.Username} [1] deleted {pkgbase.Name} [2].
|
||||
|
||||
You will no longer receive notifications about this package.
|
||||
|
||||
[1] {aur_location}/account/{user2.Username}/
|
||||
[2] {aur_location}/pkgbase/{pkgbase.Name}/\
|
||||
"""
|
||||
assert email.body == expected
|
||||
|
||||
|
||||
def test_merge(user: User, user2: User, pkgbases: List[PackageBase]):
|
||||
source, target = pkgbases[:2]
|
||||
notif = notify.DeleteNotification(user2.ID, source.ID, target.ID)
|
||||
notif.send()
|
||||
assert Email.count() == 1
|
||||
|
||||
email = Email(1).parse()
|
||||
assert email.headers.get("To") == user.Email
|
||||
expected = f"AUR Package deleted: {source.Name}"
|
||||
assert email.headers.get("Subject") == expected
|
||||
|
||||
expected = f"""\
|
||||
{user2.Username} [1] merged {source.Name} [2] into {target.Name} [3].
|
||||
|
||||
--
|
||||
If you no longer wish receive notifications about the new package,
|
||||
please go to [3] and click "Disable notifications".
|
||||
|
||||
[1] {aur_location}/account/{user2.Username}/
|
||||
[2] {aur_location}/pkgbase/{source.Name}/
|
||||
[3] {aur_location}/pkgbase/{target.Name}/\
|
||||
"""
|
||||
assert email.body == expected
|
||||
|
||||
|
||||
def set_tu(users: List[User]) -> User:
|
||||
with db.begin():
|
||||
for user in users:
|
||||
user.AccountTypeID = TRUSTED_USER_ID
|
||||
|
||||
|
||||
def test_open_close_request(user: User, user2: User,
|
||||
pkgreq: PackageRequest,
|
||||
pkgbases: List[PackageBase]):
|
||||
set_tu([user])
|
||||
pkgbase = pkgbases[0]
|
||||
|
||||
# Send an open request notification.
|
||||
notif = notify.RequestOpenNotification(
|
||||
user2.ID, pkgreq.ID, pkgreq.RequestType.Name, pkgbase.ID)
|
||||
notif.send()
|
||||
assert Email.count() == 1
|
||||
|
||||
email = Email(1).parse()
|
||||
assert email.headers.get("To") == aur_request_ml
|
||||
assert email.headers.get("Cc") == ", ".join([user.Email, user2.Email])
|
||||
expected = f"[PRQ#{pkgreq.ID}] Orphan Request for {pkgbase.Name}"
|
||||
assert email.headers.get("Subject") == expected
|
||||
|
||||
expected = f"""\
|
||||
{user2.Username} [1] filed an orphan request for {pkgbase.Name} [2]:
|
||||
|
||||
This is a request test comment.
|
||||
|
||||
[1] {aur_location}/account/{user2.Username}/
|
||||
[2] {aur_location}/pkgbase/{pkgbase.Name}/\
|
||||
"""
|
||||
assert email.body == expected
|
||||
|
||||
# Now send a closure notification on the pkgbase we just opened.
|
||||
notif = notify.RequestCloseNotification(user2.ID, pkgreq.ID, "rejected")
|
||||
notif.send()
|
||||
assert Email.count() == 2
|
||||
|
||||
email = Email(2).parse()
|
||||
assert email.headers.get("To") == aur_request_ml
|
||||
assert email.headers.get("Cc") == ", ".join([user.Email, user2.Email])
|
||||
expected = f"[PRQ#{pkgreq.ID}] Orphan Request for {pkgbase.Name} Rejected"
|
||||
assert email.headers.get("Subject") == expected
|
||||
|
||||
expected = f"""\
|
||||
Request #{pkgreq.ID} has been rejected by {user2.Username} [1].
|
||||
|
||||
[1] {aur_location}/account/{user2.Username}/\
|
||||
"""
|
||||
assert email.body == expected
|
||||
|
||||
# Test auto-accept.
|
||||
notif = notify.RequestCloseNotification(0, pkgreq.ID, "accepted")
|
||||
notif.send()
|
||||
assert Email.count() == 3
|
||||
|
||||
email = Email(3).parse()
|
||||
assert email.headers.get("To") == aur_request_ml
|
||||
assert email.headers.get("Cc") == ", ".join([user.Email, user2.Email])
|
||||
expected = (f"[PRQ#{pkgreq.ID}] Orphan Request for "
|
||||
f"{pkgbase.Name} Accepted")
|
||||
assert email.headers.get("Subject") == expected
|
||||
|
||||
expected = (f"Request #{pkgreq.ID} has been accepted automatically "
|
||||
"by the Arch User Repository\npackage request system.")
|
||||
assert email.body == expected
|
||||
|
||||
|
||||
def test_close_request_auto_accept():
|
||||
pass
|
||||
|
||||
|
||||
def test_close_request_comaintainer_cc(user: User, user2: User,
|
||||
pkgreq: PackageRequest,
|
||||
pkgbases: List[PackageBase]):
|
||||
# TODO: Check this in fastapi code!
|
||||
pkgbase = pkgbases[0]
|
||||
with db.begin():
|
||||
db.create(models.PackageComaintainer, PackageBase=pkgbase,
|
||||
User=user2, Priority=1)
|
||||
|
||||
notif = notify.RequestCloseNotification(0, pkgreq.ID, "accepted")
|
||||
notif.send()
|
||||
assert Email.count() == 1
|
||||
|
||||
email = Email(1).parse()
|
||||
assert email.headers.get("To") == aur_request_ml
|
||||
assert email.headers.get("Cc") == ", ".join([user.Email, user2.Email])
|
||||
|
||||
|
||||
def test_close_request_closure_comment(user: User, user2: User,
|
||||
pkgreq: PackageRequest,
|
||||
pkgbases: List[PackageBase]):
|
||||
pkgbase = pkgbases[0]
|
||||
with db.begin():
|
||||
pkgreq.ClosureComment = "This is a test closure comment."
|
||||
|
||||
notif = notify.RequestCloseNotification(user2.ID, pkgreq.ID, "accepted")
|
||||
notif.send()
|
||||
assert Email.count() == 1
|
||||
|
||||
email = Email(1).parse()
|
||||
assert email.headers.get("To") == aur_request_ml
|
||||
assert email.headers.get("Cc") == ", ".join([user.Email, user2.Email])
|
||||
expected = f"[PRQ#{pkgreq.ID}] Orphan Request for {pkgbase.Name} Accepted"
|
||||
assert email.headers.get("Subject") == expected
|
||||
|
||||
expected = f"""\
|
||||
Request #{pkgreq.ID} has been accepted by {user2.Username} [1]:
|
||||
|
||||
This is a test closure comment.
|
||||
|
||||
[1] {aur_location}/account/{user2.Username}/\
|
||||
"""
|
||||
assert email.body == expected
|
||||
|
||||
|
||||
def test_tu_vote_reminders(user: User):
|
||||
set_tu([user])
|
||||
|
||||
vote_id = 1
|
||||
notif = notify.TUVoteReminderNotification(vote_id)
|
||||
notif.send()
|
||||
assert Email.count() == 1
|
||||
|
||||
email = Email(1).parse()
|
||||
assert email.headers.get("To") == user.Email
|
||||
expected = f"TU Vote Reminder: Proposal {vote_id}"
|
||||
assert email.headers.get("Subject") == expected
|
||||
|
||||
expected = f"""\
|
||||
Please remember to cast your vote on proposal {vote_id} [1]. The voting period
|
||||
ends in less than 48 hours.
|
||||
|
||||
[1] {aur_location}/tu/?id={vote_id}\
|
||||
"""
|
||||
assert email.body == expected
|
||||
|
||||
|
||||
def test_notify_main(user: User):
|
||||
""" Test TU vote reminder through aurweb.notify.main(). """
|
||||
set_tu([user])
|
||||
|
||||
vote_id = 1
|
||||
args = ["aurweb-notify", "tu-vote-reminder", str(vote_id)]
|
||||
with mock.patch("sys.argv", args):
|
||||
notify.main()
|
||||
|
||||
assert Email.count() == 1
|
||||
|
||||
email = Email(1).parse()
|
||||
assert email.headers.get("To") == user.Email
|
||||
expected = f"TU Vote Reminder: Proposal {vote_id}"
|
||||
assert email.headers.get("Subject") == expected
|
||||
|
||||
expected = f"""\
|
||||
Please remember to cast your vote on proposal {vote_id} [1]. The voting period
|
||||
ends in less than 48 hours.
|
||||
|
||||
[1] {aur_location}/tu/?id={vote_id}\
|
||||
"""
|
||||
assert email.body == expected
|
||||
|
||||
|
||||
# Save original config.get; we're going to mock it and need
|
||||
# to be able to fallback when we are not overriding.
|
||||
config_get = config.get
|
||||
|
||||
|
||||
def mock_smtp_config(cls):
|
||||
def _mock_smtp_config(section: str, key: str):
|
||||
if section == "notifications":
|
||||
if key == "sendmail":
|
||||
return cls()
|
||||
elif key == "smtp-use-ssl":
|
||||
return cls(0)
|
||||
elif key == "smtp-use-starttls":
|
||||
return cls(0)
|
||||
elif key == "smtp-user":
|
||||
return cls()
|
||||
elif key == "smtp-password":
|
||||
return cls()
|
||||
return cls(config_get(section, key))
|
||||
return _mock_smtp_config
|
||||
|
||||
|
||||
def test_smtp(user: User):
|
||||
with db.begin():
|
||||
user.ResetKey = "12345678901234567890123456789012"
|
||||
|
||||
SMTP = FakeSMTP()
|
||||
|
||||
get = "aurweb.config.get"
|
||||
getboolean = "aurweb.config.getboolean"
|
||||
with mock.patch(get, side_effect=mock_smtp_config(str)):
|
||||
with mock.patch(getboolean, side_effect=mock_smtp_config(bool)):
|
||||
with mock.patch("smtplib.SMTP", side_effect=lambda a, b: SMTP):
|
||||
config.rehash()
|
||||
notif = notify.WelcomeNotification(user.ID)
|
||||
notif.send()
|
||||
config.rehash()
|
||||
assert len(SMTP.emails) == 1
|
||||
|
||||
|
||||
def mock_smtp_starttls_config(cls):
|
||||
def _mock_smtp_starttls_config(section: str, key: str):
|
||||
if section == "notifications":
|
||||
if key == "sendmail":
|
||||
return cls()
|
||||
elif key == "smtp-use-ssl":
|
||||
return cls(0)
|
||||
elif key == "smtp-use-starttls":
|
||||
return cls(1)
|
||||
elif key == "smtp-user":
|
||||
return cls("test")
|
||||
elif key == "smtp-password":
|
||||
return cls("password")
|
||||
return cls(config_get(section, key))
|
||||
return _mock_smtp_starttls_config
|
||||
|
||||
|
||||
def test_smtp_starttls(user: User):
|
||||
# This test does two things: test starttls path and test
|
||||
# path where we have a backup email.
|
||||
|
||||
with db.begin():
|
||||
user.ResetKey = "12345678901234567890123456789012"
|
||||
user.BackupEmail = "backup@example.org"
|
||||
|
||||
SMTP = FakeSMTP()
|
||||
|
||||
get = "aurweb.config.get"
|
||||
getboolean = "aurweb.config.getboolean"
|
||||
with mock.patch(get, side_effect=mock_smtp_starttls_config(str)):
|
||||
with mock.patch(
|
||||
getboolean, side_effect=mock_smtp_starttls_config(bool)):
|
||||
with mock.patch("smtplib.SMTP", side_effect=lambda a, b: SMTP):
|
||||
notif = notify.WelcomeNotification(user.ID)
|
||||
notif.send()
|
||||
assert SMTP.starttls_enabled
|
||||
assert SMTP.user
|
||||
assert SMTP.passwd
|
||||
|
||||
assert len(SMTP.emails) == 2
|
||||
to = SMTP.emails[0][1]
|
||||
assert to == [user.Email]
|
||||
|
||||
to = SMTP.emails[1][1]
|
||||
assert to == [user.BackupEmail]
|
||||
|
||||
|
||||
def mock_smtp_ssl_config(cls):
|
||||
def _mock_smtp_ssl_config(section: str, key: str):
|
||||
if section == "notifications":
|
||||
if key == "sendmail":
|
||||
return cls()
|
||||
elif key == "smtp-use-ssl":
|
||||
return cls(1)
|
||||
elif key == "smtp-use-starttls":
|
||||
return cls(0)
|
||||
elif key == "smtp-user":
|
||||
return cls("test")
|
||||
elif key == "smtp-password":
|
||||
return cls("password")
|
||||
return cls(config_get(section, key))
|
||||
return _mock_smtp_ssl_config
|
||||
|
||||
|
||||
def test_smtp_ssl(user: User):
|
||||
with db.begin():
|
||||
user.ResetKey = "12345678901234567890123456789012"
|
||||
|
||||
SMTP = FakeSMTP_SSL()
|
||||
|
||||
get = "aurweb.config.get"
|
||||
getboolean = "aurweb.config.getboolean"
|
||||
with mock.patch(get, side_effect=mock_smtp_ssl_config(str)):
|
||||
with mock.patch(getboolean, side_effect=mock_smtp_ssl_config(bool)):
|
||||
with mock.patch("smtplib.SMTP_SSL", side_effect=lambda a, b: SMTP):
|
||||
notif = notify.WelcomeNotification(user.ID)
|
||||
notif.send()
|
||||
assert len(SMTP.emails) == 1
|
||||
assert SMTP.use_ssl
|
||||
assert SMTP.user
|
||||
assert SMTP.passwd
|
||||
|
||||
|
||||
def test_notification_defaults():
|
||||
notif = notify.Notification()
|
||||
assert notif.get_refs() == tuple()
|
||||
assert notif.get_headers() == dict()
|
||||
assert notif.get_cc() == list()
|
||||
Loading…
Add table
Add a link
Reference in a new issue