Add tests for global flash() helper function

Author: ivanvermeyen

src/BaseFlash.php

@@ -27,12 +27,13 @@ public function __construct()
      * Flash a notification.
      *
      * @param string $message
-     * @param string $level
+     * @param string|null $level
      *
      * @return \CodeZero\Flash\Notification
      */
-    public function notification($message, $level = 'default')
+    public function notification($message, $level = null)
     {
+        $level = $level ?: 'default';
         $notification = new Notification($message, $level);
 
         $notifications = $this->getFlashedNotificationsFromSession();

src/helpers.php

@@ -5,11 +5,11 @@
      * Flash a message to the session.
      *
      * @param string|null $message
-     * @param string $level
+     * @param string|null $level
      *
      * @return \CodeZero\Flash\Flash
      */
-    function flash($message = null, $level = 'info')
+    function flash($message = null, $level = null)
     {
         $flash = app('flash');
 

tests/HelperTest.php

@@ -0,0 +1,35 @@
+<?php
+
+namespace CodeZero\Flash\Tests;
+
+use CodeZero\Flash\Flash;
+use CodeZero\Flash\Notification;
+
+class HelperTest extends TestCase
+{
+    /** @test */
+    public function it_returns_a_flash_instance()
+    {
+        $this->assertInstanceOf(Flash::class, flash());
+    }
+
+    /** @test */
+    public function it_flashes_a_notification()
+    {
+        $notification = flash('message');
+
+        $this->assertInstanceOf(Notification::class, $notification);
+        $this->assertEquals('message', flash()->notifications()->first()->message);
+        $this->assertEquals('default', flash()->notifications()->first()->level);
+    }
+
+    /** @test */
+    public function it_flashes_a_notification_with_custom_level()
+    {
+        $notification = flash('message', 'custom');
+
+        $this->assertInstanceOf(Notification::class, $notification);
+        $this->assertEquals('message', flash()->notifications()->first()->message);
+        $this->assertEquals('custom', flash()->notifications()->first()->level);
+    }
+}