? ? 最近公司在用jmeter做接口自动化,我平时jmeter用得比较少,一般拿来搞搞以验收为目的的性能测试,因此也是遇到了不少的问题。 ? ? 本文记录的问题是:对整个测试脚本设置全局的Header Manager后,如何对单个接口单独设置请求头。 ? ? 我们在做接口自动化时,会创建一个全局的Header Manager去存放一些公用的属性,例如身份认证、Content type等。application/json是我们经常会使用的Content type,但如果遇到如图片、文件上传之类的接口,这个Content type就不适用了。 ? ? 一开始,我尝试对图片上传接口单独设置一个Header Manager,并坚信jmeter工具内部一定有类似变量作用域的机制,结果jmeter并没有。 ? ? 后来,我想了个比较笨的方法,那就是为图片上传接口添加一个前置的beanshell脚本,在上传接口执行之前,将Header Manager的Content type给去掉。于是乎我就去jmeter官网查看jmeter api。我们的前置处理器是要拿到针对该请求的Header Manager,因为Header Manager是属于Config Element,因此我先试着去org.apache.jmeter.config包下找相关api,结果没有。 ? ? 既然config类没有提供相关api,那么就只能从sampler类入手,该上传接口使用的是http协议,因此我又去org.apache.jmeter.protocol.http.sampler中查看。 ? ?HTTPSamplerBase中定义了一些Common constants、methods for HTTP samplers,而HTTPSampler是HTTPSampleBase的子类。所以我就先去查看HTTPSamplerBase类定义了哪些方法。 ? ?果然,我们可以通过sampler.getHeaderManager()来获取到信息头管理器。但光获取到Header Manager对象还不够,还需要将其中的Content type移除。 ? ?点击图中的Header Manager,跳转到对应的类中,该类下定义了一个removeHeaderNamed(String name)方法,我们可以通过传入Header的key,达到将对应value移除的效果。
sampler.getHeaderManager().removeHeaderNamed("Content-Type");
? ?但此时再跑一遍接口,发现上传接口之后都在报错。结果树中显示,后面接口都没有了Content type,因此报错。我的解决方案是,为上传接口的下个接口再添加个beanshell前置处理器,将Content type添加进Header Manager中。 ? ?仍然沿用之前的思路,在HeaderManager类中,找到了一个add(Header h)方法,我们可以通过调用add方法,并为add方法传入一个header对象,来将该header对象添加进HeaderManager中。
? ?第二个前置处理器内容如下:
import org.apache.jmeter.protocol.http.control.HeaderManager;
import org.apache.jmeter.protocol.http.sampler;
import org.apache.jmeter.protocol.http.control.Header;
Header header = new Header("Content-Type", "application/json");
HeaderManager hm = sampler.getHeaderManager();
hm.add(header)
? ?遗憾的是这种方法太不灵活,如果有多个类似上传图片这样的接口,就不好拓展。如果有更好的方法,希望可以评论区指点一二。
|