springboot(6)-单元测试

个人积累,请勿私自转载,转载前请联系
代码及文章资源https://github.com/jedyang/DayDayUp/tree/master/java/springboot
基于SpringBootCookBook的读书笔记,重在个人理解和实践,而非翻译。

测试

关于单元测试的全面解析请看我的这篇文章,这里只讲springboot相关的。
单元测试

测试这块springboot不同的版本变化较大。这里用的是1.5.*

在我们创建的工程里,已经存在需要的目录结构BookPubApplicationTests。

测试rest请求

package org.test.bookpub;

import org.junit.Test;
import org.junit.runner.RunWith;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.boot.test.context.SpringBootTest;
import org.springframework.boot.test.web.client.TestRestTemplate;
import org.springframework.test.context.junit4.SpringRunner;
import org.test.bookpub.entity.Book;

import static junit.framework.TestCase.assertNotNull;
import static org.junit.Assert.assertEquals;

@RunWith(SpringRunner.class)// SpringJUnit支持,由此引入Spring-Test框架支持
@SpringBootTest(webEnvironment = SpringBootTest.WebEnvironment.DEFINED_PORT)
public class BookPubApplicationTests {
    @Autowired
    private TestRestTemplate restTemplate;

    @Test
    public void webappBookIsbnApi() {
//        Book book = restTemplate.getForObject("/books/1001", Book.class);
        Book book = restTemplate.getForObject("http://localhost:8080/books/1001", Book.class);
        System.out.println(book);
        assertNotNull(book);
        assertEquals("yunsheng", book.getPublisher().getName());
    }

}
  1. webEnvironment = SpringBootTest.WebEnvironment.DEFINED_PORT是使用配置的端口。
    还有一个是webEnvironment = SpringBootTest.WebEnvironment.RANDOM_PORT随机端口。
    因为后面restTemplate.getForObject("http://localhost:8080/books/1001", Book.class);写死了url,所以这边设置哪个都可以。
  2. Book book = restTemplate.getForObject("/books/1001", Book.class);这种使用代码默认的协议和域名。因为我们前面默认配置的是https。这里直接用会有问题。

测试数据库连接

增加一个测试方法

@Autowired
private BookRepository bookRepository;

// 测试数据库可正常连接
@Test
public void contextLoads() {
    assertEquals(1, bookRepository.count());
}

服务端测试

TestRestTemplate是模拟从客户端发起请求来测试。还有一种方法是使用MockMvc对象在服务端直接测试controller层的方法。就想启动了服务器一样。

@Autowired
private WebApplicationContext context;

private MockMvc mockMvc;
@Before
public void setupMockMvc() {
    mockMvc = MockMvcBuilders.webAppContextSetup(context).build();
}

@Test
public void webappBookApi() throws Exception {
    //MockHttpServletRequestBuilder.accept方法是设置客户端可识别的内容类型
    //MockHttpServletRequestBuilder.contentType,设置请求头中的Content-Type字段,表示请求体的内容类型
    mockMvc.perform(get("http://localhost:8080/books/1001")
            .accept(MediaType.APPLICATION_JSON_UTF8))
            .andExpect(status().isOk())
            .andExpect(content().string(containsString("yunsheng")))
            .andExpect(jsonPath("$.isbn").value("1001"));
}

数据库插入数据

之前我们把往数据库插入一条记录的代码放在了StartupRunner里,其实这不是很合适。现在用单元测试来代替。
有几种方式来做,

  1. 使用hibernate这类映射工具,它会扫描@Entity注解的类,自动创建对应的表。然后使用import.sql创建数据。
  2. 使用spring jdbc,通过schema.sql文件定义表结构,使用data.sql创建数据。

做法

使用hibernate方式

  1. 注释掉之前在StartupRunner中的代码

      @Override
     public void run(String... strings) throws Exception {
         logger.info("dataSource:" + ds.toString());
         // 增加一条记录
     //        Author author = new Author("yunsheng", "yang");
     //        author = authorRepository.save(author);
     //        Publisher publisher = new Publisher("yunsheng");
     //        publisher = publisherRepository.save(publisher);
     //        Book book = new Book("1001", "test1", author, publisher);
     //        Book book1 = bookRepository.save(book);
         logger.info("number of books:" + bookRepository.count());
     }
    
  2. 在application.properties中配置

     spring.jpa.hibernate.ddl-auto=create-drop
    
  3. 在resources下创建import.sql文件

     INSERT INTO author (id, first_name, last_name) VALUES (1, 'yunsheng', 'yang');
     INSERT INTO publisher (id, name) VALUES (1, 'yunsheng');
     INSERT INTO book (isbn, title, author_id, publisher_id) VALUES ('1001', 'Spring Boot Recipes', 1,1);
    
  4. 运行测试用例,通过。

