diff options
author | Jesse Morgan <jesse@jesterpm.net> | 2023-12-21 10:40:14 -0800 |
---|---|---|
committer | Jesse Morgan <jesse@jesterpm.net> | 2023-12-21 10:40:14 -0800 |
commit | 61138e02984aa01d13dc41877b6e31054b3dd3cc (patch) | |
tree | 6a00b9232e2a0b6f64fdbdac1b0e16c94d7fef4a /tests/test_canary.py | |
parent | d9e353315aa6c3f7c75f9aa883c656b77cdd5e54 (diff) |
Fix tests broken in previous commit
Diffstat (limited to 'tests/test_canary.py')
-rw-r--r-- | tests/test_canary.py | 9 |
1 files changed, 6 insertions, 3 deletions
diff --git a/tests/test_canary.py b/tests/test_canary.py index c0ba262..12c3279 100644 --- a/tests/test_canary.py +++ b/tests/test_canary.py @@ -15,6 +15,7 @@ USER_ADDRESS2 = "user2@example.com" SERVER = "mail.example.com" PASSWORD = "secret" +MUTE = 0 class TestCanary(unittest.TestCase): def setUp(self): @@ -58,9 +59,10 @@ class TestCanary(unittest.TestCase): def testCheck(self): # Setup mocks expectedUUID = "1234-5678-9012-3456" + self.db.get_missing_pongs.return_value = [] self.db.get_accounts.return_value = [ \ - (LIST_ADDRESS, USER_ADDRESS1, SERVER, PASSWORD), \ - (LIST_ADDRESS, USER_ADDRESS2, SERVER, PASSWORD)] + (LIST_ADDRESS, USER_ADDRESS1, SERVER, PASSWORD, MUTE), \ + (LIST_ADDRESS, USER_ADDRESS2, SERVER, PASSWORD, MUTE)] canary.emailutils.get_mail_uids.return_value = [1] canary.emailutils.get_message.return_value = {'Subject': "Canary Email " + expectedUUID} @@ -83,7 +85,8 @@ class TestCanary(unittest.TestCase): def testDontDeleteOtherMail(self): # Setup mocks - self.db.get_accounts.return_value = [(LIST_ADDRESS, USER_ADDRESS1, SERVER, PASSWORD)] + self.db.get_accounts.return_value = [(LIST_ADDRESS, USER_ADDRESS1, SERVER, PASSWORD, MUTE)] + self.db.get_missing_pongs.return_value = [] canary.emailutils.get_mail_uids.return_value = [1] canary.emailutils.get_message.return_value = {'Subject': "Buy Our New Widget"} |