httprunner/ate/main.py

import argparse
import unittest

from ate import runner, utils


class ApiTestCase(unittest.TestCase):
    """ create a testcase.
    """
    def __init__(self, test_runner, testcase):
        super(ApiTestCase, self).__init__()
        self.test_runner = test_runner
        self.testcase = testcase

    def runTest(self):
        """ run testcase and check result.
        """
        result = self.test_runner.run_test(self.testcase)
        self.assertEqual(result, (True, {}))

def create_suite(testset):
    """ create test suite with a testset, it may include one or several testcases.
        each suite should initialize a seperate TestRunner() with testset config.
    """
    suite = unittest.TestSuite()

    test_runner = runner.TestRunner()
    config_dict = testset.get("config", {})
    test_runner.pre_config(config_dict)
    testcases = testset.get("testcases", [])

    for testcase in testcases:
        test = ApiTestCase(test_runner, testcase)
        suite.addTest(test)

    return suite

def create_task(testcase_path):
    """ create test task suite with specified testcase path.
        each task suite may include one or several test suite.
    """
    task_suite = unittest.TestSuite()
    testsets = utils.load_testcases_by_path(testcase_path)

    for testset in testsets:
        suite = create_suite(testset)
        task_suite.addTest(suite)

    return task_suite

def main():
    """ parse command line options and run commands.
    """
    parser = argparse.ArgumentParser(
        description='Api Test Engine.')
    parser.add_argument(
        '--testcase-path', default='testcases',
        help="testcase file path")

    args = parser.parse_args()
    task_suite = create_task(args.testcase_path)
    unittest.TextTestRunner().run(task_suite)