使用jdbc方式

  1. 在application.properties中配置

    spring.jpa.hibernate.ddl-auto=none
    即使Hibernate依赖存在也不要做任何事情

  2. 创建schema.sql用来创建表

     -- Create syntax for TABLE 'author'
     DROP TABLE IF EXISTS `author`;
     CREATE TABLE `author` (
       `id`         BIGINT(20) NOT NULL AUTO_INCREMENT,
       `first_name` VARCHAR(255)        DEFAULT NULL,
       `last_name`  VARCHAR(255)        DEFAULT NULL,
       PRIMARY KEY (`id`)
     );
     -- CREATE syntax FOR TABLE 'publisher'
     DROP TABLE IF EXISTS `publisher`;
     CREATE TABLE `publisher` (
       `id`   BIGINT(20) NOT NULL AUTO_INCREMENT,
       `name` VARCHAR(255)        DEFAULT NULL,
       PRIMARY KEY (`id`)
     );
     -- CREATE syntax FOR TABLE 'reviewer'
     DROP TABLE IF EXISTS `reviewer`;
     CREATE TABLE `reviewer` (
       `id`         BIGINT(20) NOT NULL AUTO_INCREMENT,
       `first_name` VARCHAR(255)        DEFAULT NULL,
       `last_name`  VARCHAR(255)        DEFAULT NULL,
       PRIMARY KEY (`id`)
     );
     -- CREATE syntax FOR TABLE 'book'
     DROP TABLE IF EXISTS `book`;
     CREATE TABLE `book` (
       `id`           BIGINT(20) NOT NULL AUTO_INCREMENT,
       `description`  VARCHAR(255)        DEFAULT NULL,
       `isbn`         VARCHAR(255)        DEFAULT NULL,
       `title`        VARCHAR(255)        DEFAULT NULL,
       `author_id`    BIGINT(20)          DEFAULT NULL,
       `publisher_id` BIGINT(20)          DEFAULT NULL,
       PRIMARY KEY (`id`),
       CONSTRAINT `FK_publisher` FOREIGN KEY (`publisher_id`) REFERENCES
         `publisher` (`id`),
       CONSTRAINT `FK_author` FOREIGN KEY (`author_id`) REFERENCES `author`
       (`id`)
     );
     -- CREATE syntax FOR TABLE 'book_reviewers'
     DROP TABLE IF EXISTS `book_reviewers`;
     CREATE TABLE `book_reviewers` (
       `book_id`      BIGINT(20) NOT NULL,
       `reviewers_id` BIGINT(20) NOT NULL,
       CONSTRAINT `FK_book` FOREIGN KEY (`book_id`) REFERENCES `book`
       (`id`),
       CONSTRAINT `FK_reviewer` FOREIGN KEY (`reviewers_id`) REFERENCES
         `reviewer` (`id`)
     );
    
  3. 创建data.sql用来插入数据
    和import.sql一样复制一份即可

  4. 运行测试用例,通过

默认情况下,Hibernate会使用import.sql。jdbc会使用schema.sql和data.sql。
如果你希望使用别的名字代替schema.sql或者data.sql,Spring Boot也提供了对应的配置属性,即spring.datasource.schema和spring.datasource.data。

tips,此处我之前有个错误,把reviewer写成了publisher的内部类,这里创建表会失败,已修改。

mock数据库操作

在实际工作中,我们写单测更常见的做法是mock掉数据库操作。这里演示使用mockito。

package org.test.bookpub;

import org.junit.After;
import org.junit.Before;
import org.junit.Test;
import org.junit.runner.RunWith;
import org.mockito.Mockito;
import org.springframework.boot.test.context.SpringBootTest;
import org.springframework.boot.test.mock.mockito.MockBean;
import org.springframework.test.context.junit4.SpringRunner;
import org.test.bookpub.repository.PublisherRepository;

import static org.junit.Assert.assertEquals;

@RunWith(SpringRunner.class)
@SpringBootTest
public class PublisherRepositoryTests {
    @MockBean
    private PublisherRepository repository;

    @Before
    public void setupPublisherRepositoryMock() {
        Mockito.when(repository.count()).thenReturn(100L);
    }

    @Test
    public void publishersExist() {
        assertEquals(100, repository.count());
    }

    @After
    public void resetPublisherRepositoryMock() {
        Mockito.reset(repository);
    }
}

新建一个类演示,mockito更多用法,可以google之。

这里实在没搞懂一个注解的事,cookbook搞了一堆代码在干嘛
其实写到这里我已经后悔了。cookbook太老了。
https://docs.spring.io/spring-boot/docs/current/reference/htmlsingle/
还是直接看官方文档吧,有你想要的全部。

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 204,793评论 6 478
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 87,567评论 2 381
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 151,342评论 0 338
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 54,825评论 1 277
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 63,814评论 5 368
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 48,680评论 1 281
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 38,033评论 3 399
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 36,687评论 0 258
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 42,175评论 1 300
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 35,668评论 2 321
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 37,775评论 1 332
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 33,419评论 4 321
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 39,020评论 3 307
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 29,978评论 0 19
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 31,206评论 1 260
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 45,092评论 2 351
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 42,510评论 2 343

推荐阅读更多精彩内容