REST-assured 获取日志到文件并结合 Allure 报告进行展示

使用 Rest-assured 集合 Allure 运行完用例之后，在生成的报告中只有断言信息，没有请求的日志信息。而当我们的用例失败时，特别是接口失败时，请求日志是分析原因的第一手资源。那如何将 Rest-assured 产生的日志存入 Allure 里，并且能和用例一一对应起来呢？

1. 需求与方向

1.1 问题产生

在使用 Rest-assured 集合 Allure 运行完用例之后，查看生成的报告信息如下：

我们可以看到在生成的报告中只有断言信息，而没有请求的日志信息，而当我们的用例失败时，特别是接口失败时，请求日志是分析原因的第一手资源；

1.2 需求产生

其实Rest-assured是有请求日志的，可以通过在given()和then()后面加上.log().all()来打印全部的日志信息：

这块不是本文想介绍的重点，所以想了解的可以参考以往的文章或直接阅读官方文档

那么问题来了，如何将这打印出来的日志信息都"转移"到 Allure 报告中呢？并且能和用例一一对应起来，然后就开始了探索之路~

1.3 思路产生

下面的解决方案都是在我现阶段研究 Rest-assured 及 Allure 的基础上实现的，可能研究并不透彻不全面，也许有更好的方法使用，请大家多提反馈意见。

首先来看一下 Allure 报告可以如何展示日志，在学习 Allure 的过程中发现 Allure有添加附件展示的功能，那么我就直接想到将日志能存入文件然后添加到报告附件不就可以了吗？由此，Allure 端的解决方向确定。
接下来就是要想法办将 Rest-assured 产生的日志存入文件了;

整体思路：
【Rest-assured打印日志】- 【Rest-assured日志存入文件】- 【文件以附件形式传入Allure】- 【Allure展示附件日志】

2. Allure附件

先看一下Allure 添加附件的两种方法：

@Attachment：在方法上添加注解@Attachment，方法的返回值就会作为附件上传，可添加展示文本和附件类型

@Attachment(value = "Page screenshot", type = "image/png")

Allure.addAttachment：通过addAttachment方法指定要添加的附件和展示信息

public static void addHttpLogToAllure() {
        try {
            Allure.addAttachment("接口请求响应日志",
                    new FileInputStream("src/main/resources/test.log"));
        } catch (FileNotFoundException e) {
            e.printStackTrace();
        }
    }

3、保存日志

如果是我们自己打印的日志信息，可以任意保存或直接使用 log4j 即可，但是请求的日志是由 Rest-assured 产生的，这可能就需要去查阅框架相关 log 方面的文档资料。

3.1 初始版

3.1.1 方案选择

由于在框架中，我已经进行了封装，每个接口请求后都会返回 response 信息。所以一开始我想着从拿到 response 信息进行存储，查阅官方文档，寻找 response 信息获取的相关 API，发现 response.asString();可以获取到 json body 的信息，就先尝试使用。

//// Get the response body as a String
response.asString();

// Get all headers
Headers allHeaders = response.getHeaders();
// Get a single header value:
String headerName = response.getHeader("headerName");

// Get all cookies as simple name-value pairs
Map<String, String> allCookies = response.getCookies();
// Get a single cookie value:
String cookieValue = response.getCookie("cookieName");

// Get status line
String statusLine = response.getStatusLine();
// Get status code
int statusCode = response.getStatusCode();

• 先创建方法，用于接收 response 获取信息

@Attachment("响应报文")
public static String respondBody(Response response) {


    boolean prettyFormat = true;
    JSONObject jsonObject = JSONObject.parseObject(response.asString());
    String responseBody = JSONObject.toJSONString(jsonObject,prettyFormat);


    return responseBody;
}

• 再创建方法，用于接收请求信息，由于我的所有请求信息都传入了一个 Restful 对象中且未找到 Rest-assured 关于请求信息直接获取的 API，这里我就直接取 Restful对象

Restful对象：

import lombok.Data;
import java.util.HashMap;

