Jmeter 支持CASLogin 透传重定向地址

最新推荐文章于 2022-07-26 14:29:20 发布

deng_xintao

最新推荐文章于 2022-07-26 14:29:20 发布

阅读量290

点赞数

分类专栏： java

本文链接：https://blog.csdn.net/deng_xintao/article/details/87929227

版权

这是一个关于如何使用JMeter的Java实现，`CASLogin` 类，进行CAS（Central Authentication Service）登录并获取重定向地址的示例。代码中包括了设置测试参数、初始化、执行登录操作以及解析重定向URL的方法。该类继承自`AbstractJavaSamplerClient`，并实现了测试开始和结束时的回调方法。

摘要由CSDN通过智能技术生成

package com.*.casLogin;

import java.util.regex.Matcher;
import java.util.regex.Pattern;

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;

import com.*.jmeter.util.CASHttpClient;

public class CASLogin extends AbstractJavaSamplerClient {

   private String url;
   private String username;
   private String password;
   private static final String _eventId = "submit";
   private String lt;
   private SampleResult results;

   private static final String regex = "(\\d+\\.\\d+\\.\\d+\\.\\d+)\\:(\\d+)";

   // 这个方法是用来自定义java方法入参的。
   // params.addArgument("num1","");表示入参名字叫num1，默认值为空。
   public Arguments getDefaultParameters() {
       Arguments params = new Arguments();
       params.addArgument("url", "");
       params.addArgument("username", "");
       params.addArgument("password", "");
       return params;
   }

   //每个线程测试前执行一次，做一些初始化工作；
   public void setupTest(JavaSamplerContext context) {
       results = new SampleResult();
       results.setSampleLabel( "JAVA【CAS登录-返回重定向地址】");
       results.setDataEncoding("UTF-8");//因为响应的数据有中文，所以最好先设置编码
       url = context.getParameter("url");
       username = context.getParameter("username");
       password = context.getParameter("password");
       results.setSamplerData("url:"+url+"\n"+"username:"+username+"\n"+"password:"+password+"\n");
   }

   @Override
   public SampleResult runTest(JavaSamplerContext context) {
       try {
           results.sampleStart();// jmeter 开始统计响应时间标记
           lt = CASHttpClient.getCasLt(url);
           if ("".equals(lt)||null==lt) {
               throw new Exception("lt没有取到~！");
           }
           String redirect_url = CASHttpClient.sendCasLogin(url, username, password, lt,_eventId);

           redirect_url = rebuildRedirectUrl(redirect_url);
           if (redirect_url != null && redirect_url.length() > 0) {
               System.out.println("redirect_url:"+redirect_url);
               results.setResponseData(redirect_url, "UTF-8");
               results.setContentType("application/json");
               //results.setDataType(SampleResult.TEXT);
           }
           // System.out.println(resultData);
           results.setSuccessful(true);
       } catch (Throwable e) {
           results.setSuccessful(false);
           e.printStackTrace();
       }finally {
           results.sampleEnd();// jmeter 结束统计响应时间标记
       }
       return results;
   }