Question

我正在为控制器编写单元测试。我正在嘲笑服务层并使用独立设置来休息控制器。

ProductSupplierControllerUnitTest.java

public class ProductSupplierControllerUnitTest {

    @Mock
    private ProductSupplierService productSupplierService;

    @InjectMocks
    private ProductSupplierRestController productSupplierRestController;

    private MockMvc mockMvc;

    @Before
    public void setUp() {
        MockitoAnnotations.initMocks(this);
        mockMvc = MockMvcBuilders.standaloneSetup(productSupplierRestController)
                .setControllerAdvice(new ServiceExceptionHandler()).build();
    }

    @Test
    public void productNotFound() throws Exception {

        Long incorrectProductId = 2L;

        Mockito.when(productSupplierService.getProductSuppliers(incorrectProductId, tenantId))
                .thenThrow(new EntityNotFoundException(Product.class, String.valueOf(incorrectProductId)));

        RequestBuilder requestBuilder = MockMvcRequestBuilders.get(prepareRequestUrl(incorrectProductId))
                .requestAttr(TENANT_ID, tenantId).contentType(MediaType.APPLICATION_JSON_UTF8);

        mockMvc.perform(requestBuilder).andExpect(status().isNotFound())
                .andExpect(content().contentType(MediaType.APPLICATION_JSON_UTF8)).andExpect(jsonPath("$.message",
                        is(String.format("Product was not found for parameter(s) %s", incorrectProductId))));

        Mockito.verify(productSupplierService, times(1)).getProductSuppliers(incorrectProductId, tenantId);
        Mockito.verifyNoMoreInteractions(productSupplierService);
    }

    @Test
    public void getProductSuppliersSuccess() throws Exception {

        Map<String, List<? extends BaseDTO>> result = new HashMap<>(0);
        ProductSupplierDTO productSupplierDTO = new ProductSupplierDTO();
        productSupplierDTO.setSupplierId(correctSupplierId);
        productSupplierDTO.setBuyPrice(validBuyPrice);
        productSupplierDTO.setDefaultSupplier(isDefaultSupplier);

        result.put("product_suppliers", Collections.singletonList(productSupplierDTO));

        Mockito.when(productSupplierService.getProductSuppliers(productId, tenantId)).thenReturn(result);

        RequestBuilder requestBuilder = MockMvcRequestBuilders.get(prepareRequestUrl(productId))
                .requestAttr(TENANT_ID, tenantId).contentType(MediaType.APPLICATION_JSON_UTF8);

        mockMvc.perform(requestBuilder).andExpect(status().isOk())
                .andExpect(content().contentType(MediaType.APPLICATION_JSON_UTF8))
                .andExpect(jsonPath("$.product_suppliers", hasSize(1)))
                .andExpect(jsonPath("$.product_suppliers[0].supplier_id", is(correctSupplierId.intValue())))
                .andExpect(jsonPath("$.product_suppliers[0].buy_price", is(validBuyPrice)))
                .andExpect(jsonPath("$.product_suppliers[0].default_supplier", is(isDefaultSupplier)));

        Mockito.verify(productSupplierService, times(1)).getProductSuppliers(productId, tenantId);
        Mockito.verifyNoMoreInteractions(productSupplierService);
    }

    // some more tests
}

测试工作正常，但问题是这个独立设置是在每个测试用例之前完成的，因为每个测试用例控制器的方法都被映射到url并初始化了Spring FrameworkServlet。

是否可以对所有测试用例进行一次初始化？因为我想减少测试时间。

编辑：我已将代码更改为以下内容，并且按预期工作。问题是它是正确的测试方法吗？

@RunWith(MockitoJUnitRunner.class)
public class ProductSupplierControllerUnitTest {

    @Mock
    private ProductSupplierService productSupplierService;

    @InjectMocks
    private static ProductSupplierRestController productSupplierRestController = new ProductSupplierRestController();

    private static MockMvc mockMvc =  MockMvcBuilders.standaloneSetup(productSupplierRestController)
            .setControllerAdvice(new ServiceExceptionHandler()).build();

    // tests
}

将控制器和mockMvc设为静态是不错的做法？

Answer 1

如果使用JUnit，则使用注释@BeforeClass而不是@Before。具有此批注的方法将在类中的所有测试之前调用一次。

如何在Spring Boot中的所有测试用例之前只设置一次独立控制器？

1 个答案: