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_clear_log(self): """ Makes sure queries are cleared """ logger_one = Logger('one') logger_one.start_logging() query = Query().from_table(Account) # run a query and update the logger's query list query.select() logger_one.update_log() # the index should be at 1 self.assertEqual(1, logger_one.query_index) # increment the connection query count query.select() # clear the log logger_one.clear_log() # make sure no queries self.assertEqual(0, len(logger_one.queries)) # query index should match that of the connection log self.assertEqual(2, logger_one.query_index)
def test_clear_log_no_index(self): """ Makes sure that the query index doesn't change """ logger_one = Logger('one') query = Query().from_table(Account) query.select() self.assertIsNone(logger_one.query_index) # clear the log logger_one.clear_log() # make sure no query index self.assertIsNone(logger_one.query_index)
def test_clear_log(self): """ Makes sure queries are cleared """ logger_one = Logger('one') logger_one.start_logging() query = Query().from_table(Account) # run a query and update the logger's query list query.select() logger_one.update_log() # there should be one query self.assertEqual(logger_one.count(), 1) # increment the connection query count query.select() # clear the log logger_one.clear_log() # make sure no queries self.assertEqual(0, len(logger_one.queries))