Improve tests.

Signed-off-by: Tim Lieberman <[email protected]>
mezzio · Apr 15, 2021 · 22aafb1 · 22aafb1
1 parent 8f561c4
commit 22aafb1
Showing 1 changed file with 16 additions and 2 deletions.
diff --git a/test/FlashMessagesTest.php b/test/FlashMessagesTest.php
@@ -404,7 +404,7 @@ public function testCreationAggregatesThrowsExceptionIfInvalidNumberOfHops()
     public function testFlashNowAcceptsZeroHops()
     {
         $flash = FlashMessages::createFromSession($this->session);
-        $flash->flashNow('test', 'value', 0);
+        $flash->flashNow('test', 'value');
 
         $this->assertSame('value', $flash->getFlash('test'));
     }
@@ -416,6 +416,20 @@ public function testFlashNowWithZeroHopsShouldNotSetValueToSession()
             ->method('set');
 
         $flash = FlashMessages::createFromSession($this->session);
-        $flash->flashNow('test', 'value', 0);
+        $flash->flashNow('test', 'value');
+    }
+
+    public function testFlashNowWithNonzeroHopsShouldSetValueToSession()
+    {
+        $this->session
+            ->expects($this->once())
+            ->method('set')
+            ->with(
+                FlashMessagesInterface::FLASH_NEXT,
+                ['test' => ['value' => 'value', 'hops' => 1]]
+            );
+
+        $flash = FlashMessages::createFromSession($this->session);
+        $flash->flashNow('test', 'value', 1);
     }
 }