在本系列的第一篇文章中，我们逐步介绍了如何使用 Spring Boot、Cloudinary、Docker 和 PostgreSQL 构建一个强大的图像上传器。我们涵盖了从设置项目到向保存图像和信息的端点发出请求的所有内容。如果您还没有阅读那篇文章，我强烈建议您从那里开始，为我们将要使用的应用程序打下坚实的基础。
现在，是时候确保我们的应用程序可靠并随着时间的推移保持其完整性。这给我们带来了软件开发的一个关键方面：测试。在本文中，我们将重点为我们的图像上传器 API 编写单元测试。我们将探索如何模拟依赖关系，并编写涵盖服务不同部分的测试。
单元测试不仅有助于及早发现错误，还能确保我们的代码可维护和可扩展。在本文结束时，您将拥有一套针对图像上传器 API 的全面测试，让您确信您的应用程序按预期工作。
让我们深入单元测试的世界，让我们的图像上传器 API 防弹！
我正在使用 VSCode 和 Java 扩展包。现在我们准备好编写测试了。
如果您使用其他 IDE，请参阅 JUnit5 文档中对所有这些 IDE 的支持。
右键单击 BookService 类，单击“Go to Test”，然后从菜单中选择要为其生成测试的方法。
将生成一个类似的文件，如下所示：
import org.junit.jupiter.api.Test;
public class BookServiceTest {
@Test
void testAddBook() {
}
}
请记住，在本文中，我们将使用 AAA 模式 进行测试（安排 - 执行 - 断言）。
@ExtendWith(MockitoExtension.class)
public class BookServiceTest {
@Mock
private BookRepository bookRepository;
@Mock
private Cloudinary cloudinary;
@Mock
private MultipartFile multipartFile;
@Mock
private Uploader uploader;
@Captor
private ArgumentCaptor<Book> bookArgumentCaptor;
@InjectMocks
private BookService bookService;
}
@ExtendWith(MockitoExtension.class)
public class BookServiceTest {
@Mock
private BookRepository bookRepository;
@Mock
private Cloudinary cloudinary;
@Mock
private MultipartFile multipartFile;
@Mock
private Uploader uploader;
@Captor
private ArgumentCaptor<Book> bookArgumentCaptor;
@InjectMocks
private BookService bookService;
@Nested
class AddBook {
@Test
void shouldCreateANewBook() throws Exception {
// Arrange
Map<String, Object> uploadResult = Map.of("url", "http://example.com/image.jpg");
when(cloudinary.uploader()).thenReturn(uploader);
when(uploader.upload(any(File.class), anyMap())).thenReturn(uploadResult);
Book book = new Book();
book.setName("Test Book");
book.setImgUrl(uploadResult.get("url").toString());
when(bookRepository.save(any(Book.class))).thenReturn(book);
when(multipartFile.getOriginalFilename()).thenReturn("test.jpg");
when(multipartFile.getBytes()).thenReturn("test content".getBytes());
// Act
Book result = bookService.addBook("Test Book", multipartFile);
// Assert
assertNotNull(result);
assertEquals("Test Book", result.getName());
assertEquals("http://example.com/image.jpg", result.getImgUrl());
}
@Test
void shouldCallRepositorySave() throws Exception {
// Arrange
Map<String, Object> uploadResult = Map.of("url", "http://example.com/image.jpg");
when(cloudinary.uploader()).thenReturn(uploader);
when(uploader.upload(any(File.class), anyMap())).thenReturn(uploadResult);
Book book = new Book();
book.setName("Test Book");
book.setImgUrl(uploadResult.get("url").toString());
when(bookRepository.save(any(Book.class))).thenReturn(book);
when(multipartFile.getOriginalFilename()).thenReturn("test.jpg");
when(multipartFile.getBytes()).thenReturn("test content".getBytes());
// Act
bookService.addBook("Test Book", multipartFile);
// Assert
verify(bookRepository, times(1)).save(bookArgumentCaptor.capture());
Book capturedBook = bookArgumentCaptor.getValue();
assertEquals("Test Book", capturedBook.getName());
assertEquals("http://example.com/image.jpg", capturedBook.getImgUrl());
}
@Test
void shouldFailTheUpload() throws Exception {
// Arrange
when(multipartFile.getOriginalFilename()).thenReturn("test.jpg");
when(multipartFile.getBytes()).thenReturn("test content".getBytes());
when(cloudinary.uploader()).thenReturn(uploader);
when(uploader.upload(any(File.class),
anyMap())).thenThrow(IOException.class);
// Act & Assert
ResponseStatusException exception = assertThrows(ResponseStatusException.class, () -> {
bookService.addBook("Test Book", multipartFile);
});
assertEquals(HttpStatus.BAD_GATEWAY, exception.getStatusCode());
assertEquals("Failed to upload the file.", exception.getReason());
}
}
}
package cloudinary.upload.imageUpload.controllers;
import static org.mockito.ArgumentMatchers.any;
import static org.mockito.Mockito.when;
import static org.springframework.test.web.servlet.request.MockMvcRequestBuilders.multipart;
import static org.springframework.test.web.servlet.result.MockMvcResultMatchers.jsonPath;
import static org.springframework.test.web.servlet.result.MockMvcResultMatchers.status;
import org.junit.jupiter.api.Nested;
import org.junit.jupiter.api.Test;
import org.junit.jupiter.api.extension.ExtendWith;
import org.mockito.InjectMocks;
import org.mockito.Mock;
import org.mockito.junit.jupiter.MockitoExtension;
import org.springframework.http.HttpStatus;
import org.springframework.http.MediaType;
import org.springframework.mock.web.MockMultipartFile;
import org.springframework.test.web.servlet.MockMvc;
import org.springframework.test.web.servlet.setup.MockMvcBuilders;
import org.springframework.web.server.ResponseStatusException;
import cloudinary.upload.imageUpload.configs.GlobalExceptionHandler;
import cloudinary.upload.imageUpload.entities.Book;
import cloudinary.upload.imageUpload.services.BookService;
@ExtendWith(MockitoExtension.class)
public class BookControllerTest {
@Mock
private BookService bookService;
@InjectMocks
private BookController bookController;
private MockMvc mockMvc;
@Nested
class AddBook {
@Test
void shouldReturnSuccess() throws Exception {
// Arrange
MockMultipartFile image = new MockMultipartFile("imgUrl", "test.jpg", MediaType.IMAGE_JPEG_VALUE,
"test content".getBytes());
Book book = new Book();
book.setName("Test Book");
book.setImgUrl("http://example.com/image.jpg");
when(bookService.addBook(any(String.class), any(MockMultipartFile.class))).thenReturn(book);
mockMvc = MockMvcBuilders.standaloneSetup(bookController).build();
// Act & Assert
mockMvc.perform(multipart("/addBook")
.file(image)
.param("name", "Test Book"))
.andExpect(status().isOk())
.andExpect(jsonPath("$.name").value("Test Book"))
.andExpect(jsonPath("$.imgUrl").value("http://example.com/image.jpg"));
}
@Test
void shouldFailToUploadImage() throws Exception {
// Arrange
MockMultipartFile image = new MockMultipartFile("imgUrl", "test.jpg", MediaType.IMAGE_JPEG_VALUE,
"test content".getBytes());
when(bookService.addBook(any(String.class), any(MockMultipartFile.class)))
.thenThrow(new ResponseStatusException(HttpStatus.INTERNAL_SERVER_ERROR,
"Failed to upload the file."));
mockMvc = MockMvcBuilders.standaloneSetup(bookController).setControllerAdvice(new GlobalExceptionHandler())
.build();
// Act & Assert
mockMvc.perform(multipart("/addBook")
.file(image)
.param("name", "Test Book"))
.andExpect(status().isInternalServerError())
.andExpect(result -> result.getResponse().equals("Failed to upload the file."));
}
@Test
void shouldFailWithMissingNameParameter() throws Exception {
// Arrange
MockMultipartFile image = new MockMultipartFile("imgUrl", "test.jpg", MediaType.IMAGE_JPEG_VALUE,
"test content".getBytes());
mockMvc = MockMvcBuilders.standaloneSetup(bookController).build();
// Act & Assert
mockMvc.perform(multipart("/addBook")
.file(image))
.andExpect(status().isBadRequest());
}
@Test
void shouldFailWithMissingImageParameter() throws Exception {
// Arrange
mockMvc = MockMvcBuilders.standaloneSetup(bookController).build();
// Act & Assert
mockMvc.perform(multipart("/addBook")
.param("name", "Test Book"))
.andExpect(status().isBadRequest());
}
}
}
这些是一些简单的测试用例，供您开始测试您的应用程序。请记住，我们可以通过添加一些工厂来重构这些测试以避免重复。
感谢您的阅读。
JUnit5 - 文档
Mockito - 文档
以上是使用 JUnitnd Mockito 对图像上传器 API 进行单元测试的详细内容。更多信息请关注PHP中文网其他相关文章！