@Data
public class Restful {

    public String url;
    public String method;
    public HashMap<String,Object> header = new HashMap<>();
    public HashMap<String,Object> query = new HashMap<>();
    public HashMap<String,Object> pathQuery = new HashMap<>();
    public String body;
}

接收请求信息方法

@Attachment("请求信息")
public static String requestBody(Restful restful) {
    //报告展现请求信息
    return restful.toString();
}

• 最后创建一个总的接收方法加入请求流程中，在每个请求结束后获取日志信息进行附件添加

public static void getRequestAndRespondBody(Restful restful, Response response){
        requestBody(restful);
        respondBody(response);
    }

3.1.2 结果展示

从结果可以看到请求和响应报文已经成功展示，说明这种实现的思路的可行性，只是展示的日志信息还不满意，还是先想要全部的请求和响应信息且是格式化后的，不仅仅只有报文，继续探索~

3.2 RestAssured.config().logConfig

3.2.1 方法实现
在研究过程中发现RestAssured提供了logConfig方法，可以将原本在Console中打印的信息指定格式化输出到文件中，具体用法如下(这里指演示重点实现原理部分，其余封装细节太冗余就不展示了)：

• WriterOutputStream用到依赖如下：

<dependency>
    <groupId>commons-io</groupId>
    <artifactId>commons-io</artifactId>
    <version>2.4</version>
</dependency>

• 在请求中设置log打印方式和存储路径

public void addLogToFile(){
  try (FileWriter fileWriter = new FileWriter("src/main/resources/test.log");
       PrintStream printStream = new PrintStream(new WriterOutputStream(fileWriter), true)) {
       RestAssured.config = RestAssured.config().logConfig(LogConfig.logConfig().defaultStream(printStream));
       given().XXX.log().all().
       when().XXX
       then().log().all();
  } catch (IOException e) {
       e.printStackTrace();
  }
}

RestAssured提供的logConfig方法目前发现只能覆盖，无法append，不过这正好符合我们的需求，每个用例的接口请求都只一一对应各自的日志信息，这样每执行一个接口，保存一份日志信息；下一个接口执行的时候就会覆盖成新的当前接口信息保存展示

• 创建附件添加方法加入请求流程中，在每个请求结束后获取日志信息进行附件添加

public static void addHttpLogToAllure() {
   try {
        Allure.addAttachment("接口请求响应日志",
                new FileInputStream("src/main/resources/test.log"));
    } catch (FileNotFoundException e) {
        e.printStackTrace();
    }
}

• 整体的流程思路就是：

@Test
void testAllureReport(){
  addLogToFile();
  addHttpLogToAllure();
}

3.2.2 结果展示
在下面展示的用例中有2个接口请求，可以看到分别记录展示了，且格式与 Console 中格式化打印的保持一致。

3.3 RestAssured.filters

3.3.1 方法实现
Rest-assured 提供了过滤器 Filters，利用它可以串改请求，设置鉴权信息，过滤 log 等，具体的可在官网中进行学习研究，这里主要用到 RequestLoggingFilter()和 ResponseLoggingFilter() 来实现我们的需求。

RequestLoggingFilter() 和 ResponseLoggingFilter() 可以将所有的请求和响应的 log 进行打印，而我们想要的是将 log 存入文件，因此还要借助方法 logRequestTo(PrintStream stream)，指定 log 的格式化输出到文件中：

FileWriter fileWriter = null;
try {
    fileWriter = new FileWriter("src/main/resources/test.log");
} catch (IOException e) {
    e.printStackTrace();
}
PrintStream printStream = new PrintStream(new WriterOutputStream(fileWriter), true);
RestAssured.filters(new RequestLoggingFilter().logRequestTo(printStream),new ResponseLoggingFilter().logResponseTo(printStream));
附件添加复用上述的方法：

AllureAttachment.addHttpLogToAllure();

3.3.2 结果展示

结果依然实现了我们的需求

更多技术文章分享及测试资料点此获取