Lubuntu/britney2-ubuntu
8
0
Fork 0
mirror of https://git.launchpad.net/~ubuntu-release/britney/+git/britney2-ubuntu synced 2025-05-29 19:31:30 +00:00
Code Issues Packages Projects Releases Wiki Activity

Make it possible to test the email frequency for a package that is sometimes a valid candidate and sometimes not

Browse Source
This commit is contained in:
Steve Langasek 2017-06-29 12:46:37 -07:00
parent 27ef44e478
commit 6459977822
1 changed files with 8 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
tests/test_email.py
View File

@ -225,8 +225,10 @@ class T(unittest.TestCase):
@patch('britney2.policies.email.EmailPolicy.lp_get_emails') @patch('britney2.policies.email.EmailPolicy.lp_get_emails')
@patch('britney2.policies.email.smtplib', autospec=True) @patch('britney2.policies.email.smtplib', autospec=True)
def smtp_repetition(self, smtp, lp, valid=False, expected=None): def smtp_repetition(self, smtp, lp, valid, expected):
"""Resend mails periodically, with decreasing frequency.""" """Resend mails periodically, with decreasing frequency."""
if not isinstance(valid,list):
valid = [valid]*len(expected)
FakeExcuse.is_valid = valid FakeExcuse.is_valid = valid
lp.return_value = ['email@address.com'] lp.return_value = ['email@address.com']
sendmail = smtp.SMTP().sendmail sendmail = smtp.SMTP().sendmail
@ -237,6 +239,11 @@ class T(unittest.TestCase):
previous = sendmail.call_count previous = sendmail.call_count
age = hours / 24 age = hours / 24
FakeExcuse.daysold = age FakeExcuse.daysold = age
try:
FakeExcuse.is_valid = valid[len(called)]
except IndexError:
# we've already gotten all the mails we expect
pass
e.apply_policy_impl(None, None, 'unity8', None, FakeSourceData, FakeExcuse) e.apply_policy_impl(None, None, 'unity8', None, FakeSourceData, FakeExcuse)
if sendmail.call_count > previous: if sendmail.call_count > previous:
e.initialise(None) # Refill e.cache from disk e.initialise(None) # Refill e.cache from disk