コード例 #1
0
ファイル: utils.py プロジェクト: vlad0337187/aiopeewee 
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 setUp(self):
     self.qh = QueryLogHandler()
     logger.setLevel(logging.DEBUG)
     logger.addHandler(self.qh)
コード例 #3
0
ファイル: base.py プロジェクト: knightliao/peewee 
 def setUp(self):
     self.qh = QueryLogHandler()
     logger.setLevel(logging.DEBUG)
     logger.addHandler(self.qh)