Merge pull request #9 from interactive-solutions/add-entities-to-aliases

Added a exposed method to add entity to alias from outside EntityFixt…
interactive-solutions · Jan 10, 2017 · 3245391 · 3245391
2 parents 9dd8bc6 + a98cb78
commit 3245391
Showing 1 changed file with 12 additions and 3 deletions.
diff --git a/src/Context/EntityFixtureContext.php b/src/Context/EntityFixtureContext.php
@@ -74,6 +74,15 @@ public function getEntityFromAlias(string $alias)
         return $this->aliases[$alias];
     }
 
+    /**
+     * @param $entity
+     * @param $alias
+     */
+    public function addEntityToAlias($entity, $alias)
+    {
+        $this->aliases[$alias] = $entity;
+    }
+
     /**
      * Reset the alias list
      *
@@ -265,7 +274,7 @@ public function anExistingTypeCreatedWithStaticMethodAndAlias($type, $staticMeth
     {
         $entity = $this->anExistingTypeCreatedWithStaticMethod($type, $staticMethodName);
 
-        $this->aliases[$alias] = $entity;
+        $this->addEntityToAlias($entity, $alias);
     }
 
     /**
@@ -318,7 +327,7 @@ public function anExistingAs($type, $alias)
     {
         $entity = $this->anExisting($type);
 
-        $this->aliases[$alias] = $entity;
+        $this->addEntityToAlias($entity, $alias);
     }
 
     /**
@@ -335,7 +344,7 @@ public function anExistingAsWithValues($type, $alias, TableNode $values)
     {
         $entity = $this->anExistingWithValues($type, $values);
 
-        $this->aliases[$alias] = $entity;
+        $this->addEntityToAlias($entity, $alias);
     }
 
     /**