def test_logger(self): logger_one = Logger('one') logger_two = Logger('two') logger_one.start_logging() query = Query().from_table(Account) query.select() self.assertEqual(logger_one.count(), 1, 'Incorrect number of queries') query.select() logger_two.start_logging() query.select() logger_one.stop_logging() query.select() self.assertEqual(logger_one.count(), 3, 'Incorrect number of queries') self.assertEqual(logger_two.count(), 2, 'Incorrect number of queries') query.select() logger_one.start_logging() query.select() self.assertEqual(logger_one.count(), 4, 'Incorrect number of queries') self.assertEqual(logger_two.count(), 4, 'Incorrect number of queries') query.select() logger_two.clear_log() query.select() self.assertEqual(logger_one.count(), 6, 'Incorrect number of queries') self.assertEqual(logger_two.count(), 1, 'Incorrect number of queries')
def test_stop_logging(self): """ Verifies that the logger stops caring about queries """ logger = Logger() logger.start_logging() query = Query().from_table(Account) query.select() query.select() self.assertEqual(2, logger.count()) logger.stop_logging() query.select() query.select() self.assertEqual(2, logger.count()) logger.start_logging() query.select() self.assertEqual(3, logger.count())