Add temporary registry

Author: 1blankz7

src/ObjectGraphGenerator.php

@@ -28,6 +28,9 @@ class ObjectGraphGenerator
     /** @var array */
     private $registry;
 
+    /** @var array */
+    private $temporaryRegistry = [];
+
     public function __construct(array $registry = [])
     {
         $this->fakerInstance = Factory::create();
@@ -40,16 +43,18 @@ public function __construct(array $registry = [])
         $this->registry = $registry;
     }
 
-    public function generateWithSeed(string $className, int $seed): object
+    public function generate(string $className): object
     {
-        $this->fakerInstance->seed($seed);
-
         return $this->generateObject($className);
     }
 
-    public function generate(string $className): object
+    public function generateWithTemporaryConfig(string $className, array $config): object
     {
-        return $this->generateObject($className);
+        $this->temporaryRegistry = $config;
+        $object = $this->generateObject($className);
+        $this->temporaryRegistry = [];
+
+        return $object;
     }
 
     /**
@@ -146,11 +151,14 @@ function () use ($argumentName, $className, $type) {
 
     private function isInRegistry(string $key): bool
     {
-        return array_key_exists($key, $this->registry);
+        return array_key_exists($key, $this->temporaryRegistry) || array_key_exists($key, $this->registry);
     }
 
     private function getFromRegistry(string $key)
     {
+        if (isset($this->temporaryRegistry[$key])) {
+            return $this->temporaryRegistry[$key]($this, $this->fakerInstance);
+        }
         return $this->registry[$key]($this, $this->fakerInstance);
     }
 }

tests/ObjectGraphGeneratorTest.php

@@ -28,6 +28,20 @@ public function itShouldGenerateObjectOfGivenType(string $class): void
         $this->assertInstanceOf($class, $actual);
     }
 
+    /**
+     * @test
+     */
+    public function itShouldUseTemporaryConfigBeforeRegistry(): void
+    {
+        $objectGraphGenerator = $this->getUnitUnderTest();
+        /** @var SimpleTypesInConstructor $actual */
+        $actual = $objectGraphGenerator->generateWithTemporaryConfig(SimpleTypesInConstructor::class, [
+            'Cubicl\ObjectGraphGenerator\Tests\Fixture\SimpleTypesInConstructor:bar' => function () {return true;},
+        ]);
+
+        $this->assertTrue($actual->isBar());
+    }
+
     public function dataForTest(): array
     {
         return [