From 9af073977c16d0010aabc273756c1abf8addb20f Mon Sep 17 00:00:00 2001
From: Morven Lewis-Everley <morven@i-lateral.com>
Date: Wed, 20 Dec 2017 17:38:59 +0000
Subject: [PATCH] update tests

---
 tests/SystemMessageTest.php | 20 +++++++++++++++++---
 1 file changed, 17 insertions(+), 3 deletions(-)

diff --git a/tests/SystemMessageTest.php b/tests/SystemMessageTest.php
index 9dda695..f75ae89 100644
--- a/tests/SystemMessageTest.php
+++ b/tests/SystemMessageTest.php
@@ -13,8 +13,8 @@ public function testClosedByUser()
     {
         $member = $this->objFromFixture('Member', 'User1');
         $message = $this->objFromFixture('SystemMessage', 'SecondMessage');
-        $this->assertNotFalse($message->isClosedByMember($member));
-        $this->assertNotFalse($message->isClosed($member));
+        $this->assertTrue($message->isClosedByMember($member));
+        $this->assertTrue($message->isClosed($member));
         $this->assertFalse($message->isOpen($member));
     }
 
@@ -28,7 +28,21 @@ public function testClosedByCookie()
         $message = $this->objFromFixture('SystemMessage', 'FirstMessage');
         Cookie::set("systemmessage.closed.{$message->ID}", true);
 
-        $this->assertNotFalse($message->isClosed());
+        $this->assertTrue($message->isClosed());
+        $this->assertFalse($message->isOpen());
+    }
+
+    /**
+     * Test to see if message is closed via cookie is flagged as "closed" or
+     * "open"
+     */
+    public function testClosedBySession()
+    {
+        // Get message and set Cookie
+        $message = $this->objFromFixture('SystemMessage', 'FirstMessage');
+        Session::set("systemmessage.closed.{$message->ID}", true);
+
+        $this->assertTrue($message->isClosed());
         $this->assertFalse($message->isOpen());
     }