blob: 658541c0941d63e8d960c6d105ddbee3ae70d092 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
|
import tempfile, shutil, os
from django.core import management
from django.conf import settings
import common
# we need to set DATABASE_ENGINE now, at import time, before the Django database
# system gets initialized.
# django.conf.settings.LazySettings is buggy and requires us to get something
# from it before we set stuff on it.
getattr(settings, 'DATABASE_ENGINE')
settings.DATABASE_ENGINE = 'autotest_lib.frontend.db.backends.afe_sqlite'
settings.DATABASE_NAME = ':memory:'
from django.db import connection
from autotest_lib.frontend.afe import readonly_connection
def run_syncdb(verbosity=0):
management.call_command('syncdb', verbosity=verbosity, interactive=False)
def destroy_test_database():
connection.close()
# Django brilliantly ignores close() requests on in-memory DBs to keep us
# naive users from accidentally destroying data. So reach in and close
# the real connection ourselves.
# Note this depends on Django internals and will likely need to be changed
# when we move to Django 1.x.
real_connection = connection.connection
if real_connection is not None:
real_connection.close()
connection.connection = None
def set_up():
run_syncdb()
readonly_connection.ReadOnlyConnection.set_globally_disabled(True)
def tear_down():
readonly_connection.ReadOnlyConnection.set_globally_disabled(False)
destroy_test_database()
def print_queries():
"""
Print all SQL queries executed so far. Useful for debugging failing tests -
you can call it from tearDown(), and then execute the single test case of
interest from the command line.
"""
for query in connection.queries:
print query['sql'] + ';\n'
|
