@ -16,7 +16,6 @@
@@ -16,7 +16,6 @@
package org.springframework.web.method.annotation ;
import java.util.HashSet ;
import org.junit.jupiter.api.Test ;
@ -35,14 +34,15 @@ import static org.assertj.core.api.Assertions.assertThat;
@@ -35,14 +34,15 @@ import static org.assertj.core.api.Assertions.assertThat;
/ * *
* Test fixture with { @link SessionAttributesHandler } .
*
* @author Rossen Stoyanchev
* /
class SessionAttributesHandlerTests {
private final SessionAttributeStore sessionAttributeStore = new DefaultSessionAttributeStore ( ) ;
private final SessionAttributesHandler sessionAttributesHandler = new SessionAttributesHandler (
SessionAttributeHandler . class , sessionAttributeStore ) ;
private final SessionAttributesHandler sessionAttributesHandler =
new SessionAttributes Handler ( TestSessionAttributesHold er . class , sessionAttributeStore ) ;
private final NativeWebRequest request = new ServletWebRequest ( new MockHttpServletRequest ( ) ) ;
@ -62,12 +62,16 @@ class SessionAttributesHandlerTests {
@@ -62,12 +62,16 @@ class SessionAttributesHandlerTests {
sessionAttributeStore . storeAttribute ( request , "attr3" , new TestBean ( ) ) ;
sessionAttributeStore . storeAttribute ( request , "attr4" , new TestBean ( ) ) ;
assertThat ( sessionAttributesHandler . retrieveAttributes ( request ) . keySet ( ) ) . as ( "Named attributes (attr1, attr2) should be 'known' right away" ) . isEqualTo ( new HashSet < > ( asList ( "attr1" , "attr2" ) ) ) ;
assertThat ( sessionAttributesHandler . retrieveAttributes ( request ) . keySet ( ) )
. as ( "Named attributes (attr1, attr2) should be 'known' right away" )
. isEqualTo ( new HashSet < > ( asList ( "attr1" , "attr2" ) ) ) ;
// Resolve 'attr3' by type
sessionAttributesHandler . isHandlerSessionAttribute ( "attr3" , TestBean . class ) ;
assertThat ( sessionAttributesHandler . retrieveAttributes ( request ) . keySet ( ) ) . as ( "Named attributes (attr1, attr2) and resolved attribute (att3) should be 'known'" ) . isEqualTo ( new HashSet < > ( asList ( "attr1" , "attr2" , "attr3" ) ) ) ;
assertThat ( sessionAttributesHandler . retrieveAttributes ( request ) . keySet ( ) )
. as ( "Named attributes (attr1, attr2) and resolved attribute (attr3) should be 'known'" )
. isEqualTo ( new HashSet < > ( asList ( "attr1" , "attr2" , "attr3" ) ) ) ;
}
@Test
@ -105,8 +109,8 @@ class SessionAttributesHandlerTests {
@@ -105,8 +109,8 @@ class SessionAttributesHandlerTests {
}
@SessionAttributes ( names = { "attr1" , "attr2" } , types = { TestBean . class } )
private static class SessionAttributeHandl er {
@SessionAttributes ( names = { "attr1" , "attr2" } , types = TestBean . class )
private static class TestSessionAttributesHold er {
}
}