Python生成JMeter测试脚本----响应断言和查看结果树

前言

之前的文章写了怎么用python生成JMeter脚本，生成了测试计划、线程组、Http监听器组件，详细内容查看文章《Python生成JMeter测试脚本》。本篇内容接着讲怎么用python生成响应断言和查看结果树组件。

响应断言

响应断言的测试字段和模型匹配规则，在代码中已经注释了，对应着JMeter GUI工具的操作。首先使用set方法修改响应断言的参数，然后使用get方法生成响应断言组件，将组件添加到线程组中，如果想添加到Http监听器中，只需要get方法的parent传http监听器就可以。

import xml.etree.ElementTree as ET
from base import General
from HashCode import GetHashCode


class JMeterResponseAssert:
    def __init__(self):
        # 断言名称
        self.__assert_name = '响应断言'
        # 注释
        self.__comments = ''
        # 自定义失败消息
        self.__custom_message = ''
        # 测试字段    Assertion.response_data 响应文本    Assertion.response_code 响应代码    Assertion.response_message 响应信息
        # Assertion.request_headers 请求头    Assertion.sample_label url样本    Assertion.response_data_as_document 文档(文本)
        # Assertion.request_data 请求数据
        self.__test_field = 'Assertion.response_data'
        self.__assume_success = 'false'
        # 模式匹配规则   1 匹配    2 包含    5 不匹配    6 不包含    8 相等    12 不相等    16 子字符串    20 不是子字符串    33 或者匹配    34 或者包含
        # 37 或者不匹配    38 或者不包含    40 或者相等    44 或者不相等    48 或者是子字符串    52 或者不是子字符串
        self.__test_type = ''
        # 适用方式  all 主样本和分支样本    children 分支样本    variable 使用jmeter变量名称    默认为空仅主样本
        self.__scope = ''
        # 变量名称
        self.__scope_variable = ''
        self.__assert_text = []

    def get_assert_name(self):
        return self.__assert_name

    def set_assert_name(self, name):
        self.__assert_name = name

    def get_comments(self):
        return self.__comments

    def set_comments(self, comments):
        self.__comments = comments

    def get_custom_message(self):
        return self.__custom_message

    def set_custom_message(self, custom_message):
        self.__custom_message = custom_message

    def get_test_field(self):
        return self.__test_field

    def set_test_field(self, test_field):
        self.__test_field = test_field

    def get_assume_type(self):
        return self.__assume_success

    def set_assume_type(self, assume_type):
        self.__assume_success = assume_type

    def get_test_type(self):
        return self.__test_type

    def set_test_type(self, test_type):
        self.__test_type = test_type

    def get_scope(self):
        return self.__scope

    def set_scope(self, scope):
        self.__scope = scope

    def get_scope_variable(self):
        return self.__scope_variable

    def get_scope_variable(self, scope_variable):
        self.__scope_variable = scope_variable

    def get_assert_text(self):
        return self.__assert_text

    def set_assert_text(self, assert_text):
        self.__assert_text = assert_text

    def items(self):
        dicts = {
            'comments': self.__comments,
            'custom_message': self.__custom_message,
            'test_field': self.__test_field,
            'assume_success': self.__assume_success,
            'test_type': self.__test_type,
            'scope': self.__scope,
            'scope_variable': self.__scope_variable,
        }
        return dicts

    def get(self, parent):
        prop_map = {
            'comments': General.add_str_prop,
            'custom_message': General.add_str_prop,
            'test_field': General.add_str_prop,
            'assume_success': General.add_bool_prop,
            'test_type': General.add_int_prop,
            'scope': General.add_str_prop,
            'scope_variable': General.add_str_prop
        }

        keys_map = {
            'comments': 'TestPlan.comments',
            'custom_message': 'Assertion.custom_message',
            'test_field': 'Assertion.test_field',
            'assume_success': 'Assertion.assume_success',
            'test_type': 'Assertion.test_type',
            'scope': 'Assertion.scope',
            'scope_variable': 'Scope.variable'
        }

        assert_prop = ET.SubElement(parent, 'ResponseAssertion')
        assert_prop.set('guiclass', 'AssertionGui')
        assert_prop.set('testclass', 'ResponseAssertion')
        assert_prop.set('testname', self.__assert_name)

        if self.__assert_text:
            collection_prop = ET.SubElement(assert_prop, 'collectionProp')
            collection_prop.set('name', 'Asserion.test_strings')
            for text in self.__assert_text:
                General.add_str_prop(self, collection_prop, str(GetHashCode.getHashCode(text)), text)

        for key, value in self.items().items():
            if key in prop_map:
                if value == '' or value == 'false':
                    pass
                else:
                    prop_map[key](self, assert_prop, keys_map[key], str(value))
        hashtree = ET.SubElement(parent, 'hashTree')
        return assert_prop

查看结果树

查看结果树组件中的__sample_save_config参数，对应GUI工具中的配置，可以配置查看结果树。

import xml.etree.ElementTree as ET
from base import General


