trappar#10 - add ability to skip not initialized collections

zhil · May 30, 2017 · 1c19eba · 1c19eba
1 parent 1abef57
commit 1c19eba
Show file tree

Hide file tree

Showing 3 changed files with 46 additions and 1 deletion.
diff --git a/src/Trappar/AliceGenerator/FixtureGenerationContext.php b/src/Trappar/AliceGenerator/FixtureGenerationContext.php
@@ -10,6 +10,10 @@
 
 class FixtureGenerationContext
 {
+    /**
+     * @var int
+     */
+    private $skipNotInitializedCollections = false;
     /**
      * @var int
      */
@@ -143,4 +147,22 @@ public function setSortResults($sortResults)
 
         return $this;
     }
+
+    /**
+     * @return int
+     */
+    public function getSkipNotInitializedCollections(): int
+    {
+        return $this->skipNotInitializedCollections;
+    }
+
+    /**
+     * @param int $skipNotInitializedCollections
+     * @return self
+     */
+    public function setSkipNotInitializedCollections(int $skipNotInitializedCollections)
+    {
+        $this->skipNotInitializedCollections = $skipNotInitializedCollections;
+        return $this;
+    }
 }
diff --git a/src/Trappar/AliceGenerator/ObjectHandler/CollectionHandler.php b/src/Trappar/AliceGenerator/ObjectHandler/CollectionHandler.php
@@ -16,7 +16,12 @@ public function handle(ValueContext $valueContext)
             return false;
         }
 
-        $valueContext->setValue($collection->toArray());
+        $fixturesGenerationContext = $valueContext->getValueVisitor()->getFixtureGenerationContext();
+        if($fixturesGenerationContext->getSkipNotInitializedCollections() && !$collection->isInitialized()) {
+            $valueContext->setValue([]);
+        } else {
+            $valueContext->setValue($collection->toArray());
+        }
         $valueContext->getValueVisitor()->visitArray($valueContext);
 
         return true;

diff --git a/src/Trappar/AliceGenerator/ValueVisitor.php b/src/Trappar/AliceGenerator/ValueVisitor.php
@@ -243,4 +243,22 @@ private function handlePersistedObject($object, $reference)
             return true;
         }
     }
+
+    /**
+     * @return FixtureGenerationContext
+     */
+    public function getFixtureGenerationContext(): FixtureGenerationContext
+    {
+        return $this->fixtureGenerationContext;
+    }
+
+    /**
+     * @param FixtureGenerationContext $fixtureGenerationContext
+     * @return self
+     */
+    public function setFixtureGenerationContext(FixtureGenerationContext $fixtureGenerationContext)
+    {
+        $this->fixtureGenerationContext = $fixtureGenerationContext;
+        return $this;
+    }
 }