在java接口自动化里用上Py的requests？这样做就好

Posted 2021-05-29 测试baby

tags:

篇首语：本文由小常识网(cha138.com)小编为大家整理，主要介绍了在java接口自动化里用上Py的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);
    }
}