[HttpKernel] Fix session test cases for symfony

alexander-schranz · nicolas-grekas · commit b047a2d7b16b · 2022-01-27T10:55:14.000+01:00
diff --git a/src/Symfony/Component/HttpKernel/Tests/EventListener/SessionListenerTest.php b/src/Symfony/Component/HttpKernel/Tests/EventListener/SessionListenerTest.php
@@ -22,7 +22,9 @@
 use Symfony\Component\HttpFoundation\Session\Session;
 use Symfony\Component\HttpFoundation\Session\SessionFactory;
 use Symfony\Component\HttpFoundation\Session\Storage\NativeSessionStorage;
-use Symfony\Component\HttpFoundation\Session\Storage\PhpBridgeSessionStorage;
+use Symfony\Component\HttpFoundation\Session\Storage\NativeSessionStorageFactory;
+use Symfony\Component\HttpFoundation\Session\Storage\PhpBridgeSessionStorageFactory;
+use Symfony\Component\HttpFoundation\Session\Storage\SessionStorageFactoryInterface;
 use Symfony\Component\HttpKernel\DataCollector\RequestDataCollector;
 use Symfony\Component\HttpKernel\Event\FinishRequestEvent;
 use Symfony\Component\HttpKernel\Event\RequestEvent;
@@ -134,20 +136,8 @@ public function testPhpBridgeAlreadyStartedSession()
         session_start();
         $sessionId = session_id();
 
-        $requestStack = new RequestStack();
         $request = new Request();
-        $requestStack->push($request);
-
-        $session = new Session();
-        $sessionStorage = new PhpBridgeSessionStorage();
-
-        $container = new Container();
-        $container->set('request_stack', $requestStack);
-        $container->set('session', $session);
-        $container->set('session_storage', $sessionStorage);
-
-        $request = new Request();
-        $listener = new SessionListener($container);
+        $listener = $this->createListener($request, new PhpBridgeSessionStorageFactory());
 
         $event = new RequestEvent($this->createMock(HttpKernelInterface::class), $request, HttpKernelInterface::MAIN_REQUEST);
 
