Allow use of @-refs in faker provider calls, fixes #50

nelmio · Feb 5, 2014 · 62abcdc · 62abcdc
1 parent ecfed6a
commit 62abcdc
Show file tree

Hide file tree

Showing 2 changed files with 30 additions and 0 deletions.
diff --git a/src/Nelmio/Alice/Loader/Base.php b/src/Nelmio/Alice/Loader/Base.php
@@ -556,6 +556,19 @@ private function process($data, array $variables)
                 return $match[0];
             }, $args);
 
+            // replace references to other objects
+            $args = preg_replace_callback('{(?:\b|^)(?:(?<multi>\d+)x )?(?<!\\\\)@(?<reference>[a-z0-9_.*-]+)(?:\->(?<property>[a-z0-9_-]+))?(?:\b|$)}i', function ($match) use ($that) {
+                $multi    = ('' !== $match['multi']) ? $match['multi'] : null;
+                $property = isset($match['property']) ? $match['property'] : null;
+                if (strpos($match['reference'], '*')) {
+                    return '$that->getRandomReferences(' . var_export($match['reference'], true) . ', ' . var_export($multi, true) . ', ' . var_export($property, true) . ')';
+                }
+                if (null !== $multi) {
+                    throw new \UnexpectedValueException('To use multiple references you must use a mask like "'.$match['multi'].'x @user*", otherwise you would always get only one item.');
+                }
+                return '$that->getReference(' . var_export($match['reference'], true) . ', ' . var_export($property, true) . ')';
+            }, $args);
+
             $locale = var_export($matches['locale'], true);
             $name = var_export($matches['name'], true);
 

diff --git a/tests/Nelmio/Alice/Loader/BaseTest.php b/tests/Nelmio/Alice/Loader/BaseTest.php
@@ -221,6 +221,23 @@ public function testLoadParsesPropertyReferencesGetter()
         $this->assertEquals($res['user1']->getAge(), $res['user2']->favoriteNumber);
     }
 
+    public function testLoadParsesReferencesInFakerProviders()
+    {
+        $loader = new Base('en_US', array(new FakerProvider));
+        $res = $loader->load(array(
+            self::USER => array(
+                'bob' => array(
+                    'username' => 'Bob',
+                ),
+                'user' => array(
+                    'username' => '<noop(@bob)>',
+                ),
+            ),
+        ));
+
+        $this->assertEquals($res['bob'], $res['user']->username);
+    }
+
     /**
      * @expectedException UnexpectedValueException
      * @expectedExceptionMessage Property doesnotexist is not defined for reference user1