fix getWithClass for HABTM relations

bytic · Aug 21, 2016 · da1de03 · da1de03
1 parent 847428c
commit da1de03
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 3 deletions.
diff --git a/src/Records/Relations/HasAndBelongsToMany.php b/src/Records/Relations/HasAndBelongsToMany.php
@@ -74,6 +74,7 @@ protected function initJoinFields()
 
     /**
      * Simple select query from the link table
+     * @param bool $specific
      * @return Query
      */
     public function getLinkQuery($specific = true)
@@ -91,8 +92,6 @@ public function getLinkQuery($specific = true)
         return $query;
     }
 
-
-
     /**
      * @param RecordCollection $collection
      * @return RecordCollection
@@ -168,6 +167,11 @@ protected function _delete()
     }
 
 
+    public function getWithClass()
+    {
+        return $this->getName();
+    }
+
     /**
      * @return Connection
      */

diff --git a/src/Records/Relations/Relation.php b/src/Records/Relations/Relation.php
@@ -216,10 +216,15 @@ public function getWith()
 
     public function initWith()
     {
-        $className = inflector()->pluralize($this->getName());
+        $className = $this->getWithClass();
         $this->setWithClass($className);
     }
 
+    public function getWithClass()
+    {
+        return inflector()->pluralize($this->getName());
+    }
+
     /**
      * @param string $name
      */