AC-15107: PHPUnit 12 Upgrade | Moved common function into seperate file

Author: manishranjan-adobe

lib/internal/Magento/Framework/TestFramework/Unit/Helper/ObjectManager.php

@@ -1,7 +1,7 @@
 <?php
 /**
- * Copyright © Magento, Inc. All rights reserved.
- * See COPYING.txt for license details.
+ * Copyright 2012 Adobe
+ * All Rights Reserved.
  */
 namespace Magento\Framework\TestFramework\Unit\Helper;
 
@@ -93,20 +93,22 @@ protected function _processSpecialCases($className, $arguments)
      */
     protected function _getResourceModelMock()
     {
-        $resourceMock = $this->_testObject->getMockBuilder(\Magento\Framework\Module\ModuleResource::class)
-            ->disableOriginalConstructor()
-            ->disableOriginalClone()
-            ->disableArgumentCloning()
-            ->disallowMockingUnknownTypes()
-            ->onlyMethods(['getIdFieldName', '__sleep', '__wakeup'])
-            ->getMock();
+        $reflection = new \ReflectionClass($this->_testObject);
+        $method = $reflection->getMethod('createPartialMock');
+        $method->setAccessible(true);
+        $resourceMock = $method->invoke(
+            $this->_testObject,
+            \Magento\Framework\Module\ModuleResource::class,
+            ['getIdFieldName', '__sleep', '__wakeup']
+        );
+        $reflection = new \ReflectionClass($this->_testObject);
+        $anyMethod = $reflection->getMethod('any');
+        $anyMethod->setAccessible(true);
         $resourceMock->expects(
-            $this->_testObject->any()
+            $anyMethod->invoke($this->_testObject)
         )->method(
             'getIdFieldName'
-        )->will(
-            $this->_testObject->returnValue('id')
-        );
+        )->willReturn('id');
 
         return $resourceMock;
     }
@@ -119,17 +121,21 @@ protected function _getResourceModelMock()
      */
     protected function _getTranslatorMock($className)
     {
-        $translator = $this->_testObject->getMockBuilder($className)->disableOriginalConstructor()->getMock();
+        $reflection = new \ReflectionClass($this->_testObject);
+        $method = $reflection->getMethod('createMock');
+        $method->setAccessible(true);
+        $translator = $method->invoke($this->_testObject, $className);
         $translateCallback = function ($arguments) {
             return is_array($arguments) ? vsprintf(array_shift($arguments), $arguments) : '';
         };
+        $reflection = new \ReflectionClass($this->_testObject);
+        $anyMethod = $reflection->getMethod('any');
+        $anyMethod->setAccessible(true);
         $translator->expects(
-            $this->_testObject->any()
+            $anyMethod->invoke($this->_testObject)
         )->method(
             'translate'
-        )->will(
-            $this->_testObject->returnCallback($translateCallback)
-        );
+        )->willReturnCallback($translateCallback);
         return $translator;
     }
 
