想在java接口自动化里用上Python的requests?这样做就可以了

想在java接口自动化里用上Python的requests?这样做就可以了

相信现在很多的公司自动化测试重点都在接口层,因为接口测试更加接近代码底层,相对于UI自动化,接口自动化有着开发更快、覆盖更全、回报率高等优点。

接口自动化代码实现不难,本质上就是代码模拟发送请求,然后判断返回结果是否符合预期即可。目前也有很多成熟开源的可以发送http请求的库,比如python的requests库。requests库易上手,使用起来也非常便捷。比如,我们要测试一个查询用户名的接口。

import pytest
import requests

def test_user_info():
    ‘‘‘查询用户信息‘‘‘
    r = requests.get(url, headers=HEADER)
    result = r.json()
    assert result["status"] == 0
    assert result["data"]["username"] == "把苹果咬哭"

这样就实现了一个接口的测试,用python(语言)+requests(第三方库)+pytest(测试框架)来搭建一套接口自动化项目其实还是很便捷的。

用java造一个“requests”出来

但是对于想使用java语言来搭建的话,暂时就没有像Python的requests这么便捷的库了。java中也有一些可以模拟发送请求的库,比如httpclient、okhttp等。不过这两个用起来的便捷程度跟requests就差远了。不过我们可以去封装出一个类似于requests使用方式的类,组里大佬提供了主要封装,我也根据自己项目需求进行补充,这里分享一下。

在java里实现需要借助另一个核心类,那就是RestTemplate,是Spring提供的用于访问Rest服务的客户端。具体有兴趣的小伙伴可以自行搜索进行更全面的学习。下面就来demo出一个可以模仿requests发送带请求body的post请求,还要带上header。

首先,我们用requests的时候,通常是这样实现的:

import pytest
import requests

header = {
        ‘Authorization‘:"我是鉴权"
    }

event = {
    "id": 1,
    "create_user": "把苹果咬哭" 
}

def test_event_add():
    ‘‘‘测试添加项目‘‘‘
    r = requests.post(url, data=event, headers=header)
    result = r.json()
    
    assert result["status"] == 0

所以,我们可以知道关键就在于requests.post(url, data=event, headers=header),传入测试接口url,请求body参数,还有header之后发送请求,这样的调用方式。其实只需要将restTempate简单的封装一下即可。

import org.springframework.web.client.RestTemplate;
import com.alibaba.fastjson.JSONObject;

public class DemoRestUtils {
    RestTemplate restTemplate;
    //提供一个默认构造方法
    public RestSuperUtils() {
        restTemplate = new RestTemplate();
    }
    /**
     * POST Body请求方式,带有header 请求Body JSON Body体
     *
     * @param url
     * @param headers
     * @param jsonBody
     * @return
     */
    public JSONObject postForBody(String url, HttpHeaders headers, JSONObject jsonBody) {

        HttpEntity<?> httpEntity = new HttpEntity<>(jsonBody, headers);
        ResponseEntity<JSONObject> responseEntity = restTemplate.exchange(url, HttpMethod.POST, httpEntity,
                JSONObject.class);
        JSONObject jsonResponse = responseEntity.getBody();
        return jsonResponse;
    }
}

这里可以将header跟请求body整合成HttpEntity,这样的话就可以直接调用restTemplate提供的exchange方法,传入接口地址,请求方式,还有请求实体即可完成调用,而且还可以通过ResponseEntity的getBody方法获得返回,再加一卜转化后,返回json格式的请求响应即可。这样我们的用例可以这样写了:

@SpringBootTest
public class TestDemo {
    public DemoRestUtils requests;
    private final String apiPath = "/addEvent";

    void testQueryDemo() {
        String reqData = "{\"id\": 1}";
        //调用接口
        JSONObject reqData = JSON.parseObject(testEvent);
        JSONObject response = requests.postForBody(apiPath, reqData);
    }
}

最后可以对返回的response做分析,拿到你想要断言的字段,可以使用fastjson的JSONPath.eval方法,可以支持xpath定位到你要的字段,拿到值。

上述只是伪代码,导入依赖如果你使用jetbrain IDea的话会有提示导入,本文目的只是介绍一种思路,并不适合直接复制引用。欢迎交流!

相关推荐