sanity.py

#!/usr/bin/env python
# encoding: utf-8
from optparse import OptionParser
from clint.textui import colored
import sys
import os
import re
import time

testfile_pattern = 'test_(.+)\.[json|yaml]'


def run_tests(folder):
    """
    searches for testfiles in a folder recursively
    """
    test_files = []
    start = time.clock()

    for root, subfolder, files in os.walk(folder):
        for file_ in files:
            match_ = re.match(testfile_pattern, file_)
            if match_:
                testfile_ = os.path.abspath(file_)
                test_files.append(testfile_)

    elapsed = (time.clock() - start)
    print 'RAN %s tests in %s' % (len(test_files), elapsed)
    print colored.green('OK')
    #print colored.red('OK')


def main():

    usage = 'usage: %prog [options] user -h for help'
    parser = OptionParser(usage)
    parser.add_option('-f', '--testfolder', action='store',
    type='string', dest='testfolder', help='folder containg the tests')
    #parser.add_option('-t', '--type', action='store',
    #type='string', dest='testfolder', help='folder containg the tests')

    (options, args) = parser.parse_args()
    if options.testfolder:
        run_tests(options.testfolder)
    else:
        parser.print_help()

if __name__ == '__main__':
    status = main()
    sys.exit(status)