ACP2E-3492: added test coverage

Author: alzota

app/code/Magento/CustomerGraphQl/Test/Unit/Model/Resolver/Cache/Customer/ModelHydratorTest.php

@@ -0,0 +1,172 @@
+<?php
+/**
+ * Copyright 2024 Adobe.
+ * All Rights Reserved.
+ */
+declare(strict_types=1);
+
+namespace Magento\CustomerGraphQl\Test\Unit\Model\Resolver\Cache\Customer;
+
+use Magento\Customer\Model\Data\Customer;
+use Magento\Customer\Model\Data\CustomerFactory;
+use Magento\CustomerGraphQl\Model\Resolver\Cache\Customer\ModelHydrator;
+use Magento\Framework\EntityManager\HydratorInterface;
+use Magento\Framework\EntityManager\HydratorPool;
+use PHPUnit\Framework\MockObject\Exception;
+use PHPUnit\Framework\MockObject\MockObject;
+use PHPUnit\Framework\TestCase;
+
+class ModelHydratorTest extends TestCase
+{
+    /**
+     * @var CustomerFactory|MockObject
+     */
+    private $customerFactory;
+
+    /**
+     * @var HydratorPool|MockObject
+     */
+    private $hydratorPool;
+
+    /**
+     * @var HydratorInterface|MockObject
+     */
+    private $hydrator;
+
+    /**
+     * @var ModelHydrator
+     */
+    private $modelHydrator;
+
+    protected function setUp(): void
+    {
+        $this->customerFactory = $this->createMock(CustomerFactory::class);
+        $this->hydratorPool = $this->createMock(HydratorPool::class);
+        $this->hydrator = $this->createMock(HydratorInterface::class);
+
+        $this->modelHydrator = new ModelHydrator(
+            $this->customerFactory,
+            $this->hydratorPool
+        );
+    }
+
+    /**
+     * Test hydrate method with existing model
+     *
+     * @return void
+     * @throws Exception
+     */
+
+    public function testHydrateWithExistingModel()
+    {
+        $customer = $this->createMock(Customer::class);
+        $resolverData = [
+            'model_id' => 1,
+            'model_entity_type' => 'customer',
+            'model_data' => ['id' => 1]
+        ];
+
+        $this->customerFactory
+            ->method('create')
+            ->willReturn($customer);
+        $this->hydrator
+            ->method('hydrate')
+            ->with($customer, $resolverData['model_data'])
+            ->willReturnSelf();
+        $this->hydratorPool
+            ->method('getHydrator')
+            ->with('customer')
+            ->willReturn($this->hydrator);
+
+        $this->modelHydrator->hydrate($resolverData);
+        $this->modelHydrator->hydrate($resolverData);
+        $this->assertSame($customer, $resolverData['model']);
+    }
+
+    /**
+     * Test hydrate method with new model
+     *
+     * @return void
+     * @throws Exception
+     */
+
+    public function testHydrateWithNewModel()
+    {
+        $customer = $this->createMock(Customer::class);
+        $resolverData = [
+            'model_id' => 1,
+            'model_entity_type' => 'customer',
+            'model_data' => ['id' => 1]
+        ];
+
+        $this->customerFactory
+            ->method('create')
+            ->willReturn($customer);
+        $this->hydratorPool
+            ->method('getHydrator')
+            ->willReturn($this->hydrator);
+        $this->hydrator->expects($this->once())
+            ->method('hydrate')
+            ->with($customer, $resolverData['model_data']);
+
+        $this->modelHydrator->hydrate($resolverData);
+        $this->assertSame($customer, $resolverData['model']);
+    }
+
+    /**
+     * Test that resetState method resets the state of the modelHydrator
+     *
+     * @return void
+     * @throws Exception
+     */
+    public function testResetState()
+    {
+        $customer1 = $this->createMock(Customer::class);
+        $customer2 = $this->createMock(Customer::class);
+
+        $resolverData1 = [
+            'model_id' => 1,
+            'model_entity_type' => 'customer',
+            'model_data' => ['id' => 1]
+        ];
+
+        $resolverData2 = [
+            'model_id' => 2,
+            'model_entity_type' => 'customer',
+            'model_data' => ['id' => 2]
+        ];
+
+        $this->customerFactory
+            ->method('create')
+            ->willReturnOnConsecutiveCalls($customer1, $customer2);
+        $this->hydratorPool
+            ->method('getHydrator')
+            ->willReturn($this->hydrator);
+
+        $matcher = $this->exactly(2);
+        $expected1 = $resolverData1['model_data'];
+        $expected2 = $resolverData2['model_data'];
+
+        $this->hydrator
+            ->expects($matcher)
+            ->method('hydrate')
+            ->willReturnCallback(function ($model, $data) use ($matcher, $expected1, $expected2) {
+                match ($matcher->numberOfInvocations()) {
+                    1 => $this->assertEquals($expected1, $data),
+                    2 => $this->assertEquals($expected2, $data),
+                };
+            });
+
+        $this->modelHydrator->hydrate($resolverData1);
+
+        $this->assertArrayHasKey('model', $resolverData1);
+        $this->assertEquals(1, $resolverData1['model_id']);
+
+        $this->modelHydrator->_resetState();
+
+        $this->modelHydrator->hydrate($resolverData2);
+
+        $this->assertArrayHasKey('model', $resolverData2);
+        $this->assertEquals(2, $resolverData2['model_id']);
+    }
+}