From 61138e02984aa01d13dc41877b6e31054b3dd3cc Mon Sep 17 00:00:00 2001 From: Jesse Morgan Date: Thu, 21 Dec 2023 10:40:14 -0800 Subject: Fix tests broken in previous commit --- tests/test_canary.py | 9 ++++++--- 1 file 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"} -- cgit v1.2.3