Calling setLocation on a RepeaterBuilder should delegate to the Paren…

…tContext Close issue #1
StoutLogic · May 16, 2016 · f6994a3 · f6994a3
1 parent 1c4e8fb
commit f6994a3
Show file tree

Hide file tree

Showing 2 changed files with 21 additions and 0 deletions.
diff --git a/src/RepeaterBuilder.php b/src/RepeaterBuilder.php
@@ -31,4 +31,9 @@ public function endRepeater()
     {
         return $this->getParentContext();
     }
+
+    public function setLocation($param, $operator, $value)
+    {
+        return $this->getParentContext()->setLocation($param, $operator, $value);
+    }
 }
diff --git a/tests/RepeaterBuilderTest.php b/tests/RepeaterBuilderTest.php
@@ -45,4 +45,20 @@ public function testEndRepeater()
                 ->endRepeater()
                 ->addText('parent_title');
     }
+
+    public function testSetLocation()
+    {
+        $fieldsBuilder = $this->getMockBuilder('StoutLogic\AcfBuilder\FieldsBuilder')
+                              ->setConstructorArgs(['parent'])
+                              ->getMock();
+
+        $repeaterBuilder = new RepeaterBuilder('slides');
+        $repeaterBuilder->setParentContext($fieldsBuilder);
+
+        $fieldsBuilder->expects($this->once())->method('setLocation');
+
+        $repeaterBuilder->addText('title')
+                ->addWysiwyg('content')
+                ->setLocation('post_type', '==', 'page');
+    }
 }