Test LocatorAwareTrait seperately.

cakephp · Apr 21, 2017 · d58c185 · d58c185
1 parent 27a9897
commit d58c185
Showing 1 changed file with 23 additions and 0 deletions.
diff --git a/tests/TestCase/ORM/Locator/LocatorAwareTraitTest.php b/tests/TestCase/ORM/Locator/LocatorAwareTraitTest.php
@@ -41,10 +41,33 @@ public function setUp()
      * @return void
      */
     public function testTableLocator()
+    {
+        $tableLocator = $this->subject->tableLocator();
+        $this->assertSame(TableRegistry::getTableLocator(), $tableLocator);
+
+        $newLocator = $this->getMockBuilder('Cake\ORM\Locator\LocatorInterface')->getMock();
+        $subjectLocator = $this->subject->tableLocator($newLocator);
+        $this->assertSame($newLocator, $subjectLocator);
+    }
+
+    /**
+     * Tests testGetTableLocator method
+     *
+     * @return void
+     */
+    public function testGetTableLocator()
     {
         $tableLocator = $this->subject->getTableLocator();
         $this->assertSame(TableRegistry::getTableLocator(), $tableLocator);
+    }
 
+    /**
+     * Tests testSetTableLocator method
+     *
+     * @return void
+     */
+    public function testSetTableLocator()
+    {
         $newLocator = $this->getMockBuilder('Cake\ORM\Locator\LocatorInterface')->getMock();
         $this->subject->setTableLocator($newLocator);
         $subjectLocator = $this->subject->getTableLocator();