package com.test.jmeterdemo;

 import com.test.http.HTTPClient;

 import com.test.http.HTTPUtils;

 import org.apache.jmeter.config.Arguments;

 import org.apache.jmeter.protocol.java.sampler.AbstractJavaSamplerClient;

 import org.apache.jmeter.protocol.java.sampler.JavaSamplerContext;

 import org.apache.jmeter.samplers.SampleResult;

 public class TestHTTP extends AbstractJavaSamplerClient  {

     private String url;

     private String method;

     private String header;

     private String reqBody;

     private String responseCode;

     private String responseHeader;

     private String responseBody;

     /**

      * 这个方法用来控制显示在GUI页面的属性，由用户来进行设置。

      * 此方法不用调用，是一个与生命周期相关的方法，类加载则运行。

      */

     public Arguments getDefaultParameters() {

         Arguments arguments = new Arguments();

         arguments.addArgument("Url", "");

         arguments.addArgument("Method", "");

         arguments.addArgument("Header", "");

         arguments.addArgument("ReqBody", "");

         return arguments;

     }

     /**

      * 初始化方法，初始化性能测试时的每个线程

      * 实际运行时每个线程仅执行一次，在测试方法运行前执行，类似于LoadRunner中的init方法

      */

     public void setupTest(JavaSamplerContext jsc){

         url = jsc.getParameter("Url");

         method = jsc.getParameter("Method");

         header = jsc.getParameter("Header");

         reqBody = jsc.getParameter("ReqBody");

     }

     /**

      * 性能测试时的线程运行体

      * 测试执行的循环体，根据线程数和循环次数的不同可执行多次，类似于Loadrunner中的Action方法

      */

     public SampleResult runTest(JavaSamplerContext javaSamplerContext) {

         SampleResult results = new SampleResult();

         results.setSamplerData(url+"\n"+method+"\n"+header+"\n"+reqBody);

         results.setRequestHeaders(header);

         //标记事务开始

         results.sampleStart();

         HTTPUtils.Response response = HTTPClient.sendHttpRequest(url, method, header, reqBody);

         //标记事务结束

         results.sampleEnd();

         if (response != null) {

             responseCode = String.valueOf(response.getResponseCode());

             responseHeader = response.getResponseHeader().toString();

             responseBody = (String) response.getResponseBody();

             results.setSuccessful(true);

         }

         results.setResponseCode(responseCode);

         results.setResponseHeaders(responseHeader);

         results.setResponseData(responseBody, "UTF-8");

         return results;

     }

     /**

      * 测试结束方法，结束测试中的每个线程

      * 实际运行时，每个线程仅执行一次，在测试方法运行结束后执行，类似于Loadrunner中的End方法

      */

     public void teardownTest(JavaSamplerContext args) {

     }

 }