java 编写拦截器单元测试

nfg76nw0 于 2022-11-27 发布在 Java

关注(0)|答案(1)|浏览(301)

我正在做一个项目，我需要写一个拦截器，它基本上是修改头。

class AuthRequestInterceptor implements Interceptor {
  @Override public Response intercept(Interceptor.Chain chain) throws IOException {
        Request original = chain.request();

        // Request customization: add request headers
        Request.Builder requestBuilder = original.newBuilder()
                .header("Authorization", "auth-value"); 
    
        Request request = requestBuilder.build();
        return chain.proceed(request);
  }
}

我想测试拦截器是否将Authorization添加到了请求头中。

class AuthRequestInterceptorTest extends Specification {
    AuthRequestInterceptor authRequestInterceptor = new AuthRequestInterceptor();
    OkHttpClient okHttpClient = new OkHttpClient();

    void setup() {
        try {
            okHttpClient = new OkHttpClient()
                .newBuilder()
                .addInterceptor(authRequestInterceptor)
                .build();
        } catch (NoSuchAlgorithmException | KeyManagementException e) {
            throw new WebClientException(e)
        }
    }

    def "Get Authorization in to header"() {

        given:
        HashMap<String, String> headers = new HashMap<>()

        when:
        Request mockRequest = new Request.Builder()
            .url("http://1.1.1.1/heath-check")
            .headers(Headers.of(headers))
            .build()

       Response res = okHttpClient.newCall(mockRequest).execute()

        then:
        res.headers("Authorization")
    }
}

测试失败，因此我调试了AuthRequestInterceptor返回的响应，如下所示

Response{protocol=h2, code=200, message=, url=https://1.1.1.1/}

我想测试请求标头中是否添加了授权密钥，但我只能获取响应，无法确定如何获取请求标头

Java

来源：https://stackoverflow.com/questions/74575745/writing-unit-test-of-interceptors

1条答案

按热度按时间

efzxgjgh1#

帮助程序类：

package de.scrum_master.stackoverflow.q74575745;

class WebClientException extends RuntimeException {
  public WebClientException(Throwable cause) {
    super(cause);
  }
}

测试中的类别：

package de.scrum_master.stackoverflow.q74575745;

import okhttp3.Interceptor;
import okhttp3.Request;
import okhttp3.Response;

import java.io.IOException;

class AuthRequestInterceptor implements Interceptor {
  @Override
  public Response intercept(Interceptor.Chain chain) throws IOException {
    Request original = chain.request();

    // Request customization: add request headers
    Request.Builder requestBuilder = original.newBuilder()
      .header("Authorization", "auth-value");

    Request request = requestBuilder.build();
    return chain.proceed(request);
  }
}

斯波克规格：

为了隔离地测试拦截器，只需模拟拦截器链参数，确保它返回一个没有头的请求，然后验证proceed(Request)方法是用一个包含所需头的请求参数调用的。

package de.scrum_master.stackoverflow.q74575745

import okhttp3.Interceptor
import okhttp3.Request
import spock.lang.Specification

class AuthRequestInterceptorTest extends Specification {
  def "request contains authorization header"() {
    given: "a mock interceptor chain returning a prepared request without headers"
    def chain = Mock(Interceptor.Chain) {
      request() >> new Request.Builder()
        .url("http://1.1.1.1/heath-check")
        .build()
    }

    when: "running the interceptor under test"
    new AuthRequestInterceptor().intercept(chain)

    then: "the expected authorization header is added to the request before proceeding"
    1 * chain.proceed({ Request request -> request.headers("Authorization") == ["auth-value"] })
  }
}

请在Groovy Web Console中尝试。
您可能想了解Spock论证约束。

赞(0）回复(0）举报 2022-11-27

我来回答

java 编写拦截器单元测试

1条答案

相关问题

热门标签

最新问答