B2B-2464: Improve category children loading

- Add test coverage for new $excludedMethodsClassMap dependency

Author: danmooney2

lib/internal/Magento/Framework/Reflection/Test/Unit/DataObjectProcessorTest.php

@@ -26,6 +26,11 @@ class DataObjectProcessorTest extends TestCase
      */
     private $dataObjectProcessor;
 
+    /**
+     * @var MethodsMap
+     */
+    private $methodsMapProcessor;
+
     /**
      * @var ExtensionAttributesProcessor|MockObject
      */
@@ -34,7 +39,7 @@ class DataObjectProcessorTest extends TestCase
     protected function setUp(): void
     {
         $objectManager = new ObjectManager($this);
-        $methodsMapProcessor = $objectManager->getObject(
+        $this->methodsMapProcessor = $objectManager->getObject(
             MethodsMap::class,
             [
                 'fieldNamer' => $objectManager->getObject(FieldNamer::class),
@@ -48,35 +53,40 @@ protected function setUp(): void
             ->willReturn(['unserializedData']);
 
         $objectManager->setBackwardCompatibleProperty(
-            $methodsMapProcessor,
+            $this->methodsMapProcessor,
             'serializer',
             $serializerMock
         );
 
         $this->extensionAttributesProcessorMock = $this->getMockBuilder(ExtensionAttributesProcessor::class)
             ->disableOriginalConstructor()
             ->getMock();
+    }
+
+    /**
+     * @param array $extensionAttributes
+     * @param array $excludedMethodsClassMap
+     * @param array $expectedOutput
+     * @dataProvider buildOutputDataArrayDataProvider
+     */
+    public function testBuildOutputDataArray(
+        array $extensionAttributes,
+        array $excludedMethodsClassMap,
+        array $expectedOutput
+    ) {
+        $objectManager = new ObjectManager($this);
 
         $this->dataObjectProcessor = $objectManager->getObject(
             DataObjectProcessor::class,
             [
-                'methodsMapProcessor' => $methodsMapProcessor,
+                'methodsMapProcessor' => $this->methodsMapProcessor,
                 'typeCaster' => $objectManager->getObject(TypeCaster::class),
                 'fieldNamer' => $objectManager->getObject(FieldNamer::class),
-                'extensionAttributesProcessor' => $this->extensionAttributesProcessorMock
+                'extensionAttributesProcessor' => $this->extensionAttributesProcessorMock,
+                'excludedMethodsClassMap' => $excludedMethodsClassMap,
             ]
         );
-    }
 
-    /**
-     * @param array $extensionAttributes
-     * @param array $expectedOutputDataArray
-     *
-     * @dataProvider buildOutputDataArrayDataProvider
-     */
-    public function testBuildOutputDataArray($extensionAttributes, $expectedOutputDataArray)
-    {
-        $objectManager =  new ObjectManager($this);
         /** @var TestDataObject $testDataObject */
         $testDataObject = $objectManager->getObject(
             TestDataObject::class,
@@ -87,40 +97,70 @@ public function testBuildOutputDataArray($extensionAttributes, $expectedOutputDa
             ]
         );
 
-        $this->extensionAttributesProcessorMock->expects($this->once())
+        if (in_array('getExtensionAttributes', $excludedMethodsClassMap[TestDataInterface::class] ?? [])) {
+            $expectedTimes = $this->never();
+        } else {
+            $expectedTimes = $this->once();
+        }
+
+        $this->extensionAttributesProcessorMock->expects($expectedTimes)
             ->method('buildOutputDataArray')
             ->willReturn($extensionAttributes);
 
         $outputData = $this->dataObjectProcessor
             ->buildOutputDataArray($testDataObject, TestDataInterface::class);
-        $this->assertEquals($expectedOutputDataArray, $outputData);
+        $this->assertEquals($expectedOutput, $outputData);
     }
 
     /**
      * @return array
      */
     public function buildOutputDataArrayDataProvider()
     {
-        $expectedOutputDataArray = [
+        $expectedOutput = [
             'id' => '1',
             'address' => 'someAddress',
             'default_shipping' => 'true',
             'required_billing' => 'false',
         ];
-        $extensionAttributeArray = [
+
+        $extensionAttributes = [
             'attribute1' => 'value1',
-            'attribute2' => 'value2'
+            'attribute2' => 'value2',
         ];
 
         return [
-            'No Attributes' => [[], $expectedOutputDataArray],
-            'With Attributes' => [
-                $extensionAttributeArray,
+            'No Extension Attributes or Excluded Methods' => [
+                [],
+                [],
+                $expectedOutput,
+            ],
+            'With Extension Attributes' => [
+                $extensionAttributes,
+                [],
                 array_merge(
-                    $expectedOutputDataArray,
-                    ['extension_attributes' => $extensionAttributeArray]
-                )
-            ]
+                    $expectedOutput,
+                    ['extension_attributes' => $extensionAttributes]
+                ),
+            ],
+            'With Excluded Method' => [
+                [],
+                [
+                    TestDataInterface::class => [
+                        'getAddress',
+                    ],
+                ],
+                array_diff_key($expectedOutput, array_flip(['address'])),
+            ],
+            'With getExtensionAttributes as Excluded Method' => [
+                $extensionAttributes,
+                [
+                    TestDataInterface::class => [
+                        'getExtensionAttributes',
+                    ],
+                ],
+                $expectedOutput,
+            ],
         ];
     }
 }