MQE-2088 : Test generation error on invalid entities

Author: Manjusha.S

src/Magento/FunctionalTestingFramework/Util/TestGenerator.php

@@ -31,6 +31,7 @@
 use Magento\FunctionalTestingFramework\Util\Path\FilePathFormatter;
 use Mustache_Engine;
 use Mustache_Loader_FilesystemLoader;
+use Magento\FunctionalTestingFramework\DataGenerator\Handlers\DataObjectHandler;
 
 /**
  * Class TestGenerator
@@ -904,7 +905,7 @@ public function generateStepsPhp($actionObjects, $generationScope = TestGenerato
                     break;
                 case "createData":
                     $entity = $customActionAttributes['entity'];
-
+                    $this->entityExistsCheck($entity , $stepKey);
                     //TODO refactor entity field override to not be individual actionObjects
                     $customEntityFields =
                         $customActionAttributes[ActionObjectExtractor::ACTION_OBJECT_PERSISTENCE_FIELDS] ?? [];
@@ -922,7 +923,6 @@ public function generateStepsPhp($actionObjects, $generationScope = TestGenerato
                     if (!empty($requiredEntityKeys)) {
                         $requiredEntityKeysArray = '"' . implode('", "', $requiredEntityKeys) . '"';
                     }
-
                     $scope = $this->getObjectScope($generationScope);
 
                     $createEntityFunctionCall = "\t\t\${$actor}->createEntity(";
@@ -1511,7 +1511,8 @@ public function generateStepsPhp($actionObjects, $generationScope = TestGenerato
             }
             $testSteps .= PHP_EOL;
         }
-
+        
+       
         return $testSteps;
     }
 
@@ -1673,6 +1674,9 @@ private function resolveStepKeyReferences($input, $actionGroupOrigin, $matchAll
         $testInvocationKey = ucfirst($actionGroupOrigin[ActionGroupObject::ACTION_GROUP_ORIGIN_TEST_REF]);
 
         foreach ($stepKeys as $stepKey) {
+
+           
+            
             // MQE-1011
             $stepKeyVarRef = "$" . $stepKey;
 
@@ -1877,7 +1881,6 @@ private function generateTestPhp($test)
             $testPhp .= "\t\t\$this->isSuccess = true;" . PHP_EOL;
             $testPhp .= "\t}\n";
         }
-
         return $testPhp;
     }
 
@@ -2049,6 +2052,23 @@ private function addDollarSign($input)
         return sprintf("$%s", ltrim($this->stripQuotes($input), '$'));
     }
 
+    /**
+     * Check if the entity exists
+     *
+     * @param string $entity
+     * @param string $stepKey
+     * @throws TestReferenceException
+     */
+    public function entityExistsCheck($entity , $stepKey)
+    {
+        $retrievedEntity = DataObjectHandler::getInstance()->getObject($entity);
+        if ($retrievedEntity === null) {
+            throw new TestReferenceException(
+                "Test generation failed as entity \"" . $entity . "\" does not exist. at stepkey ".$stepKey
+            );
+        }
+    }
+
     /**
      * Wrap parameters into a function call.
      *