Merge branch 'customise-storage-key'

Closes #35, closes #10
slimphp · Oct 22, 2017 · 9aaff5f · 9aaff5f
2 parents 4434777 + a34986f
commit 9aaff5f
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 1 deletion.
diff --git a/src/Messages.php b/src/Messages.php
@@ -52,8 +52,12 @@ class Messages
      * @throws RuntimeException if the session cannot be found
      * @throws InvalidArgumentException if the store is not array-like
      */
-    public function __construct(&$storage = null)
+    public function __construct(&$storage = null, $storageKey = null)
     {
+        if (is_string($storageKey) && $storageKey) {
+            $this->storageKey = $storageKey;
+        }
+
         // Set storage
         if (is_array($storage) || $storage instanceof ArrayAccess) {
             $this->storage = &$storage;

diff --git a/tests/MessagesTest.php b/tests/MessagesTest.php
@@ -269,4 +269,14 @@ public function testClearMessage()
         $this->assertFalse($flash->hasMessage('Now'));
         $this->assertTrue($flash->hasMessage('Foo'));
     }
+
+    public function testSettingCustomStorageKey()
+    {
+        $storage = ['some-key' => [ 'Test' => ['Test']]];
+        $flash = new Messages($storage);
+        $this->assertFalse($flash->hasMessage('Test'));
+
+        $flash = new Messages($storage, 'some-key');
+        $this->assertTrue($flash->hasMessage('Test'));
+    }
 }