@@ -345,15 +345,22 @@ public Object m1() {
345345 private MockUnit .Block session = unit -> {
346346 Binder binder = unit .get (Binder .class );
347347
348- AnnotatedBindingBuilder <SessionManager > smbinding = unit .mock (AnnotatedBindingBuilder .class );
349- smbinding . toInstance ( isA ( SessionManager . class ) );
348+ AnnotatedBindingBuilder <SessionManager > smABB = unit .mock (AnnotatedBindingBuilder .class );
349+ smABB . asEagerSingleton ( );
350350
351- expect (binder .bind (SessionManager .class )).andReturn (smbinding );
351+ ScopedBindingBuilder ssSBB = unit .mock (ScopedBindingBuilder .class );
352+ ssSBB .asEagerSingleton ();
352353
353- AnnotatedBindingBuilder <Session . Definition > binding = unit .mock (AnnotatedBindingBuilder .class );
354- binding . toInstance ( isA ( Session .Definition .class ));
354+ AnnotatedBindingBuilder <Store > ssABB = unit .mock (AnnotatedBindingBuilder .class );
355+ expect ( ssABB . to ( Session .Mem .class )). andReturn ( ssSBB );
355356
356- expect (binder .bind (Session .Definition .class )).andReturn (binding );
357+ expect (binder .bind (SessionManager .class )).andReturn (smABB );
358+ expect (binder .bind (Session .Store .class )).andReturn (ssABB );
359+
360+ AnnotatedBindingBuilder <Session .Definition > sdABB = unit .mock (AnnotatedBindingBuilder .class );
361+ sdABB .toInstance (isA (Session .Definition .class ));
362+
363+ expect (binder .bind (Session .Definition .class )).andReturn (sdABB );
357364 };
358365
359366 private MockUnit .Block boot = unit -> {
@@ -2105,17 +2112,22 @@ public void useStore() throws Exception {
21052112 .expect (unit -> {
21062113 Binder binder = unit .get (Binder .class );
21072114
2108- AnnotatedBindingBuilder <SessionManager > smbinding = unit
2109- .mock (AnnotatedBindingBuilder .class );
2110- smbinding .toInstance (isA (SessionManager .class ));
2115+ AnnotatedBindingBuilder <SessionManager > smABB = unit .mock (AnnotatedBindingBuilder .class );
2116+ smABB .asEagerSingleton ();
2117+
2118+ ScopedBindingBuilder ssSBB = unit .mock (ScopedBindingBuilder .class );
2119+ ssSBB .asEagerSingleton ();
2120+
2121+ AnnotatedBindingBuilder <Store > ssABB = unit .mock (AnnotatedBindingBuilder .class );
2122+ expect (ssABB .to (unit .get (Session .Store .class ).getClass ())).andReturn (ssSBB );
21112123
2112- expect (binder .bind (SessionManager .class )).andReturn (smbinding );
2124+ expect (binder .bind (SessionManager .class )).andReturn (smABB );
2125+ expect (binder .bind (Session .Store .class )).andReturn (ssABB );
21132126
2114- AnnotatedBindingBuilder <Session .Definition > binding = unit
2115- .mock (AnnotatedBindingBuilder .class );
2116- binding .toInstance (unit .capture (Session .Definition .class ));
2127+ AnnotatedBindingBuilder <Session .Definition > sdABB = unit .mock (AnnotatedBindingBuilder .class );
2128+ sdABB .toInstance (unit .capture (Session .Definition .class ));
21172129
2118- expect (binder .bind (Session .Definition .class )).andReturn (binding );
2130+ expect (binder .bind (Session .Definition .class )).andReturn (sdABB );
21192131 })
21202132 .expect (routes )
21212133 .expect (routeHandler )
@@ -2127,14 +2139,14 @@ public void useStore() throws Exception {
21272139 .run (unit -> {
21282140
21292141 Jooby jooby = new Jooby ();
2130- jooby .session (unit .get (Store .class ));
2142+ jooby .session (unit .get (Store .class ). getClass () );
21312143
21322144 jooby .start ();
21332145
21342146 }, boot ,
21352147 unit -> {
21362148 Definition def = unit .captured (Session .Definition .class ).iterator ().next ();
2137- assertEquals (unit .get (Store .class ), def .store ());
2149+ assertEquals (unit .get (Store .class ). getClass () , def .store ());
21382150 });
21392151 }
21402152
0 commit comments