terrain.py

import os
import time
import shutil
from django.core.management import call_command
from django.conf import settings
from splinter.browser import Browser
from lettuce import after, before, world
from web_steps import *


PROJECT_ROOT_PATH = os.path.dirname(os.path.abspath(__file__))

@before.harvest
def initial_run_time(variables):
    world.time_before_harvest = time.time()
    test_directories_list = [
        os.path.join(PROJECT_ROOT_PATH, 'site_media', 'test', 'files', 'projects'),
        os.path.join(PROJECT_ROOT_PATH, 'site_media', 'test', 'images', 'members'),
        os.path.join(PROJECT_ROOT_PATH, 'site_media', 'test', 'images', 'news'),
        os.path.join(PROJECT_ROOT_PATH, 'site_media', 'test', 'images', 'projects'),
        os.path.join(PROJECT_ROOT_PATH, 'site_media', 'test', 'images', 'tools'),
    ]

    for directory in test_directories_list:
        if not os.path.exists(directory):
            os.makedirs(directory)

@before.all
def set_browser():
    world.browser = Browser()

@before.each_scenario
def clean_database(scenario):
    clean_data()
    clean_media()

def clean_data():
    call_command('flush', interactive=False)
    call_command('loaddata', 'all')

def clean_media():
    clean_media_by_kind('images')
    clean_media_by_kind('files')

def clean_media_by_kind(kind, skip_dir=None):
    images_dir = os.path.join(settings.MEDIA_ROOT, 'test', kind)
    for file_name in os.listdir(images_dir):
        clean_all(os.path.join(images_dir, file_name))

def clean_all(directory, skip_dir=None):
    for file_name in os.listdir(directory):
        absname = os.path.join(directory, file_name)
        if skip_dir and skip_dir in absname:
            pass
        elif os.path.isdir(absname) and file_name not in ['.', '..']:
            clean_all(absname)
        elif not file_name.startswith('.'):
            os.unlink(absname)

@after.all
def finish_him(total_result):
    world.browser.quit()
    clean_media()

@after.harvest
def run_time(results):
    shutil.rmtree(os.path.join(PROJECT_ROOT_PATH, 'site_media', 'test'))
    time_after_harvest = time.time()
    time_before_harvest = world.time_before_harvest
    total_time = int(time_after_harvest - time_before_harvest)
    minutes = total_time / 60
    seconds = total_time % 60
    total_features_ran = 0
    total_scenarios_ran = 0
    total_scenarios_passed = 0
    total_scenarios_failed = dict()
    for app in results:
        total_features_ran += app.features_ran
        total_scenarios_ran += app.scenarios_ran
        total_scenarios_passed += app.scenarios_passed
        for result in app.scenario_results:
            if result.passed == False:
                total_scenarios_failed.update({result.scenario.name: result.scenario.feature})

    print '\n================================================'
    print '=========== Acceptance tests results ===========\n'
    print '\033[32m\033[1m%i features ran.\033[0;0m' % total_features_ran
    print '\033[32m\033[1m%i scenarios ran.\033[0;0m' % total_scenarios_ran
    print '\033[32m\033[1m%i scenarios passed.\033[0;0m' % total_scenarios_passed
    if total_scenarios_failed:
        print '\033[31m\033[1mError in:\033[0;0m'
        for scenario, feature in total_scenarios_failed.iteritems():
            print '\033[31m\033[1m' + feature.name + ' ~> ' + scenario + '\033[0;0m'
    else:
        print '\033[32m\033[1mNo errors! Congratulations!\033[0;0m'
    print '----------------------------------------------------------------------'
    print 'Everything ran in %i minute(s) and %i second(s).' % (minutes, seconds)