Example #1
0
def load_tests(loader, tests, patterns):
    """
    Custom test loader
    """
    suite = VagrantTestSuite(BASE_BOXES)

    # Add a test case for each fabric task
    path = os.path.join(os.path.dirname(__file__), 'fabfile.py')
    _, tasks, _ = load_fabfile(path)
    for name, task in tasks.items():
        suite.addTest(VagrantFunctionTestCase(task))

    return suite