如何在 Spring WebTestClient 中斷言 FlashAttributes？

Question

我想測試 spring @Controller的FlashMap分配：

@Controller
public class MyServlet {
    @GetMapping("/test")
    public String get() {
            ServletRequestAttributes reqAttr = (ServletRequestAttributes) RequestContextHolder.getRequestAttributes();
            FlashMap outAttr = RequestContextUtils.getOutputFlashMap(reqAttr.getRequest());
            outAttr.put("success", "test");
        }
    }
}

我如何才能真正從WebTestClient class 訪問 flash 屬性？

@SpringBootTest
public class ServletTest {
    @Autowired
    private WebTestClient;

    @Test
    public void test() {
        client.get().uri("/test")
            .expectStatus().isOk()
            .expectBody().isEmpty(); 
            //TODO how to validate flash attributes?
    }
}

Answer 1

EntityExchangeResult<byte[]> result = client...
                                            .expectBody().returnResult();

MockMvcWebTestClient.resultActionsFor(result)
                    .andExpect(flash()
                    .attribute("success", "test");