fix(接口测试): Mock期望的后置脚本中增加对第三方jar包的调用

Mock期望的后置脚本中增加对第三方jar包的调用
This commit is contained in:
song-tianyang 2022-02-22 18:15:43 +08:00 committed by 刘瑞斌
parent bf5f8b5cbd
commit 3db22f6061
2 changed files with 57 additions and 7 deletions

View File

@ -2,11 +2,18 @@ package io.metersphere.api.mock.utils;
import com.alibaba.fastjson.JSONObject;
import io.metersphere.api.dto.mock.RequestMockParams;
import io.metersphere.base.domain.JarConfig;
import io.metersphere.commons.utils.CommonBeanFactory;
import io.metersphere.commons.utils.LogUtil;
import io.metersphere.service.JarConfigService;
import org.apache.commons.lang3.StringUtils;
import javax.script.ScriptEngine;
import javax.script.ScriptEngineManager;
import java.io.File;
import java.lang.reflect.Method;
import java.net.URL;
import java.net.URLClassLoader;
import java.util.ArrayList;
import java.util.List;
import java.util.Map;
@ -36,23 +43,66 @@ public class MockScriptEngineUtils {
}
}
public void loadJar(String jarPath) throws Exception {
File jarFile = new File(jarPath);
// 从URLClassLoader类中获取类所在文件夹的方法jar也可以认为是一个文件夹
Method method = null;
try {
method = URLClassLoader.class.getDeclaredMethod("addURL", URL.class);
} catch (NoSuchMethodException | SecurityException e1) {
e1.printStackTrace();
}
// 获取方法的访问权限以便写回
boolean accessible = method.isAccessible();
try {
method.setAccessible(true);
// 获取系统类加载器
URLClassLoader classLoader = (URLClassLoader) ClassLoader.getSystemClassLoader();
URL url = jarFile.toURI().toURL();
method.invoke(classLoader, url);
} catch (Exception e) {
e.printStackTrace();
} finally {
method.setAccessible(accessible);
}
}
/**
* 加载jar包
*/
private void loadJars() {
JarConfigService jarConfigService = CommonBeanFactory.getBean(JarConfigService.class);
if (jarConfigService != null) {
List<JarConfig> jars = jarConfigService.list();
jars.forEach(jarConfig -> {
try {
this.loadJar(jarConfig.getPath());
} catch (Exception e) {
e.printStackTrace();
LogUtil.error(e.getMessage(), e);
}
});
}
}
public ScriptEngine getBaseScriptEngine(String scriptLanguage, String url, Map<String, String> headerMap, RequestMockParams requestMockParams) {
ScriptEngine engine = null;
try {
if (StringUtils.isEmpty(scriptLanguage)) {
return null;
}
String preScript = "";
if (StringUtils.equalsIgnoreCase(scriptLanguage, "beanshell")) {
ScriptEngineManager scriptEngineFactory = new ScriptEngineManager();
engine = scriptEngineFactory.getEngineByName(scriptLanguage);
String preScript = this.genBeanshellPreScript(url, headerMap, requestMockParams);
engine.eval(preScript);
engine = scriptEngineFactory.getEngineByName("beanshell");
preScript = this.genBeanshellPreScript(url, headerMap, requestMockParams);
} else if (StringUtils.equalsIgnoreCase(scriptLanguage, "python")) {
ScriptEngineManager scriptEngineFactory = new ScriptEngineManager();
engine = scriptEngineFactory.getEngineByName(scriptLanguage);
String preScript = this.genPythonPreScript(url, headerMap, requestMockParams);
engine.eval(preScript);
preScript = this.genPythonPreScript(url, headerMap, requestMockParams);
}
this.loadJars();
engine.eval(preScript);
} catch (Exception e) {
LogUtil.error(e);
}

View File

@ -38,8 +38,8 @@ export default {
lineDivMarginTopHeight: 0,
lineDivBottomHeight: 0,
filterTypes: [
{id: 'And', label: 'And'},
{id: 'Or', label: 'Or'},
{id: 'And', label: 'AND'},
{id: 'Or', label: 'OR'},
],
}
},