terrain.py for running lettuce with django

terrain.py

import logging

from django.core.management import call_command

from lettuce import before, after

import yourprojec.test_settings

@before.all
def before_all():
    logging.info("Setting up a test database...\n")
    
    # Sync the test database
    call_command('syncdb',
      **{
        'settings': yourproject.test_settings,
        'interactive': False,
        'verbosity': 0,
      }
    )
    
    # Migrate test database with south
    call_command('migrate',
      **{
        'settings': yourproject.test_settings,
        'interactive': False,
        'verbosity': 0,
      }
    )

@before.each_scenario
def before_scenario(scenario):
    logging.info('Flushing test database...\n')
    
    # Flush the database
    call_command('flush',
      **{
        'settings': yourproject.test_settings,
        'interactive': False,
        'verbosity': 0,
      }
    )