httprunner/ate/task.py

import logging
import unittest

from ate import exception, runner, testcase, 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.
        """
        self.assertTrue(self.test_runner._run_test(self.testcase))

class ApiTestSuite(unittest.TestSuite):
    """ create test suite with a testset, it may include one or several testcases.
        each suite should initialize a separate Runner() with testset config.
    """
    def __init__(self, testset):
        super(ApiTestSuite, self).__init__()
        self.test_runner = runner.Runner()
        self.config_dict = testset.get("config", {})
        self.test_runner.init_config(self.config_dict, level="testset")
        testcases = testset.get("testcases", [])
        self._add_tests_to_suite(testcases)

    def _add_tests_to_suite(self, testcases):
        for testcase in testcases:
            if utils.PYTHON_VERSION == 3:
                ApiTestCase.runTest.__doc__ = testcase['name']
            else:
                ApiTestCase.runTest.__func__.__doc__ = testcase['name']

            test = ApiTestCase(self.test_runner, testcase)
            self.addTest(test)

    def print_output(self):
        output_variables_list = self.config_dict.get("output", [])
        self.test_runner.generate_output(output_variables_list)

class TaskSuite(unittest.TestSuite):
    """ create test task suite with specified testcase path.
        each task suite may include one or several test suite.
    """
    def __init__(self, testcase_path):
        super(TaskSuite, self).__init__()
        self.suite_list = []
        testsets = testcase.load_testcases_by_path(testcase_path)
        if not testsets:
            raise exception.FileNotFoundError

        for testset in testsets:
            suite = ApiTestSuite(testset)
            self.addTest(suite)
            self.suite_list.append(suite)

    @property
    def tasks(self):
        return self.suite_list