#10 - another approach of limiting collections

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()
+    {
+        return $this->skipNotInitializedCollections;
+    }
+
+    /**
+     * @param int $skipNotInitializedCollections
+     * @return self
+     */
+    public function setSkipNotInitializedCollections($skipNotInitializedCollections)
+    {
+        $this->skipNotInitializedCollections = $skipNotInitializedCollections;
+        return $this;
+    }
 }

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;

src/Trappar/AliceGenerator/ValueVisitor.php

@@ -243,4 +243,22 @@ private function handlePersistedObject($object, $reference)
             return true;
         }
     }
+
+    /**
+     * @return FixtureGenerationContext
+     */
+    public function getFixtureGenerationContext()
+    {
+        return $this->fixtureGenerationContext;
+    }
+
+    /**
+     * @param FixtureGenerationContext $fixtureGenerationContext
+     * @return self
+     */
+    public function setFixtureGenerationContext(FixtureGenerationContext $fixtureGenerationContext)
+    {
+        $this->fixtureGenerationContext = $fixtureGenerationContext;
+        return $this;
+    }
 }