接口测试中Groovy引擎的接入实现！

接口测试中Groovy可以作为上下游接口参数化传递的前置脚本和后置脚本使用，无缝衔接Java语法，groovy的引入对于动态参数化的设置方便很多。

其中核心部分就是接入groovy的引擎，下面介绍groovy引擎的实现逻辑。

• pom文件groovy核心依赖

            org.codehaus.groovy
            groovy
            3.0.7
        
        
            org.kohsuke
            groovy-sandbox
            1.19
        

1、Groovy引擎中添加常用的import
静态代码块中引入

一些基础的依赖，比如：JSONObject、JSONArray、JsonPath等工具类

static {
        config = new CompilerConfiguration();
        // 添加默认import
        config.addCompilationCustomizers(new ImportCustomizer().addImports(
                "com.taltools.script.exception.StepFailureException",
                "com.jayway.jsonpath.JsonPath",
                "com.alibaba.fastjson.JSONObject",
                "com.alibaba.fastjson.JSONArray",
                "groovy.json.JsonSlurper",
                "org.apache.commons.codec.digest.DigestUtils",
                "org.apache.commons.codec.digest.HmacUtils"));
        // 添加线程中断拦截器
        config.addCompilationCustomizers(new ASTTransformationCustomizer(ThreadInterrupt.class));
        // 添加线程中断拦截器，可中断超时线程，当前定义超时时间为30s
        config.addCompilationCustomizers(new ASTTransformationCustomizer(Collections.singletonMap("value", INTERRUPT_TIME_OUT), TimedInterrupt.class));
        // 沙盒环境
        config.addCompilationCustomizers(new SandboxTransformer());

        NO_RUNTIME_SANDBOX = new NoRuntimeSandbox();
        NO_SYSTEM_HARM_SANDBOX = new NoSystemHarmSandbox();
    }

• 设置缓存容量&过期时间

{
        lruCache = CacheBuilder.newBuilder()
                .maximumSize(1000) //最大容量
                .expireAfterAccess(12, TimeUnit.HOURS) //缓存过期时长
                .concurrencyLevel(Runtime.getRuntime().availableProcessors())// 设置并发级别为cpu核心数
                .build();
    }

2、Groovy核心实现，上下文参数绑定

• 上下文参数变量名绑定

@Override
    public ScriptContext runScript(String scriptContent, ScriptContext context) {
        if (scriptContent == null || scriptContent.isEmpty()) {
            throw new IllegalArgumentException("输入的script内容不能为空");
        }
        log.debug("执行Groovy脚本: {}", scriptContent);
        Binding binding = generateBinding(context);
        parseScript(scriptContent, binding).run();
        ScriptContext retContext = new ScriptContext();
        retContext.setVariables(binding.getVariables());
        return retContext;
    }

    public long getCacheCount() {
        return lruCache.size();
    }

    protected Script parseScript(String scriptContent, Binding binding) {
        String md5 = DigestUtils.md5Hex(scriptContent);
        String sha1 = DigestUtils.sha1Hex(scriptContent);
        String key = md5 + "-" + sha1;
        Class scriptClass = lruCache.getIfPresent(key);
        if (scriptClass == null) {
            registerSandbox();
            GroovyShell shell = new GroovyShell(config);
            Script script = shell.parse(scriptContent);
            scriptClass = script.getClass();
            lruCache.put(key, scriptClass);
            log.debug("未命中脚本LRU缓存，创建脚步对象并存入缓存，当前缓存数量: {}", getCacheCount());
        } else {
            log.debug("命中脚本LRU缓存, key: {}, 当前缓存数量: {}", key, getCacheCount());
        }
        return InvokerHelper.createScript(scriptClass, binding);
    }

3、调佣Groovy引擎设置参数名测试

• 在单元测试中测试接受groovy脚本的返回值

@Test
    public void testGroovy(){
        String req = "    outParams['name'] = testGroovy();\n" +
                "    static String testGroovy(){\n" +
                "        def name = 'java'\n" +
                "        def greeting = \"Hello ${name}\"\n" +
                "        return greeting\n" +
                "    }";
        GroovyEngine engine = new GroovyEngine();
        ScriptContext context = new ScriptContext();
        LinkedHashMap outParams = new LinkedHashMap<>();
        context.setVariable("outParams",outParams);
        ScriptContext ret = engine.runScript(req,context);
        System.out.println(JsonUtils.obj2json(ret));
        System.out.println(outParams);

    }

• 控制台打印结果

• 前端页面效果

关于groovy在接口测试平台中的落地后续不断补充，欢迎关注！Coding测试