如何在 RestController 类中模拟私有变量

Question

我正在使用 Spring 进行一些 Web 单元测试。有没有办法可以模拟在@PostConstruct 中设置的MyProcessor 实例？我尝试使用@MockBean，但它被设置为空并出现空指针异常？

我正在使用工厂根据一些布尔标志来实例化MyProcessor。但是，如果有一种完全不同的方法可以使我的测试更清晰，我愿意接受想法。

请注意我使用的是 Junit 5。

public class Controller {

    private final AProcessorFactory factory;

    @Value("${tt.f.processor.async}")
    private boolean isAsync;

    private MyProcessor myProcessor;

    public Controller(final AProcessorFactory factory) {
        this.factory= factory;
    }

    @PostConstruct
    public void init() {
        myProcessor = factory.getInstance(isAsync);
    }

}


@WebMvcTest(Controller.class)
public class ControllerTest{

    @MockBean
    private MyProcessor processor;


    @MockBean
    private AProcessorFactory factory;

    @Autowired
    private MockMvc mockMvc;


    @Test
    public void test() throws Exception {           

        when(processor.process(any(Request.class), any(String.class)))
                .thenReturn(new TestResponse("123", SUCCESS, "", ""));

}

Answer 1

看起来您的 myProcessor 实际上是由您的 AProcessorFactory 在 PostConstruct init() 方法中构建的。

您需要为 AProcessorFactory 模拟提供一种行为。

首先，您可能希望在构造函数中设置 myProcessor，因为 @PostConstruct init() 方法没有特殊的上下文加载逻辑。

public class Controller {

    @Value("${tt.f.processor.async}")
    private boolean isAsync;

    private MyProcessor myProcessor;

    public Controller(final AProcessorFactory factory) {
        this.myProcessor = factory.getInstance(isAsync);
    }
}

您可以在测试的@Before 步骤中指定此项。

 @MockBean
 private AProcessorFactory factory;

 @Before
 public void setUp() {
     when(processor.process(any(Request.class), any(String.class)))
            .thenReturn(new TestResponse("123", SUCCESS, "", ""));
     when(factory.getInstance(any())).thenReturn(processor)
 }

Spring 将注册您的已加载适当行为的 AProcessorFactory 模拟。