testExposeAllMethodsWillExposeAllMethods

elliotchance · Dec 21, 2014 · 45f191e · 45f191e
1 parent 8825f46
commit 45f191e
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 3 deletions.
diff --git a/src/Concise/Mock/MockBuilder.php b/src/Concise/Mock/MockBuilder.php
@@ -11,8 +11,6 @@
 use ReflectionClass;
 use Closure;
 use Concise\Validation\ArgumentChecker;
-use ReflectionException;
-use ReflectionMethod;
 
 class MockBuilder
 {
@@ -516,6 +514,12 @@ public function getClassName()
      */
     public function exposeAll()
     {
+        $reflection = new ReflectionClass($this->className);
+        foreach ($reflection->getMethods() as $method) {
+            if (!$method->isFinal() && !$method->isPrivate()) {
+                $this->expose($method->getName());
+            }
+        }
         return $this;
     }
 }
diff --git a/tests/Concise/Mock/BuilderExposeTest.php b/tests/Concise/Mock/BuilderExposeTest.php
@@ -82,8 +82,20 @@ public function testExposeTwoMethodsWithArraySyntax(MockBuilder $builder, $type)
      * @group #189
      * @dataProvider allBuilders
      */
-    public function testExposeAllCanBeChained(MockBuilder $builder)
+    public function testExposeAllCanBeChained(MockBuilder $builder, $type)
     {
+        $this->youCannotExposeAMethodOnAMockThatIsNotNice($type);
         $this->assert($builder->exposeAll(), is_the_same_as, $builder);
     }
+
+    /**
+     * @group #189
+     * @dataProvider allBuilders
+     */
+    public function testExposeAllMethodsWillExposeAllMethods(MockBuilder $builder, $type)
+    {
+        $this->youCannotExposeAMethodOnAMockThatIsNotNice($type);
+        $mock = $builder->exposeAll()->get();
+        $this->assert($mock->mySecretMethod(), equals, 'abc');
+    }
 }