added shouldReceive method to facade.

taylorotwell · taylorotwell · commit e748e76448b4 · 2013-02-15T09:23:37.000-06:00
diff --git a/src/Illuminate/Support/Facades/Facade.php b/src/Illuminate/Support/Facades/Facade.php
@@ -16,6 +16,34 @@ abstract class Facade {
 	 */
 	protected static $resolvedInstance;
 
+	/**
+	 * Hotswap the underlying instance behind the facade.
+	 *
+	 * @param  mixed  $instance
+	 * @return void
+	 */
+	public static function swap($instance)
+	{
+		static::$resolvedInstance = $instance;
+
+		static::$app->instance(static::getFacadeAccessor(), $instance);
+	}
+
+	/**
+	 * Initiate a mock expectation on the facade.
+	 *
+	 * @param  dynamic
+	 * @return Mockery\Expectation
+	 */
+	public static function shouldReceive()
+	{
+		static::$resolvedInstance = $mock = \Mockery::mock(get_class(static::getFacadeRoot()));
+
+		static::$app->instance(static::getFacadeAccessor(), $mock);
+
+		return call_user_func_array(array($mock, 'shouldReceive'), func_get_args());
+	}
+
 	/**
 	 * Get the root object behind the facade.
 	 *
diff --git a/src/Illuminate/Support/composer.json b/src/Illuminate/Support/composer.json
@@ -11,6 +11,9 @@
         "php": ">=5.3.0",
         "patchwork/utf8": "1.0.*"
     },
+    "require-dev": {
+        "mockery/mockery": "0.7.2"
+    },
     "autoload": {
         "psr-0": {
             "Illuminate\\Support": ""
diff --git a/tests/Support/SupportFacadeTest.php b/tests/Support/SupportFacadeTest.php
@@ -1,34 +1,61 @@
 <?php
 
+use Mockery as m;
+
 class SupportFacadeTest extends PHPUnit_Framework_TestCase {
 
+	public function setUp()
+	{
+		Illuminate\Support\Facades\Facade::clearResolvedInstances();
+	}
+
+
+	public function tearDown()
+	{
+		m::close();
+	}
+
+
 	public function testFacadeCallsUnderlyingApplication()
 	{
-		FacadeStub::setFacadeApplication(array('foo' => new ApplicationStub));
+		$app = new ApplicationStub;
+		$app->setAttributes(array('foo' => $mock = m::mock('StdClass')));
+		$mock->shouldReceive('bar')->once()->andReturn('baz');
+		FacadeStub::setFacadeApplication($app);
 		$this->assertEquals('baz', FacadeStub::bar());
 	}
 
+
+	public function testShouldReceiveReturnsAMockeryMock()
+	{
+		$app = new ApplicationStub;
+		$app->setAttributes(array('foo' => new StdClass));
+		FacadeStub::setFacadeApplication($app);
+
+		$this->assertInstanceOf('Mockery\MockInterface', $mock = FacadeStub::shouldReceive('foo')->once()->with('bar')->andReturn('baz')->getMock());
+		$this->assertEquals('baz', $app['foo']->foo('bar'));
+	}
+
 }
 
 class FacadeStub extends Illuminate\Support\Facades\Facade {
 
-	/**
-	 * Get the registered name of the component.
-	 *
-	 * @return string
-	 */
 	protected static function getFacadeAccessor()
 	{
 		return 'foo';
 	}
 
 }
 
-class ApplicationStub {
+class ApplicationStub implements ArrayAccess {
 
-	public function bar()
-	{
-		return 'baz';
-	}
+	protected $attributes = array();
+
+	public function setAttributes($attributes) { $this->attributes = $attributes; }
+	public function instance($key, $instance) { $this->attributes[$key] = $instance; }
+	public function offsetExists($offset) { isset($this->attributes[$offset]); }
+	public function offsetGet($key) { return $this->attributes[$key]; }
+	public function offsetSet($key, $value) { $this->attributes[$key] = $value; }
+	public function offsetUnset($key) { unset($this->attributes[$key]); }
 
 }