class JMeterResultCollector:
    def __init__(self):
        self.__result_collector_name = '查看结果树'
        # 显示日志内容  为true时勾选仅错误日志
        self.__error_logging = 'false'
        # 显示日志内容 为true时勾选仅成功日志，与错误日志只能选择一个
        self.__success_only_logging = ''
        # 数据写入文件的文件名称
        self.__filename = ''
        # 示例保存配置
        self.__sample_save_config = {
            'time': 'true',
            'latency': 'true',
            'timestamp': 'true',
            'success': 'true',
            'label': 'true',
            'code': 'true',
            'message': 'true',
            'threadName': 'true',
            'dataType': 'true',
            'encoding': 'false',
            'assertions': 'true',
            'subresults': 'true',
            'responseData': 'false',
            'samplerData': 'false',
            'xml': 'false',
            'fieldNames': 'true',
            'responseHeaders': 'false',
            'requestHeaders': 'false',
            'responseDataOnError': 'false',
            'saveAssertionResultsFailureMessage': 'true',
            'assertionsResultsToSave': '0',
            'bytes': 'true',
            'sentBytes': 'true',
            'url': 'true',
            'threadCounts': 'true',
            'idleTime': 'true',
            'connectTime': 'true'
        }

    def get_result_collector_name(self):
        return self.__result_collector_name

    def set_result_collector_name(self, result_collector_name):
        self.__result_collector_name = result_collector_name

    def get_error_logging(self):
        return self.__error_logging

    def set_error_logging(self, error_logging):
        self.__error_logging = error_logging

    def get_success_only_logging(self):
        return self.__success_only_logging

    def set_success_only_logging(self, success_only_logging):
        self.__success_only_logging = success_only_logging

    def get_filename(self):
        return self.__filename

    def set_filename(self, filename):
        self.__filename = filename

    def get_sample_save_config(self):
        return self.__sample_save_config

    def set_sample_save_config(self, sample_save_config: dict):
        self.__sample_save_config = sample_save_config

    def set_config(self, config: dict):
        for key, value in config.items():
            self.__sample_save_config[key] = value

    def items(self):
        dicts = {
            'error_logging': self.__error_logging,
            'success_only_logging': self.__success_only_logging,
            'filename': self.__filename
        }
        return dicts

    def get(self, parent):
        prop_map = {
            'error_logging': General.add_bool_prop,
            'success_only_logging': General.add_bool_prop,
            'filename': General.add_str_prop
        }

        keys_map = {
            'error_logging': 'ResultCollector.error_logging',
            'success_only_logging': 'ResultCollector.success_only_logging',
            'filename': 'filename'
        }

        result_collector_prop = ET.SubElement(parent, 'ResultCollector')
        result_collector_prop.set('guiclass', 'ViewResultsFullVisualizer')
        result_collector_prop.set('testclass', 'ResultCollector')
        result_collector_prop.set('testname', self.__result_collector_name)

        for key, value in self.items().items():
            if key in prop_map:
                prop_map[key](self, result_collector_prop, keys_map[key], str(value))
                # if value == '' or value == 'false':
                #     pass
                # else:
                #     prop_map[key](self, result_collector_prop, keys_map[key], str(value))
        obj_prop = ET.SubElement(result_collector_prop, 'objProp')
        config_name_prop = ET.SubElement(obj_prop, 'name')
        config_name_prop.text = 'saveConfig'
        value_prop = ET.SubElement(obj_prop, 'value')
        value_prop.set('class', 'SampleSaveConfiguration')
        for key, value in self.__sample_save_config.items():
            props = ET.SubElement(value_prop, key)
            props.text = str(value)
        hashtree = ET.SubElement(parent, 'hashTree')
        return result_collector_prop

生成JMeter脚本demo

http监听器组件、响应 断言组件、查看结果树组件都在同一个hashtree下，所以先创建一个test_hashtree，然后生成方法都穿这个标签。

import xml.etree.ElementTree as ET
import JMeter
from JMeterThreadGroup import JMeterThreadGroup
from JMeterTestPlan import JMeterTestPlan
from JMeterHttpSampler import JMeterHttpSampler
from JMeterResponseAssert import JMeterResponseAssert
from JMeterResultCollector import JMeterResultCollector
import base


class GenJMeter:
    jmeter_script = JMeter.get()
    hash_tree = ET.SubElement(jmeter_script, 'hashTree')
    test_plan = JMeterTestPlan()
    test_plan = test_plan.get(hash_tree)
    thread_group_tree = ET.SubElement(hash_tree, 'hashTree')
    thread_group = JMeterThreadGroup().get(thread_group_tree)
    test_hashtree = ET.SubElement(thread_group_tree, 'hashTree')
    http_sampler = JMeterHttpSampler()
    http_sampler.set_params({'test': 123})
    http_sampler.set_files({'file': 'test.txt'})
    http_sampler.get(test_hashtree)
    response_assert = JMeterResponseAssert()
    response_assert.set_assert_name('断言')
    response_assert.set_assert_text(['test', '123'])
    response_assert.set_test_type('2')
    response_assert.get(test_hashtree)
    result_collector_prop = JMeterResultCollector()
    result_collector_prop.get(test_hashtree)
    bases = base.General()
    tree = ET.ElementTree(jmeter_script)
    bases.indent(jmeter_script)
    with open('test.jmx', "wb") as f:
        tree.write(f, encoding="utf-8", xml_declaration=True, method="xml")

其他组件实现方式待续