浅谈Jmockit使用

最新推荐文章于 2024-06-24 09:37:01 发布

fenix_Lee

最新推荐文章于 2024-06-24 09:37:01 发布

阅读量2.9k

点赞数 1

本文链接：https://blog.csdn.net/changsu4615/article/details/46802517

版权

最近公司里要求使用Jmockit来辅助单元测试。之前也没接触过，研究了几天，今天就跟大家谈谈我怎么使用Jmockit。写一个小例子供大家参考。

如有不对的地方还请大家多多指正

对于Mock的作用这里就不详细介绍了，不清楚的可以自行百度。笔者使用的是1.15版本的Jmockit（默认读者已经把相应的包导好）

在开始测试之前我们先写一个简单的Java类，代码如下

public class GetServiceForBook {


  public int getService(String bookType) {

    if ("novel".equals(bookType)){
      return 1;
    }
    else {
      return 2;
    }
  }
}

以上是一个极其简单的一个Java类，里面定义了一个getService方法。看了这个方法后可以确定，调用这个方法后得到的结果不是1就2.下面我们就来写一个小单元测试。

import static org.junit.Assert.*;
import org.junit.Test;

public class GetServiceForBookTest {
  
  private GetServiceForBook getServiceForBook = new GetServiceForBook();
  
  @Test
  public void testBookService() {
      
   int test = getServiceForBook.getService("novel");
    
   assertEquals(1,test);
 }
}

对于上面的单元测试我相信不会有人看不懂，而且简单运行一下就是一个长长的绿条。那下面我们就开始使用Mock来“修改”我们的GetServiceForBook类。

刚刚也说明了，对于getService方法的返回值不是1就是2。那如果我想让getService返回3怎么办？很简单，使用Mock。先上Expectations的方式。

import static org.junit.Assert.*;
import mockit.Expectations;
import mockit.Injectable;

import org.junit.Test;

public class GetServiceForBookTest {
  
  @Injectable
  private GetServiceForBook getServiceForBook;
  
  @Test
  public void testBookService() {
      
   new Expectations(){
     {
       getServiceForBook.getService(anyString);
       result = 3;
       times = 1;
     }
     
   };
    
   int test = getServiceForBook.getService("novel");
   assertEquals(3,test);
 }
}

以上就是使用Mock的方式把getService方法给“修改”了。所以底下的断言是看test和3的比较。很明显，这里用了Expectations把getService的返回值改成了3,也就是说在这个测试里不论我传进去的参数是什么，它的返回都是3.（使用@Injectable系统会自动注入，创建实例。）下面在给一个使用MockUp方式

import static org.junit.Assert.*;
import mockit.Mock;
import mockit.MockUp;

import org.junit.Test;

public class GetServiceForBookTest {
  

  private GetServiceForBook getServiceForBook = new GetServiceForBook();
  
  @Test
  public void testBookService() {
    
   new MockUp<GetServiceForBook>(){
     @Mock
     public int getService(String bookType)
     {
       return 3;
     }
   };
    
   int test = getServiceForBook.getService("novel");
   assertEquals(3,test);
 }
}

可以看出这两个Mock的结果是一样的都是把getService的方法改成3. 但是MockUp方式跟Expectations不同的是，MockUp把getService方法重新写了一遍并不像Expectations那样只要设定下变量就可以改变getService方法。

现在，我们稍微修改下原来的GetServiceForBook类。

package cn.com.liandisys.book.service;

public class GetServiceForBook {


  JustForTest jft = new JustForTest();

  public GetServiceForBook()
  {

  }

  public GetServiceForBook(JustForTest jft)
  {
    this.jft = jft;
  }

  public int getService(String bookType) {

    boolean flag = jft.getResult();
    System.out.println(flag);
    if ("novel".equals(bookType)) {
      return 1;
    }
    else {
      return 2;
    }
  }
}

看到里面多了一个JustForTest的类，下面把JustForTest类代码贴出来

public class JustForTest {

  public boolean getResult() {
    return false;
  }

}

依然是一个简单的Java类，我们可以看出这个JustForTest类里有一个getResult的方法，这个方法返回的永远是false. 在GetServiceForBook类里申明了一个JustForTest类并调用了getResult方法然后打印出来。不出意外这个输出结果应该永远都是false。那下面我们用Mock把getResult方法“修改”成true。

import static org.junit.Assert.*;
import mockit.Expectations;
import mockit.Injectable;
import mockit.Tested;

import org.junit.Test;

public class GetServiceForBookTest {
  
  @Tested
  private GetServiceForBook getServiceForBook;
  
  @Injectable
  private JustForTest jft;
  
  @Test
  public void testBookService() {
    
   new Expectations(){
     
     {
        jft.getResult();
        result = true;
        times = 1;
     }
     
   };
    
   int test = getServiceForBook.getService("novel");
   assertEquals(1,test);
 }
}

由于我们并没有mock掉getService所以返回值不是1就是2，对于上面的单元测试，给的参数是novel那返回值就应该是1。然后我们运行一下可以发现控制台输出的是true而不是false了，这是因为我们把JustForTest类的getResult方法给Mock掉了。

里面还有两个注解一个@Tested一个是@Injectable。第一个@Tested不能变只能是这个，第二个可以写成@Mocked。如果第一个改成@Mocked的话输出结果就变成0，也就是说不过你的参数是什么结果都是输出0。这是因为在你使用Mocked注解时，如果没有进行“修改”，那系统就会自己帮你把这个类里的所有方法都Mock掉，而mock的内容就是返回默认值。（int的默认值是0）。

下面给出使用MockUp方式的

import static org.junit.Assert.*;
import mockit.Mock;
import mockit.MockUp;

import org.junit.Test;

public class GetServiceForBookTest {
  
  private GetServiceForBook getServiceForBook = new GetServiceForBook();
  
  @Test
  public void testBookService() {
    
    new MockUp<JustForTest>(){
      @Mock
      public boolean getResult()
      {
        return true;
      }
    };
    
    int test = getServiceForBook.getService("novel");
    assertEquals(1,test);
 }
}

这里就不多解释了，跟上面的MockUp例子很像，同样也是把getResult输出true。

通过查阅Jmockit文档，发现MockUp有一个方法叫getMockInstance。也就是创建一个Mock实例。我们可以这样理解，每一个需要被Mock的代码都可以看作是一个对象。就拿上面getResult例子来说JustForTest可以把它看作是一个Mock对象。所以我们还可以改成下面的形式。

import static org.junit.Assert.*;
import mockit.Mock;
import mockit.MockUp;

import org.junit.Test;

public class GetServiceForBookTest {
  
  private GetServiceForBook getServiceForBook;
  
  @Test
  public void testBookService() {
    
    JustForTest mock = new MockUp<JustForTest>(){
      @Mock
      public boolean getResult()
      {
        return true;
      }
    }.getMockInstance();
    
    getServiceForBook = new GetServiceForBook(mock);
    int test = getServiceForBook.getService("novel");
    assertEquals(1,test);
 }
}

这就是为什么我去重写getServiceForBook的构造函数，把Mock创建出来的对象赋给getServiceForBook里的JustForTest，这样也能达到Mock的目的

这次就先写到这里，其实对于Jmockit来说，这个只是沧海一栗，里面还有多东西需要研究，这里就先跟大家分享这么多，希望能对读者有用。

这是我第一次写博客，可能还有很多不足的地方，还请大大多多指教。