Moved array entity data replacement conditional

Author: nathanjosiah

src/Magento/FunctionalTestingFramework/Test/Objects/ActionObject.php

@@ -495,6 +495,10 @@ private function findAndReplaceReferences($objectHandler, $inputString)
                 $this->setTimeout($obj->getElement($objField)->getTimeout());
             } elseif (get_class($obj) == EntityDataObject::class) {
                 $replacement = $this->resolveEntityDataObjectReference($obj, $match);
+
+                if (is_array($replacement)) {
+                    $replacement = '["' . implode('","', array_map('addSlashes', $replacement)) . '"]';
+                }
             }
 
             if ($replacement === null) {
@@ -504,9 +508,6 @@ private function findAndReplaceReferences($objectHandler, $inputString)
                     throw new TestReferenceException("Could not resolve entity reference " . $inputString);
                 }
             }
-            elseif (is_array($replacement)) {
-                $replacement = '["' . implode('","', array_map('addSlashes',$replacement)) . '"]';
-            }
 
             $replacement = $this->resolveParameterization($parameterized, $replacement, $match, $obj);