/
djangoify.py
59 lines (47 loc) · 1.93 KB
/
djangoify.py
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
52
53
54
55
56
57
58
59
from lettuce import before, world, after
from salad.logger import logger
old_database_name = None
try:
from django.core.management import call_command
from django.conf import settings
from django.test.simple import DjangoTestSuiteRunner
from django.core import mail
try:
from south.management.commands import patch_for_test_db_setup
USE_SOUTH = getattr(settings, "SOUTH_TESTS_MIGRATE", False)
except:
USE_SOUTH = False
@before.runserver
def setup_database(actual_server):
logger.info("Setting up a test database ...\n")
if settings.DATABASES["default"]["ENGINE"] != "django.db.backends.sqlite3":
from django.db import connection
global old_database_name
old_database_name = settings.DATABASES["default"]["NAME"]
connection.creation.create_test_db()
else:
# connection.settings_dict["NAME"] = ":memory:"
pass
if USE_SOUTH:
patch_for_test_db_setup()
world.test_runner = DjangoTestSuiteRunner(interactive=False)
world.test_runner.setup_test_environment()
call_command('syncdb', interactive=False, verbosity=0)
call_command('flush', interactive=False, verbosity=0)
@after.runserver
def teardown_database(actual_server):
logger.info("Destroying test database ...\n")
world.test_runner.teardown_test_environment()
if settings.DATABASES["default"]["ENGINE"] != "django.db.backends.sqlite3":
from django.db import connection
connection.creation.destroy_test_db(old_database_name)
@after.each_feature
def reset_data(scenario):
logger.info("Flushing...")
call_command('flush', interactive=False, verbosity=0)
@before.each_feature
def empty_outbox(scenario):
logger.info("Emptying outbox...")
mail.outbox = []
except:
logger.warn("Django terrains not imported.")