Spring + Mockito –无法模拟保存方法？

最新推荐文章于 2024-06-13 07:34:43 发布

cyan20115

最新推荐文章于 2024-06-13 07:34:43 发布

阅读量466

点赞数

文章标签： json java

在Spring Boot 2 + Spring Data JPA环境中遇到使用Mockito模拟存储库方法返回null的问题。解决方案包括正确使用Mockito的参数匹配，以及考虑为模型实现equals和hashCode方法。

摘要由CSDN通过智能技术生成

尝试模拟存储库save()方法，但是它总是返回null？

PS已通过Spring Boot 2 + Spring Data JPA测试

@Test
    public void save_book_OK() throws Exception {

        Book newBook = new Book(1L, "Mockito Guide", "mkyong");
        when(mockRepository.save(newBook)).thenReturn(newBook);

		mockMvc.perform(post("/books")
			.content("{json}")
			.header(HttpHeaders.CONTENT_TYPE, MediaType.APPLICATION_JSON))
			.andExpect(status().isCreated());
				
    }

解

1. Mockito将equals用于参数匹配，请尝试将ArgumentMatchers.any用于save方法。

import static org.mockito.ArgumentMatchers.any;
import static org.mockito.Mockito.*;

	@Test
    public void save_book_OK() throws Exception {

        Book newBook = new Book(1L, "Mockito Guide", "mkyong");
        when(mockRepository.save(any(Book.class))).thenReturn(newBook);

		//...
				
    }

2.或者，为模型实现equals和hashCode 。

package com.mkyong;

import javax.persistence.Entity;
import javax.persistence.GeneratedValue;
import javax.persistence.Id;
import java.math.BigDecimal;

@Entity
public class Book {

    @Id
    @GeneratedValue
    private Long id;
    private String name;
    private String author;
   
    //...

    @Override
    public boolean equals(Object o) {
        if (this == o) return true;
        if (o == null || getClass() != o.getClass()) return false;

        Book book = (Book) o;

        if (id != null ? !id.equals(book.id) : book.id != null) return false;
        if (name != null ? !name.equals(book.name) : book.name != null) return false;
        return author != null ? author.equals(book.author) : book.author == null;
    }

    @Override
    public int hashCode() {
        int result = id != null ? id.hashCode() : 0;
        result = 31 * result + (name != null ? name.hashCode() : 0);
        result = 31 * result + (author != null ? author.hashCode() : 0);
        return result;
    }
}