Added tests for willThrowException, willReturnArgument, willReturnSelf, willReturnOnConsecutiveCalls

Author: fabian

Tests/MockObjectTest.php

@@ -159,6 +159,24 @@ public function testStubbedException()
         $this->fail();
     }
 
+    public function testStubbedWillThrowException()
+    {
+        $mock = $this->getMock('AnInterface');
+        $mock->expects($this->any())
+             ->method('doSomething')
+             ->willThrowException(new Exception);
+
+        try {
+            $mock->doSomething();
+        }
+
+        catch (Exception $e) {
+            return;
+        }
+
+        $this->fail();
+    }
+
     public function testStubbedReturnValue()
     {
         $mock = $this->getMock('AnInterface');
@@ -202,6 +220,23 @@ public function testStubbedReturnValueMap()
         $this->assertEquals(NULL, $mock->doSomething('foo', 'bar'));
     }
 
+    public function testStubbedReturnArgument()
+    {
+        $mock = $this->getMock('AnInterface');
+        $mock->expects($this->any())
+             ->method('doSomething')
+             ->will($this->returnArgument(1));
+
+        $this->assertEquals('b', $mock->doSomething('a', 'b'));
+
+        $mock = $this->getMock('AnInterface');
+        $mock->expects($this->any())
+             ->method('doSomething')
+             ->willReturnArgument(1);
+
+        $this->assertEquals('b', $mock->doSomething('a', 'b'));
+    }
+
     public function testFunctionCallback()
     {
         $mock = $this->getMock('SomeClass', array('doSomething'), array(), '', FALSE);
@@ -219,6 +254,44 @@ public function testFunctionCallback()
         $this->assertEquals('pass', $mock->doSomething('foo', 'bar'));
     }
 
+    public function testStubbedReturnSelf()
+    {
+        $mock = $this->getMock('AnInterface');
+        $mock->expects($this->any())
+             ->method('doSomething')
+             ->will($this->returnSelf());
+
+        $this->assertEquals($mock, $mock->doSomething());
+
+        $mock = $this->getMock('AnInterface');
+        $mock->expects($this->any())
+             ->method('doSomething')
+             ->willReturnSelf();
+
+        $this->assertEquals($mock, $mock->doSomething());
+    }
+
+    public function testStubbedReturnOnConsecutiveCalls()
+    {
+        $mock = $this->getMock('AnInterface');
+        $mock->expects($this->any())
+             ->method('doSomething')
+             ->will($this->onConsecutiveCalls('a', 'b', 'c'));
+
+        $this->assertEquals('a', $mock->doSomething());
+        $this->assertEquals('b', $mock->doSomething());
+        $this->assertEquals('c', $mock->doSomething());
+
+        $mock = $this->getMock('AnInterface');
+        $mock->expects($this->any())
+             ->method('doSomething')
+             ->willReturnOnConsecutiveCalls('a', 'b', 'c');
+
+        $this->assertEquals('a', $mock->doSomething());
+        $this->assertEquals('b', $mock->doSomething());
+        $this->assertEquals('c', $mock->doSomething());
+    }
+
     public function testStaticMethodCallback()
     {
         $mock = $this->getMock('SomeClass', array('doSomething'), array(), '', FALSE);