Return instance.

Signed-off-by: crynobone <[email protected]>
orchestral · Jan 4, 2014 · ba2c29a · ba2c29a
1 parent a81a64f
commit ba2c29a
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git a/src/Orchestra/Support/Messages.php b/src/Orchestra/Support/Messages.php
@@ -75,12 +75,14 @@ public function retrieve()
      * Extend Messages instance from session.
      *
      * @param  \Closure $callback
-     * @return void
+     * @return Messages
      */
     public function extend(Closure $callback)
     {
         $instance = $this->retrieve();
         call_user_func($callback, $instance);
+
+        return $instance;
     }
 
     /**

diff --git a/tests/MessagesTest.php b/tests/MessagesTest.php
@@ -116,11 +116,12 @@ public function testExtendMethod()
         };
 
         $stub = with(new Messages)->setSession($session);
-        $stub->extend($callback);
+        $output = $stub->extend($callback);
 
         $retrieve = $stub->retrieve();
         $retrieve->setFormat();
 
+        $this->assertInstanceOf('\Orchestra\Support\Messages', $output);
         $this->assertInstanceOf('\Orchestra\Support\Messages', $retrieve);
         $this->assertEquals(array('Hi World', 'Hi Orchestra Platform'), $retrieve->get('hello'));
     }