allow relation without related model (morphTo case)

jarektkaczyk · Jul 17, 2018 · 58ab7d3 · 58ab7d3
1 parent b2b3990
commit 58ab7d3
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/src/EloquentSuite.php b/src/EloquentSuite.php
@@ -48,11 +48,11 @@ trait EloquentSuite
      *
      * @param  string $model        Classname of the model under test
      * @param  string $relation     Relation method (belongsTo, hasMany etc)
-     * @param  string $related      Classname of the related model
+     * @param  string|null $related Classname of the related model
      * @param  ...string $params    Optional params for the relation
      * @return \PHPUnit\Framework\MockObject\MockObject[]
      */
-    public function createRelationMock(string $model, string $relation, string $related, ...$params) : MockObject
+    public function createRelationMock(string $model, string $relation, string $related = null, ...$params) : MockObject
     {
         return $this->createRelationChainMock($model, $relation, $related, ...$params)[0];
     }
@@ -82,11 +82,11 @@ public function createRelationMock(string $model, string $relation, string $rela
      *
      * @param  string $model        Classname of the model under test
      * @param  string $relation     Relation method (belongsTo, hasMany etc)
-     * @param  string $related      Classname of the related model
+     * @param  string|null $related Classname of the related model
      * @param  ...string $params    Optional params for the relation
      * @return [\PHPUnit\Framework\MockObject\MockObject, \Mockery\MockInterface]
      */
-    public function createRelationChainMock(string $model, string $relation, string $related, ...$params) : array
+    public function createRelationChainMock(string $model, string $relation, string $related = null, ...$params) : array
     {
         if (!array_key_exists($relation, self::$eloquent_relations)) {
             $this->fail('Unknown relation provided: ' . $relation);