@@ -141,13 +147,11 @@ protected function _getTranslatorMock($className)
      */
     protected function _getMockWithoutConstructorCall($className)
     {
-        $mock = $this->_testObject->getMockBuilder($className)
-            ->disableOriginalConstructor()
-            ->disableOriginalClone()
-            ->disableArgumentCloning()
-            ->disallowMockingUnknownTypes()
-            ->getMock();
-        return $mock;
+        // Use reflection to call protected createMock method
+        $reflection = new \ReflectionClass($this->_testObject);
+        $method = $reflection->getMethod('createMock');
+        $method->setAccessible(true);
+        return $method->invoke($this->_testObject, $className);
     }
 
     /**
@@ -191,24 +195,24 @@ public function getObject($className, array $arguments = [])
     protected function getBuilder($className, array $arguments)
     {
         if (!isset($arguments['objectFactory'])) {
-            $objectFactory = $this->_testObject->getMockBuilder(\Magento\Framework\Api\ObjectFactory::class)
+            $reflection = new \ReflectionClass($this->_testObject);
+            $method = $reflection->getMethod('getMockBuilder');
+            $method->setAccessible(true);
+            $mockBuilder = $method->invoke($this->_testObject, \Magento\Framework\Api\ObjectFactory::class);
+            
+            // Use onlyMethods() with methods that actually exist in ObjectFactory
+            $objectFactory = $mockBuilder->onlyMethods(['create', 'get'])
                 ->disableOriginalConstructor()
-                ->disableOriginalClone()
-                ->disableArgumentCloning()
-                ->disallowMockingUnknownTypes()
-                ->addMethods(['populateWithArray', 'populate'])
-                ->onlyMethods(['create'])
                 ->getMock();
 
-            $objectFactory->expects($this->_testObject->any())
-                ->method('populateWithArray')
-                ->will($this->_testObject->returnSelf());
-            $objectFactory->expects($this->_testObject->any())
-                ->method('populate')
-                ->will($this->_testObject->returnSelf());
-            $objectFactory->expects($this->_testObject->any())
+            $reflection = new \ReflectionClass($this->_testObject);
+            $anyMethod = $reflection->getMethod('any');
+            $anyMethod->setAccessible(true);
+            
+            // Only configure methods that actually exist in ObjectFactory
+            $objectFactory->expects($anyMethod->invoke($this->_testObject))
                 ->method('create')
-                ->will($this->_testObject->returnCallback(
+                ->willReturnCallback(
                     function ($className, $arguments) {
                         $reflectionClass = new \ReflectionClass($className);
                         $constructorMethod = $reflectionClass->getConstructor();
@@ -232,7 +236,15 @@ function ($className, $arguments) {
                         }
                         return new $className(...array_values($args));
                     }
-                ));
+                );
+            
+            $objectFactory->expects($anyMethod->invoke($this->_testObject))
+                ->method('get')
+                ->willReturnCallback(
+                    function ($className) {
+                        return $this->_getMockWithoutConstructorCall($className);
+                    }
+                );
 
             $arguments['objectFactory'] = $objectFactory;
         }
@@ -281,12 +293,10 @@ public function getConstructArguments($className, array $arguments = [])
                 if ($firstPosition !== false) {
                     $parameterString = substr($parameterString, $firstPosition + 11);
                     $parameterString = substr($parameterString, 0, strpos($parameterString, ' '));
-                    $object = $this->_testObject->getMockBuilder($parameterString)
-                        ->disableOriginalConstructor()
-                        ->disableOriginalClone()
-                        ->disableArgumentCloning()
-                        ->disallowMockingUnknownTypes()
-                        ->getMock();
+                    $reflection = new \ReflectionClass($this->_testObject);
+                    $method = $reflection->getMethod('createMock');
+                    $method->setAccessible(true);
+                    $object = $method->invoke($this->_testObject, $parameterString);
                 }
             }
 
@@ -310,20 +320,19 @@ public function getCollectionMock($className, array $data)
                 $className . ' does not instance of \Magento\Framework\Data\Collection'
             );
         }
-        $mock = $this->_testObject->getMockBuilder($className)
-            ->disableOriginalConstructor()
-            ->disableOriginalClone()
-            ->disableArgumentCloning()
-            ->disallowMockingUnknownTypes()
-            ->getMock();
+        $reflection = new \ReflectionClass($this->_testObject);
+        $method = $reflection->getMethod('createMock');
+        $method->setAccessible(true);
+        $mock = $method->invoke($this->_testObject, $className);
         $iterator = new \ArrayIterator($data);
+        $reflection = new \ReflectionClass($this->_testObject);
+        $anyMethod = $reflection->getMethod('any');
+        $anyMethod->setAccessible(true);
         $mock->expects(
-            $this->_testObject->any()
+            $anyMethod->invoke($this->_testObject)
         )->method(
             'getIterator'
-        )->will(
-            $this->_testObject->returnValue($iterator)
-        );
+        )->willReturn($iterator);
         return $mock;
     }
 
@@ -370,16 +379,41 @@ public function setBackwardCompatibleProperty($object, $propertyName, $propertyV
      */
     public function prepareObjectManager(array $map = [])
     {
-        $objectManagerMock = $this->_testObject->getMockBuilder(ObjectManagerInterface::class)
-            ->addMethods(['getInstance'])
-            ->onlyMethods(['get'])
-            ->getMockForAbstractClass();
+        $reflection = new \ReflectionClass($this->_testObject);
+        $method = $reflection->getMethod('createMock');
+        $method->setAccessible(true);
+        $objectManagerMock = $method->invoke(
+            $this->_testObject,
+            ObjectManagerInterface::class
+        );
 
-        $objectManagerMock->method('getInstance')->willReturnSelf();
         $objectManagerMock->method('get')->willReturnMap($map);
 
         $reflectionProperty = new \ReflectionProperty(AppObjectManager::class, '_instance');
         $reflectionProperty->setAccessible(true);
         $reflectionProperty->setValue($objectManagerMock, $objectManagerMock);
     }
+
+    /**
+     * Create a partial mock with reflection.
+     *
+     * @param string $className
+     * @param array $methods
+     * @return MockObject
+     */
+    public function createPartialMockWithReflection(string $className, array $methods): MockObject
+    {
+        $reflection = new \ReflectionClass($this->_testObject);
+        $getMockBuilderMethod = $reflection->getMethod('getMockBuilder');
+        $getMockBuilderMethod->setAccessible(true);
+        $mockBuilder = $getMockBuilderMethod->invoke($this->_testObject, $className);
+        
+        $builderReflection = new \ReflectionClass($mockBuilder);
+        $methodsProperty = $builderReflection->getProperty('methods');
+        $methodsProperty->setAccessible(true);
+        $methodsProperty->setValue($mockBuilder, $methods);
+        
+        $mockBuilder->disableOriginalConstructor();
+        return $mockBuilder->getMock();
+    }
 }