Add setClient

ElfSundae · Sep 4, 2017 · fb5d01c · fb5d01c
1 parent 78b398a
commit fb5d01c
Show file tree

Hide file tree

Showing 2 changed files with 44 additions and 3 deletions.
diff --git a/src/Message.php b/src/Message.php
@@ -69,9 +69,7 @@ class Message implements JsonSerializable
      */
     public function __construct(Client $client = null)
     {
-        if ($this->client = $client) {
-            $this->configureDefaults($client->getMessageDefaults(), true);
-        }
+        $this->setClient($client);
     }
 
     /**
@@ -84,6 +82,23 @@ public function getClient()
         return $this->client;
     }
 
+    /**
+     * Set the BearyChat Client instance.
+     *
+     * @param  \ElfSundae\BearyChat\Client  $client
+     * @return $this
+     */
+    public function setClient($client)
+    {
+        if ($client instanceof Client && $this->client != $client) {
+            $this->configureDefaults($client->getMessageDefaults(), true);
+        }
+
+        $this->client = $client;
+
+        return $this;
+    }
+
     /**
      * Get the text.
      *

diff --git a/tests/MessageTest.php b/tests/MessageTest.php
@@ -521,6 +521,32 @@ public function testAddImage()
         ], $message->toArray());
     }
 
+    public function testSetClient()
+    {
+        $message = new Message;
+        $client = m::mock(Client::class)
+            ->shouldReceive('getMessageDefaults')
+            ->once()
+            ->andReturn(['user' => 'elf'])
+            ->mock();
+        $message->setClient($client);
+        $this->assertSame($client, $message->getClient());
+        $this->assertSame('elf', $message->getUser());
+
+        $message->setClient(null);
+        $this->assertNull($message->getClient());
+
+        $client = m::mock(Client::class)
+            ->shouldReceive('getMessageDefaults')
+            ->once()
+            ->andReturn(['user' => 'sundae', 'markdown' => false])
+            ->mock();
+        $message->setClient($client);
+        $this->assertSame($client, $message->getClient());
+        $this->assertSame('elf', $message->getUser());
+        $this->assertSame(false, $message->getMarkdown());
+    }
+
     protected function getClient($defaults = ['user' => 'elf', 'notification' => 'noti', 'attachment_color' => '#f00'])
     {
         return m::mock(Client::class)