Add flash message assert

wavevision · Oct 18, 2019 · 4b1c82b · 4b1c82b
1 parent cd03f99
commit 4b1c82b
Show file tree

Hide file tree

Showing 7 changed files with 40 additions and 1 deletion.
diff --git a/src/NetteTests/Configuration.php b/src/NetteTests/Configuration.php
@@ -20,4 +20,9 @@ public static function getConfiguratorFactory(): callable
 		return self::$configuratorFactory;
 	}
 
+	public static function setup(): void
+	{
+		ob_start();
+	}
+
 }
diff --git a/src/NetteTests/TestCases/Parts/PresenterAsserts.php b/src/NetteTests/TestCases/Parts/PresenterAsserts.php
@@ -95,6 +95,18 @@ protected function extractForwardResponseRequest(PresenterResponse $presenterRes
 		return $this->extractForwardResponse($presenterResponse)->getRequest();
 	}
 
+	/**
+	 * @return array<mixed>
+	 */
+	protected function extractFlashMessages(PresenterResponse $presenterResponse): array
+	{
+		$flashes = [];
+		foreach ($presenterResponse->getPresenterRequest()->getPresenter()->getTemplate()->flashes as $flash) {
+			$flashes[] = (array)$flash;
+		}
+		return $flashes;
+	}
+
 	protected function runPresenter(PresenterRequest $presenterRequest): PresenterResponse
 	{
 		$this->presenters->setup($presenterRequest);

diff --git a/tests/app/config/common.neon b/tests/app/config/common.neon
@@ -5,6 +5,7 @@ application:
 
 session:
 	expiration: 14 days
+	autoStart: true
 
 
 services:

diff --git a/tests/app/src/Presenters/ExamplePresenter.php b/tests/app/src/Presenters/ExamplePresenter.php
@@ -48,4 +48,10 @@ public function actionForwardResponse(): void
 		$this->forward('default');
 	}
 
+	public function actionFlash(): void
+	{
+		$this->flashMessage('Hello there!', 'warning');
+		$this->terminate();
+	}
+
 }
diff --git a/tests/app/tests/Presenters/ExamplePresenterTest.php b/tests/app/tests/Presenters/ExamplePresenterTest.php
@@ -15,6 +15,21 @@
 class ExamplePresenterTest extends PresenterTestCase
 {
 
+	public function testFlashMessage(): void
+	{
+		$this->assertSame(
+			[
+				[
+					'message' => 'Hello there!',
+					'type' => 'warning',
+				],
+			],
+			$this->extractFlashMessages(
+				$this->runPresenter(new PresenterRequest(ExamplePresenter::class, 'flash'))
+			)
+		);
+	}
+
 	public function testAjaxRequest(): void
 	{
 		$this->assertInstanceOf(

diff --git a/tests/app/tests/Presenters/FormPresenterTest.php b/tests/app/tests/Presenters/FormPresenterTest.php
@@ -12,7 +12,6 @@ class FormPresenterTest extends PresenterTestCase
 
 	public function testSubmitForm(): void
 	{
-		$this->assertEquals([], []);
 		$this->assertValidForm(
 			$this->submitForm(
 				new SubmitFormRequest(

diff --git a/tests/app/tests/bootstrap.php b/tests/app/tests/bootstrap.php
@@ -4,6 +4,7 @@
 use Wavevision\NetteTests\Configuration;
 
 require __DIR__ . '/../../../vendor/autoload.php';
+Configuration::setup();
 Configuration::setConfiguratorFactory(
 	function (): Configurator {
 		$configurator = new Configurator();