现在的位置: 首页 > 自动控制 > 工业·编程 > 正文

OkHttp3系列(二)MockWebServer使用

2020-08-07 06:40 工业·编程 ⁄ 共 5785字 ⁄ 字号 暂无评论

OkHttp3是由Square贡献的HTTP客户端框架,主要用在Andorid中,但是由于其易用的API、强大的功能、请求的快速等特点,也被大量采用在后端开发领域。本系列文章讲述OkHttp3的基本使用、OkHttp3的高级功能以及OkHttp3源码的解析等,请持续关注。

本篇文章是此系列的第二篇。

Mock#

mock在测试领域是很重要的一个概念。mock测试就是在测试过程中,对于某些不容易构造或者不容易获取的对象,创建用一个虚拟的对象以方便测试的测试方法。比如在Java中可以借助JMock、EasyMock等工具创建Java对象,帮助我们快速进行单元测试。

MockWebServer#

MockWebServer则是OkHttp3提供的一个快速创建HTTP服务端的工具。当我们的服务需要依赖外部HTTP应用时,可以按照预期功能快速构建外部HTTP应用,加快开发流程,快速进行单元测试,完善代码。搭配OkHttp3使用时,可以测试我们自己编写的OkHttp3客户端代码。

目前Java版本的MockWebServer最后版本的Maven坐标如下,本编文章的代码示例均基于该版本。

<dependency>

    <groupId>com.squareup.okhttp3</groupId>

    <artifactId>mockwebserver</artifactId>

    <version>3.14.9</version>

</dependency>

使用MockWebServer#

基本示例#

MockWebServer的使用很简单。

首先创建一个MockWebServer对象。

MockWebServer server = new MockWebServer();

然后创建响应内容。

MockResponse mockResponse = new MockResponse().setBody("hello, world!")

把响应内容放入MockWebServer对象。

server.enqueue(mockResponse);

启动MockWebServer。

try {

    server.start(8080);

} catch (IOException e) {

    e.printStackTrace();

}

高级功能#

模拟弱网环境响应。#

MockWebServer server = new MockWebServer();

String filePath = "C:\\Users\\weegee\\Downloads\\dm-algo-top10.pdf";

Buffer bodyBuffer = new Buffer();

bodyBuffer.readFrom(new FileInputStream(new File(filePath)));

MockResponse bigMockResponse = new MockResponse()

        .addHeader("Cache-Control", "no-cache")

        .setBody(bodyBuffer);

bigMockResponse.setBodyDelay(5, TimeUnit.SECONDS);

bigMockResponse.throttleBody(1024 * 1024, 1, TimeUnit.SECONDS);

server.enqueue(bigMockResponse);

try {

    server.start(8099);

} catch (IOException e) {

    e.printStackTrace();

    System.exit(0);

}

throttleBody(long, long, TimeUnit):MockResponse方法的第一个参数是传输字节数,第二个参数是传输第一个参数指定的字节数需要的时间,第三个参数是时间单位。

上述示例模拟每秒仅能传输1MB字节数据的网络状况,响应体大小为4.2MB,同时设置响应延迟5秒才可开始处理。则该请求正常情况下应该需要9秒多才可以结束,我们用Postman作为请求客户端模拟这一情况。

服务端请求分发#

正常的HTTP请求一般会对不同路径的请求返回不同的结果,MockWebServer通过Dispatcher支持该功能。

MockWebServer server = new MockWebServer();

final Dispatcher dispatcher = new Dispatcher() {

    @Override

    public MockResponse dispatch(RecordedRequest request) throws InterruptedException {

        if (request.getPath().equals("/v1/login/auth/")){

            return new MockResponse().setResponseCode(200);

        } else if (request.getPath().equals("/v1/check/version")){

            return new MockResponse().setResponseCode(200).setBody("version=9");

        } else if (request.getPath().equals("/v1/profile/info")) {

            return new MockResponse().setResponseCode(200).setBody("{\\\"info\\\":{\\\"name\":\"Lucas Albuquerque\",\"age\":\"21\",\"gender\":\"male\"}}");

        }

        return new MockResponse().setResponseCode(404);

    }

};

server.setDispatcher(dispatcher);

try {

    server.start(8099);

} catch (IOException e) {

    e.printStackTrace();

    System.exit(0);

}

上述示例设置了三个请求路径/v1/login/auth/、/v1/check/version、/v1/profile/info,客户端对对应路径的请求返回对应的结果。