@@ -162,17 +152,14 @@ public function testPhpBridgeAlreadyStartedSession()
      */
     public function testSessionCookieWrittenNoCookieGiven()
     {
-        $session = new Session();
-        $session->set('hello', 'world');
-
-        $container = new Container();
-        $container->set('initialized_session', $session);
+        $request = new Request();
+        $listener = $this->createListener($request, new NativeSessionStorageFactory());
 
-        $listener = new SessionListener($container);
         $kernel = $this->createMock(HttpKernelInterface::class);
 
-        $request = new Request();
         $listener->onKernelRequest(new RequestEvent($kernel, $request, HttpKernelInterface::MAIN_REQUEST));
+        $session = $request->getSession();
+        $session->set('hello', 'world');
 
         $response = new Response();
         $listener->onKernelResponse(new ResponseEvent($kernel, $request, HttpKernelInterface::MAIN_REQUEST, $response));
@@ -191,22 +178,25 @@ public function testSessionCookieWrittenNoCookieGiven()
      */
     public function testSessionCookieNotWrittenCookieGiven()
     {
-        $session = new Session();
-        $session->set('hello', 'world');
-        $sessionId = $session->getId();
-
-        $container = new Container();
-        $container->set('initialized_session', $session);
+        $sessionId = $this->createValidSessionId();
 
-        $listener = new SessionListener($container);
-        $kernel = $this->createMock(HttpKernelInterface::class);
+        $this->assertNotEmpty($sessionId);
 
         $request = new Request();
         $request->cookies->set('PHPSESSID', $sessionId);
+
+        $listener = $this->createListener($request, new NativeSessionStorageFactory());
+
+        $kernel = $this->createMock(HttpKernelInterface::class);
         $listener->onKernelRequest(new RequestEvent($kernel, $request, HttpKernelInterface::MAIN_REQUEST));
 
+        $session = $request->getSession();
+        $this->assertSame($sessionId, $session->getId());
+        $session->set('hello', 'world');
+
         $response = new Response();
         $listener->onKernelResponse(new ResponseEvent($kernel, $request, HttpKernelInterface::MAIN_REQUEST, $response));
+        $this->assertSame($sessionId, $session->getId());
 
         $cookies = $response->headers->getCookies();
         $this->assertCount(0, $cookies);
@@ -217,21 +207,18 @@ public function testSessionCookieNotWrittenCookieGiven()
      */
     public function testSessionCookieClearedWhenInvalidated()
     {
-        $session = new Session();
-
-        $container = new Container();
-        $container->set('initialized_session', $session);
-
-        $listener = new SessionListener($container);
+        $sessionId = $this->createValidSessionId();
+        $request = new Request();
+        $request->cookies->set('PHPSESSID', $sessionId);
+        $listener = $this->createListener($request, new NativeSessionStorageFactory());
         $kernel = $this->createMock(HttpKernelInterface::class);
 
-        $request = new Request();
         $listener->onKernelRequest(new RequestEvent($kernel, $request, HttpKernelInterface::MAIN_REQUEST));
 
+        $session = $request->getSession();
         $session->start();
         $sessionId = $session->getId();
         $this->assertNotEmpty($sessionId);
-        $request->cookies->set($session->getName(), $sessionId);
         $_SESSION['hello'] = 'world'; // check compatibility to php session bridge
 
         $session->invalidate();
@@ -252,21 +239,18 @@ public function testSessionCookieClearedWhenInvalidated()
      */
     public function testSessionCookieNotClearedWhenOtherVariablesSet()
     {
-        $session = new Session();
-
-        $container = new Container();
-        $container->set('initialized_session', $session);
-
-        $listener = new SessionListener($container);
+        $sessionId = $this->createValidSessionId();
+        $request = new Request();
+        $request->cookies->set('PHPSESSID', $sessionId);
+        $listener = $this->createListener($request, new NativeSessionStorageFactory());
         $kernel = $this->createMock(HttpKernelInterface::class);
 
-        $request = new Request();
         $listener->onKernelRequest(new RequestEvent($kernel, $request, HttpKernelInterface::MAIN_REQUEST));
 
+        $session = $request->getSession();
         $session->start();
         $sessionId = $session->getId();
         $this->assertNotEmpty($sessionId);
-        $request->cookies->set($session->getName(), $sessionId);
         $_SESSION['hello'] = 'world';
 
         $response = new Response();
@@ -281,17 +265,13 @@ public function testSessionCookieNotClearedWhenOtherVariablesSet()
      */
     public function testSessionCookieSetWhenOtherNativeVariablesSet()
     {
-        $session = new Session();
-
-        $container = new Container();
-        $container->set('initialized_session', $session);
-
-        $listener = new SessionListener($container);
+        $request = new Request();
+        $listener = $this->createListener($request, new NativeSessionStorageFactory());
         $kernel = $this->createMock(HttpKernelInterface::class);
 
-        $request = new Request();
         $listener->onKernelRequest(new RequestEvent($kernel, $request, HttpKernelInterface::MAIN_REQUEST));
 
+        $session = $request->getSession();
         $session->start();
         $sessionId = $session->getId();
         $this->assertNotEmpty($sessionId);
@@ -756,4 +736,36 @@ public function testResetUnclosedSession()
         $this->assertEmpty(session_id());
         $this->assertSame(\PHP_SESSION_NONE, session_status());
     }
+
+    private function createListener(Request $request, SessionStorageFactoryInterface $sessionFactory)
+    {
+        $requestStack = new RequestStack();
+        $request = new Request();
+        $requestStack->push($request);
+
+        $sessionFactory = new SessionFactory(
+            $requestStack,
+            $sessionFactory,
+        );
+
+        $container = new Container();
+        $container->set('request_stack', $requestStack);
+        $container->set('session_factory', $sessionFactory);
+
+        $listener = new SessionListener($container);
+
+        return new SessionListener($container);
+    }
+
+    private function createValidSessionId(): string
+    {
+        session_start();
+        $sessionId = session_id();
+        $_SESSION['some'] = 'value';
+        session_write_close();
+        $_SESSION = [];
+        session_abort();
+
+        return $sessionId;
+    }
 }
