13
13
14
14
use PHPUnit \Framework \TestCase ;
15
15
use Psr \Container \ContainerInterface ;
16
+ use Symfony \Component \DependencyInjection \ContainerBuilder ;
16
17
use Symfony \Component \HttpFoundation \Request ;
17
18
use Symfony \Component \HttpFoundation \RequestStack ;
18
- use Symfony \Component \HttpFoundation \Session \SessionInterface ;
19
+ use Symfony \Component \HttpFoundation \Session \Session ;
19
20
use Symfony \Component \Security \Core \Authentication \Token \NullToken ;
20
21
use Symfony \Component \Security \Core \Authentication \Token \Storage \TokenStorage ;
21
22
use Symfony \Component \Security \Core \Authentication \Token \Storage \UsageTrackingTokenStorage ;
@@ -25,28 +26,14 @@ class UsageTrackingTokenStorageTest extends TestCase
25
26
{
26
27
public function testGetSetToken ()
27
28
{
28
- $ sessionAccess = 0 ;
29
- $ sessionLocator = new class (['request_stack ' => function () use (&$ sessionAccess ) {
30
- $ session = $ this ->createMock (SessionInterface::class);
31
-
32
- $ request = new Request ();
33
- $ request ->setSession ($ session );
34
- $ requestStack = $ this ->getMockBuilder (RequestStack::class)->onlyMethods (['getSession ' ])->getMock ();
35
- $ requestStack ->push ($ request );
36
- $ requestStack ->expects ($ this ->any ())->method ('getSession ' )->willReturnCallback (function () use ($ session , &$ sessionAccess ) {
37
- ++$ sessionAccess ;
38
-
39
- $ session ->expects ($ this ->once ())
40
- ->method ('getMetadataBag ' );
41
-
42
- return $ session ;
43
- });
44
-
45
- return $ requestStack ;
46
- }]) implements ContainerInterface {
47
- use ServiceLocatorTrait;
48
- };
49
29
$ tokenStorage = new TokenStorage ();
30
+ $ session = new Session ();
31
+ $ request = new Request ();
32
+ $ request ->setSession ($ session );
33
+ $ requestStack = new RequestStack ();
34
+ $ requestStack ->push ($ request );
35
+ $ sessionLocator = new ContainerBuilder ();
36
+ $ sessionLocator ->set ('request_stack ' , $ requestStack );
50
37
$ trackingStorage = new UsageTrackingTokenStorage ($ tokenStorage , $ sessionLocator );
51
38
52
39
$ this ->assertNull ($ trackingStorage ->getToken ());
@@ -55,15 +42,15 @@ public function testGetSetToken()
55
42
$ trackingStorage ->setToken ($ token );
56
43
$ this ->assertSame ($ token , $ trackingStorage ->getToken ());
57
44
$ this ->assertSame ($ token , $ tokenStorage ->getToken ());
58
- $ this ->assertSame (0 , $ sessionAccess );
45
+ $ this ->assertSame (0 , $ session -> getUsageIndex () );
59
46
60
47
$ trackingStorage ->enableUsageTracking ();
61
48
$ this ->assertSame ($ token , $ trackingStorage ->getToken ());
62
- $ this ->assertSame (1 , $ sessionAccess );
49
+ $ this ->assertSame (1 , $ session -> getUsageIndex () );
63
50
64
51
$ trackingStorage ->disableUsageTracking ();
65
52
$ this ->assertSame ($ token , $ trackingStorage ->getToken ());
66
- $ this ->assertSame (1 , $ sessionAccess );
53
+ $ this ->assertSame (1 , $ session -> getUsageIndex () );
67
54
}
68
55
69
56
public function testWithoutMainRequest ()
0 commit comments