注意: setDispatcher和enqueue不能同时使用,若是先使用enqueue再使用setDispatcher,则enqueue的响应体则会丢失,而先setDispatcher再enqueue则运行失败,抛出异常java.lang.ClassCastException。原因在于MockWebServer对于所有的响应体MockResponse都是通过一个分发器处理,下面截取MockWebServer的部分源码进行分析。

// 唯一分发器

private Dispatcher dispatcher = new QueueDispatcher();

// enqueue方法

public void enqueue(MockResponse response) {

    ((QueueDispatcher)this.dispatcher).enqueueResponse(response.clone());

}

// setDispatcher方法,调用该方法会把之前dispatcher内容抛弃,使用方法参数中的dispatcher覆盖原来的内容

public void setDispatcher(Dispatcher dispatcher) {

    if (dispatcher == null) {

        throw new NullPointerException();

    } else {

        this.dispatcher = dispatcher;

    }

}

先enqueue再setDispatcher比较好理解,但是先setDispatcher再enqueue时,发生了以下异常。

Exception in thread "main" java.lang.ClassCastException: class com.github.chengtengfei.http.OkHttp3MockWebServer$1 cannot be cast to class okhttp3.mockwebserver.QueueDispatcher

异常发生的位置跟踪到是。

((QueueDispatcher)this.dispatcher).enqueueResponse(response.clone());

也就是说,调用setDispatcher方法后,this.dispatcher就不再是QueueDispatcher类型的了,而变成了com.github.chengtengfei.http.OkHttp3MockWebServer$1这个类型。

一般类名加$1表明当前类中存在内部匿名类,所以编译后以数字代替类名称,反编译OkHttp3MockWebServer$1可获得以下内容。

class OkHttp3MockWebServer$1 extends Dispatcher {

    OkHttp3MockWebServer$1() {

    }

    public MockResponse dispatch(RecordedRequest request) throws InterruptedException {

        if (request.getPath().equals("/v1/login/auth/")) {

            return (new MockResponse()).setResponseCode(200);

        } else if (request.getPath().equals("/v1/check/version")) {

            return (new MockResponse()).setResponseCode(200).setBody("version=9");

        } else {

            return request.getPath().equals("/v1/profile/info") ? (new MockResponse()).setResponseCode(200).setBody("{\\\"info\\\":{\\\"name\":\"Lucas Albuquerque\",\"age\":\"21\",\"gender\":\"male\"}}") : (new MockResponse()).setResponseCode(404);

        }

    }

}

调用setDispatcher前,创建了dispatcher对象,该对象是一个内部类的对象,该内部类继承自okhttp3.mockwebserver.Dispatcher然后实现了dispatch(RecordedRequest):MockResponse方法,方法体中实现了自定义的分发机制,调用setDispatcher传入的参数类型是内部类的类型,和原来的QueueDispatcher不是同一类型,所以进行强制类型转换就会抛出异常。

请求记录#

MockWebServer支持对客户端的每次请求进行记录,借助该功能可以观察请求时携带的参数、请求头等内容。

这里我们以前一小节的请求分发为例构建服务端,然后打印每次请求的路径。

MockWebServer server = new MockWebServer();

final Dispatcher dispatcher = new Dispatcher() {

    @Override

    public MockResponse dispatch(RecordedRequest request) throws InterruptedException {

        if (request.getPath().equals("/v1/login/auth")){

            return new MockResponse().setResponseCode(200);

        } else if (request.getPath().equals("/v1/check/version")){

            return new MockResponse().setResponseCode(200).setBody("version=9");

        } else if (request.getPath().equals("/v1/profile/info")) {

            return new MockResponse().setResponseCode(200).setBody("{\\\"info\\\":{\\\"name\":\"Lucas Albuquerque\",\"age\":\"21\",\"gender\":\"male\"}}");

        }

        return new MockResponse().setResponseCode(404);

    }

};

server.setDispatcher(dispatcher);

try {

    server.start(8099);

} catch (IOException e) {

    e.printStackTrace();

    System.exit(0);

}

while (true) {

    RecordedRequest request = server.takeRequest();

    System.out.println(request.getPath());

}

小结#

本片文章介绍了OkHttp3提供的MockWebServer功能,并详细介绍了它的使用。下篇文章将会介绍一款由笔者打造的用来快速使用OkHttp3的框架,敬请期待。

作者:weegee

给我留言

留言无头像?