예제 #1
0
def assert_query_count(num, ignore_txn=False):
    qh = QueryLogHandler()
    logger.setLevel(logging.DEBUG)
    logger.addHandler(qh)
    try:
        qc0 = len(qh.queries(ignore_txn=ignore_txn))
        yield qh
    finally:
        logger.removeHandler(qh)
        qc1 = len(qh.queries(ignore_txn=ignore_txn))
        assert (qc1 - qc0) == num
예제 #2
0
 def tearDown(self):
     logger.removeHandler(self.qh)
예제 #3
0
파일: base.py 프로젝트: knightliao/peewee 
 def tearDown(self):
     logger.removeHandler(self.qh)