Check out the full series of Questions Management tutorial here.
In the previous tutorial, we prepared all the necessary configurations to build the API retrieving all the categories in the MongoDB database: a document Category object containing information about a category, a CategoryRepository to manipulate with MongoDB, a CategoryController that defines the APIs of the Core Category Service will start with “/category” and the connection information to the MongoDB server is configured in the application.properties file. Now, we are going to build this API.
To build API retrieving all the categories, I’m going to add a method to expose a GET request “/all” in CategoryController:
|
1 2 3 4 |
@GetMapping("/all") public Flux<Category> findAllCategories() { } |
Since Spring Data MongoDB Reactive has provided us with a method to search all categories then our job just calls to use:
|
1 2 3 4 |
@GetMapping("/all") public Flux<Category> findAllCategories() { return categoryRepository.findAll(); } |
Very simple, right?
Suppose currently in the database I have the following categories:
then when I run the request to get all the available categories, the result will be:
Now we will add a new Unit Test for the code that we just added.
In the previous tutorial, I created a test class for CategoryController called CategoryControllerTest, a Mock object for CategoryRepository, and injected this Mock object into the CategoryController class:
|
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 |
package com.huongdanjava.categoryservice; import org.mockito.InjectMocks; import org.mockito.Mock; import com.huongdanjava.categoryservice.repository.CategoryRepository; public class CategoryControllerTest { @Mock private CategoryRepository categoryRepository; @InjectMocks private CategoryController categoryController; } |
Initializing the mock every time I run a test case:
|
1 2 3 4 |
@Before public void init() { MockitoAnnotations.initMocks(this); } |
Now I’m going to add a method to test the findAllCategories() method of the CategoryController class, like this:
|
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 |
@Test public void testFindAllCategories() { // Create category document to return when using findAll Category category = new Category(); category.setId("123"); category.setCode("ABC"); category.setName("ZXC"); // Mock findAll() method of CategoryRepository to return a Flux when(categoryRepository.findAll()).thenReturn(Flux.just(category)); // Call method Flux<Category> categories = categoryController.findAllCategories(); Category result = categories.blockFirst(); // Assertions assertEquals("123", result.getId()); assertEquals("ABC", result.getCode()); assertEquals("ZXC", result.getName()); } |
Run “Maven test” in STS or “mvn test” with Apache Maven, you will not see any errors.
