diff --git a/matemat/db/test/test_facade.py b/matemat/db/test/test_facade.py index 70cb4dc..5b4089e 100644 --- a/matemat/db/test/test_facade.py +++ b/matemat/db/test/test_facade.py @@ -479,15 +479,18 @@ class DatabaseTest(unittest.TestCase): 'user6': user6.id }) - self.assertFalse(db.check_receipt_due(user0)) - self.assertFalse(self.db.check_receipt_due(user1)) - self.assertTrue(self.db.check_receipt_due(user2)) - self.assertTrue(self.db.check_receipt_due(user3)) - self.assertFalse(self.db.check_receipt_due(user4)) - self.assertTrue(self.db.check_receipt_due(user5)) - self.assertFalse(self.db.check_receipt_due(user6)) + import logging + logger = logging.getLogger('test') + + self.assertFalse(db.check_receipt_due(user0, logger)) + self.assertFalse(self.db.check_receipt_due(user1, logger)) + self.assertTrue(self.db.check_receipt_due(user2, logger)) + self.assertTrue(self.db.check_receipt_due(user3, logger)) + self.assertFalse(self.db.check_receipt_due(user4, logger)) + self.assertTrue(self.db.check_receipt_due(user5, logger)) + self.assertFalse(self.db.check_receipt_due(user6, logger)) with self.assertRaises(TypeError): - self.db.check_receipt_due(user7) + self.db.check_receipt_due(user7, logger) def test_create_receipt(self): with self.db as db: