Merge branch 'pr-852' into imported-magento-magento2-functional-testing-framework-852

Author: jilu1

dev/tests/unit/Magento/FunctionalTestFramework/Util/TestGeneratorTest.php

@@ -12,13 +12,15 @@
 use Magento\FunctionalTestingFramework\Exceptions\TestReferenceException;
 use Magento\FunctionalTestingFramework\Filter\FilterList;
 use Magento\FunctionalTestingFramework\ObjectManager;
+use Magento\FunctionalTestingFramework\Test\Handlers\TestObjectHandler;
 use Magento\FunctionalTestingFramework\Test\Objects\ActionObject;
 use Magento\FunctionalTestingFramework\Test\Objects\TestHookObject;
 use Magento\FunctionalTestingFramework\Test\Objects\TestObject;
 use Magento\FunctionalTestingFramework\Util\Filesystem\CestFileCreatorUtil;
 use Magento\FunctionalTestingFramework\Util\GenerationErrorHandler;
 use Magento\FunctionalTestingFramework\Util\ModuleResolver;
 use Magento\FunctionalTestingFramework\Util\TestGenerator;
+use ReflectionClass;
 use ReflectionProperty;
 use tests\unit\Util\MagentoTestCase;
 use tests\unit\Util\TestLoggingUtil;
@@ -75,6 +77,8 @@ public function testEntityException(): void
 
         $testObject = new TestObject('sampleTest', ['merge123' => $actionObject], [], [], 'filename');
         $testGeneratorObject = TestGenerator::getInstance('', ['sampleTest' => $testObject]);
+        $this->mockTestObjectHandler();
+
         $testGeneratorObject->createAllTestFiles(null, []);
 
         // assert that no exception for createAllTestFiles and generation error is stored in GenerationErrorHandler
@@ -231,5 +235,25 @@ public static function tearDownAfterClass(): void
         $mftfAppConfigInstance = new ReflectionProperty(MftfApplicationConfig::class, 'MFTF_APPLICATION_CONTEXT');
         $mftfAppConfigInstance->setAccessible(true);
         $mftfAppConfigInstance->setValue(null);
+
+        $property = new ReflectionProperty(TestObjectHandler::class, 'testObjectHandler');
+        $property->setAccessible(true);
+        $property->setValue(null);
+    }
+
+    /**
+     * Mock test object handler for test.
+     */
+    private function mockTestObjectHandler(): void
+    {
+        $testObjectHandlerClass = new ReflectionClass(TestObjectHandler::class);
+        $testObjectHandlerConstructor = $testObjectHandlerClass->getConstructor();
+        $testObjectHandlerConstructor->setAccessible(true);
+        $testObjectHandler = $testObjectHandlerClass->newInstanceWithoutConstructor();
+        $testObjectHandlerConstructor->invoke($testObjectHandler);
+
+        $property = new ReflectionProperty(TestObjectHandler::class, 'testObjectHandler');
+        $property->setAccessible(true);
+        $property->setValue($testObjectHandler);
     }
 }