24 lines
741 B
Python
24 lines
741 B
Python
|
''' Utils: tester for database operations. '''
|
||
|
|
import logging
|
||
|
|
|
||
|
|
from django.db import connection
|
||
|
|
from rest_framework.test import APITestCase
|
||
|
|
|
||
|
|
|
||
|
|
class DBTester(APITestCase):
|
||
|
|
''' Abstract base class for Testing database. '''
|
||
|
|
|
||
|
|
def setUp(self):
|
||
|
|
self.logger = logging.getLogger('django.db.backends')
|
||
|
|
self.logger.setLevel(logging.DEBUG)
|
||
|
|
|
||
|
|
def start_db_log(self):
|
||
|
|
''' Warning! Do not use this second time before calling stop_db_log. '''
|
||
|
|
''' Warning! Do not forget to enable global logging in settings. '''
|
||
|
|
logging.disable(logging.NOTSET)
|
||
|
|
connection.force_debug_cursor = True
|
||
|
|
|
||
|
|
def stop_db_log(self):
|
||
|
|
connection.force_debug_cursor = False
|
||
|
|
logging.disable(logging.CRITICAL)
|