Skip to main content

Command Palette

Search for a command to run...

喝一杯mockito

Published
2 min read
喝一杯mockito

mockito 是最常用的 java mock 库, 用于模拟创建各种对象和行为。本文会介绍一下 基础库 mockitio的用法,进行一些练习。mockitio 通常不单独使用,配合 Junit一起使用。

下面列举一下用法,

模拟对象

创建mock 对象,并且可以要求,这个对象的成员方法符合我们的行为要求。

    Foo foo= Mockito.mock(Foo.class);

直接返回

直接返回,我们需要的值

when thenReturn

  @Test
  void returnWhatIWant() {
    Foo mock = Mockito.mock(Foo.class);
    Mockito.when(mock.say()).thenReturn("nice to meet you");
    String result = mock.say();
    Assertions.assertEquals("nice to meet you", result);
  }

定制方法

定制方法,改变方法的实现 when thenAnswer

@Test
void returnBaseOneParamOrSomething() {
  Foo mock = Mockito.mock(Foo.class);
  Mockito.when(mock.echo(anyString()))
      .thenAnswer(
          new Answer<String>() {
            @Override
            public String answer(InvocationOnMock invocationOnMock) throws Throwable {
              String argument = invocationOnMock.getArgument(0, String.class);
              return "hi,"+argument;
            }
          });
  String result = mock.echo("all");
  Assertions.assertEquals("hi,all", result);
}

验证方法执行次数 和 时间

  @Test
  void verifyMethod() {
    Foo foo = Mockito.mock(Foo.class);
    foo.say();

    Mockito.verify(foo).say();
    Mockito.verify(foo, Mockito.times(1)).say();
    Mockito.verify(foo, Mockito.timeout(1000)).say();
  }

模拟静态方法

改变静态方法的行为(使用 mockito-inline 作为依赖)

@Test
  void mockStaticMethod() {
    try (MockedStatic<Foo> fooMockedStatic = Mockito.mockStatic(Foo.class)) {
      fooMockedStatic.when(() -> Foo.hold()).thenReturn("hi");
      String result = Foo.hold();
      Assertions.assertEquals("hi", result);
    }

    try(MockedStatic<Foo> fooMockedStatic = Mockito.mockStatic(Foo.class)) {
      fooMockedStatic.when(()->Foo.reply(anyString())).thenAnswer(
          new Answer<String>() {
            @Override
            public String answer(InvocationOnMock invocationOnMock) throws Throwable {
              String arg = invocationOnMock.getArgument(0,String.class);
              return "hi," + arg;
            }
          }
      );
      String result = Foo.reply("1024");
      Assertions.assertEquals("hi,1024",result);
    }
  }

模拟构造函数

改名构造函数的行为,使得创建出来的都是模拟对象。

@Test
  void mockConstruction() {
    try (MockedConstruction<Foo> fooMockedConstruction = Mockito.mockConstruction(Foo.class)) {
      Foo foo = new Foo();
      foo.say();
      Mockito.verify(foo).say();
    }

    try (MockedConstruction<Foo> fooMockedConstruction =
        Mockito.mockConstruction(
            Foo.class,
            new MockedConstruction.MockInitializer<Foo>() {
              @Override
              public void prepare(Foo foo1, MockedConstruction.Context context) {
                Mockito.when(foo1.say()).thenReturn("mock");
              }
            })) {
      Foo foo3 = new Foo();
      String result = foo3.say();
      Mockito.verify(foo3, Mockito.times(1)).say();

      Assertions.assertEquals("mock", result);
    }
  }

参数匹配

只对某个方法的特定参数进行模拟,其他参数不模拟

  @Test
  void mockSpecificArgument() {
    Foo foo = Mockito.mock(Foo.class);
    Mockito.when(foo.echo(eq("hello"))).thenReturn("nonono");
    Assertions.assertEquals("nonono", foo.echo("hello"));
    Mockito.verify(foo).echo("hello");

    Assertions.assertNull(foo.echo("other"));
    Mockito.verify(foo).echo("other");
  }

待续

TBD

总结,上面是我常用的一些 mockito 方法。

More from this blog

Google Best Practice For Java Libraries (JLBP-0)

this is a chinese tranlation from Google Best Practice For Java Libraries Google Java库最佳实践 Google Java库的最佳实践是一些规则,旨在减少使用相关的Java库时的问题。这些实践来自于多年来维护开源Java库的经验,并且受到了从错误中吸取的许多宝贵教训的启发。我们发现遵循这些规则可以得到质量更高的Java库,减少依赖冲突和其他类型的问题。这个规则列表是开放的,所以可能会不时地添加新的规则。 最佳实践...

Nov 22, 20231 min read

【阅读】uml精粹 类图

本书的整体脉络如下: class diagram 表示对象的类型以及其间存在的静态关系。用一个长方形表示类,用连线表示类之间的关系。 具体类图 长方形 对象的类型,也就是类有三部分表示而成 类名 (class name) 属性 (attribute) 操作 (operation) 属性 (attribute) 其中抽象来说,类的成员变量 可以有两种表示, 属性表示 和 关联表示。 属性表示:就是画在框里, 一般写 包括 可见行, 类型,名称 ,也可以写默认值 和 注释。 也可以标注是否是 ...

Sep 4, 20221 min read

【夜读源码】grpc-java 客户端发送 (一)

快速回顾 接上文,一次 grpc 的请求包括四个步骤,客户端发送请求到网络,是第一步。发送过程,Stub 把请求传给了ClientCall, ClientCall 传给了 ClientStream,ClientStream 有Netty 的实现,内部会把数据传输到网络上。 本文主要介绍 blockingUnaryCall 是如何使用 ClientCall ClientChannel 工作的, waitAndDrain 和 ClientCall 的方法如何对应。 Stub 外层简述 最外层的Stu...

Sep 4, 20224 min read
【夜读源码】grpc-java  客户端发送 (一)

【夜读源码】grpc-java 概览

gRPC 是 Google 开源的一套语言无关基于HTTP2协议的 RPC 框架,其中 Java 的实现称为 grpc-java,代码地址在 https://github.com/grpc/grpc-java。 RPC (remote process call)的含义是程序执行本地的一个方法(看上去像本地执行),实际上会在远程程序代码内执行,这个过程会涉及到网络通讯和传输协议。gRPC作为 RPC 一种实现,传输的网络协议是HTTP2,传输的内容按照 Protocol Buffer 进行编码,并...

Aug 16, 20223 min read
【夜读源码】grpc-java 概览

Xiao Kun's Infra Insight Blog

14 posts

Focus on interpreting FANG/Big Factory Engineering Blog, extracting practical experience of Infra that can be implemented