轉載地址：http://www.techv5.com/topic/696/

下面再介紹一下通過JMockit這個Java Mock工具來進行spring的單元測試，其特點是不需指定spring的配置檔案，任何物件都可以mock出來並進行關聯。

Controller

import com.odde.mail.model.Result;
import com.odde.mail.service.MailService;
import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;
import org.codehaus.jackson.map.ObjectMapper
 
;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Controller;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RequestMethod;
import org.springframework.web.bind.annotation.RequestParam;
 

import org.springframework.web.bind.annotation.ResponseBody;

import static java.lang.String.format;

@Controller
@RequestMapping("/mail")
public class MailController {
    private static final Log log = LogFactory.getLog(MailController.class);
    private final ObjectMapper mapper = new ObjectMapper
 
();

    @Autowired
    private MailService mailService;

    @RequestMapping(value = "/send", method = RequestMethod.POST, produces = "text/plain;charset=UTF-8")
    public
    @ResponseBody
    String send(@RequestParam("recipients") String recipients,
                @RequestParam("subject") String subject,
                @RequestParam("content") String content) throws Exception {
        log.debug("mail controller send start");
        log.debug(format("recipients:%s", recipients));
        log.debug(format("subject:%s", subject));
        log.debug(format("content:%s", content));
        Result mailResult = mailService.send(recipients, subject, content);
        String result = mapper.writeValueAsString(mailResult);
        log.debug(format("result:%s", result));
        log.debug("mail controller send finish");
        return result;
    }
}

首先我們還是來看一下使用了JMockit的Controller單元測試是怎麼寫的，Controller的功能程式碼可見上面。

import com.odde.mail.model.Result;
import com.odde.mail.service.MailService;
import mockit.Expectations;
import mockit.Injectable;
import mockit.Tested;
import mockit.integration.junit4.JMockit;
import org.junit.Test;
import org.junit.runner.RunWith;

import static org.hamcrest.CoreMatchers.is;
import static org.junit.Assert.assertThat;

@RunWith(JMockit.class)
public class MailControllerTest {
    @Tested
    MailController mailController;

    @Injectable
    private MailService mailService;

    @Test
    public void should_return_status_success_when_send_mail_success() throws Exception {
        new Expectations() { {
            mailService.send("[email protected]", "test", "test");
            result = new Result("成功");
        } };

        String result = mailController.send("[email protected]", "test", "test");

        assertThat(result, is("{\"status\":\"成功\"}"));
    }
}

• @RunWith(JMockit.class): 指定單元測試的執行類為JMockit.class;
• @Tested: 這個Annotate是指被測試類，在這個測試案例中我們要測試的是MailController，所以我們給其打上這個標籤;
• @Injectable: 這個Annotate可以將物件進行mock並自動關聯到被測試類，而不需要通過其他檔案類似spring的配置檔案等來進行關聯;
• @Expectations: mock物件mailService的send方法，讓其返回一個Result物件;

Service

再來看一下Service的單元測試要怎麼改寫，同樣Service的功能程式碼可以看上一篇Post。

import com.odde.mail.model.Recipient;
import com.odde.mail.model.Result;
import com.odde.mail.repo.RecipientRepository;
import mockit.Injectable;
import mockit.NonStrictExpectations;
import mockit.Tested;
import mockit.integration.junit4.JMockit;
import org.junit.Test;
import org.junit.runner.RunWith;

import java.util.List;

import static java.util.Arrays.asList;
import static org.hamcrest.core.Is.is;
import static org.junit.Assert.assertThat;

@RunWith(JMockit.class)
public class RecipientServiceTest {

    @Tested
    private RecipientService recipientService;

    @Injectable
    private RecipientRepository recipientRepository;

    @Test
    public void should_return_success_when_add_recipient_not_exist() throws Exception {
        Result result = recipientService.add("Tom", "[email protected]");
        assertThat(result.getStatus(), is("成功"));
    }
}

相對Controller Test這裡少了一步對recipientRepository物件findByEmail方法的mock，因為如果不通過Expectations進行方法mock的話，方法會預設返回null，而我們要測試的場景正是需要findByEmail方法返回null，所以mock方法這一步我們也省了。改寫後的整體程式碼也比原來的少了很多，而且速度更快。

適當使用Mock框架

JMockit功能非常強大，不僅可以輕鬆處理上面的這些測試場景，還可以對static,final,private等方法進行mock，可以讓你的單元測試毫無阻礙的進行。

但是如果過度的使用Mock框架，會讓功能程式碼的壞味道被掩蓋。本來單元測試的設計可以讓你發現功能程式碼上的一些設計是否合理，比如有沒有緊耦合等，但使用JMockit可以讓你在設計不合理的程式碼上也可以輕鬆地進行單元測試，這樣你就很難發現功能程式碼上的問題了。

所以建議JMockit等類似的mock框架還是要謹慎使用，首先要保證功能程式碼設計合理，滿足面向物件設計的要求，再來考慮提高單元測試效率的問題。