33293: Fixed code

Author: anzin

dev/tests/unit/Magento/FunctionalTestFramework/DataGenerator/Handlers/PersistedObjectHandlerTest.php

@@ -8,17 +8,16 @@
 namespace tests\unit\Magento\FunctionalTestFramework\DataGenerator\Handlers;
 
 use Exception;
+use Magento\FunctionalTestingFramework\DataGenerator\Handlers\DataObjectHandler;
 use Magento\FunctionalTestingFramework\DataGenerator\Handlers\PersistedObjectHandler;
 use Magento\FunctionalTestingFramework\DataGenerator\Parsers\DataProfileSchemaParser;
 use Magento\FunctionalTestingFramework\DataGenerator\Persist\CurlHandler;
 use Magento\FunctionalTestingFramework\Exceptions\TestFrameworkException;
 use Magento\FunctionalTestingFramework\Exceptions\TestReferenceException;
 use Magento\FunctionalTestingFramework\ObjectManager;
 use Magento\FunctionalTestingFramework\ObjectManagerFactory;
-use Magento\FunctionalTestingFramework\DataGenerator\Handlers\DataObjectHandler;
 use ReflectionProperty;
 use tests\unit\Util\MagentoTestCase;
-use tests\unit\Util\ObjectHandlerUtil;
 use tests\unit\Util\TestLoggingUtil;
 
 /**
@@ -261,15 +260,14 @@ public function testUpdateSimpleEntity(): void
         ";
 
         // Mock Classes
-        ObjectHandlerUtil::mockDataObjectHandlerWithData($parserOutput);
-        $this->mockCurlHandler($jsonResponse);
+        $this->mockCurlHandler($jsonResponse, $parserOutput);
         $handler = PersistedObjectHandler::getInstance();
         $handler->createEntity(
             $entityStepKey,
             $scope,
             $entityName
         );
-        $this->mockCurlHandler($updatedResponse);
+        $this->mockCurlHandler($updatedResponse, $parserOutput);
 
         // Call method
         $handler->updateEntity(