ENGCOM-6446: [Customer] Cover CustomerData\Customer and CustomerData\JsLayoutDataProviderPool by Unit Test #25984

Author: VladimirZaets

app/code/Magento/Customer/Test/Unit/CustomerData/CustomerTest.php

@@ -0,0 +1,90 @@
+<?php
+/**
+ * Copyright © Magento, Inc. All rights reserved.
+ * See COPYING.txt for license details.
+ */
+
+declare(strict_types=1);
+
+namespace Magento\Customer\Test\Unit\CustomerData;
+
+use Magento\Framework\TestFramework\Unit\Helper\ObjectManager as ObjectManagerHelper;
+use Magento\Customer\CustomerData\Customer as CustomerData;
+use Magento\Customer\Helper\Session\CurrentCustomer;
+use Magento\Customer\Helper\View;
+use Magento\Customer\Api\Data\CustomerInterface;
+use PHPUnit\Framework\TestCase;
+
+class CustomerTest extends TestCase
+{
+    /**
+     * @var ObjectManagerHelper
+     */
+    private $objectManagerHelper;
+
+    /**
+     * @var CustomerData
+     */
+    private $customerData;
+
+    /**
+     * @var CurrentCustomer
+     */
+    private $currentCustomerMock;
+
+    /**
+     * @var View
+     */
+    private $customerViewHelperMock;
+
+    /**
+     * Setup environment to test
+     */
+    protected function setUp()
+    {
+        $this->currentCustomerMock = $this->createMock(CurrentCustomer::class);
+        $this->customerViewHelperMock = $this->createMock(View::class);
+        $this->objectManagerHelper = new ObjectManagerHelper($this);
+
+        $this->customerData = $this->objectManagerHelper->getObject(
+            CustomerData::class,
+            [
+                'currentCustomer' => $this->currentCustomerMock,
+                'customerViewHelper' => $this->customerViewHelperMock
+            ]
+        );
+    }
+
+    /**
+     * Test getSectionData() without customer Id
+     */
+    public function testGetSectionDataWithoutCustomerId()
+    {
+        $this->currentCustomerMock->expects($this->any())->method('getCustomerId')->willReturn(null);
+        $this->assertEquals([], $this->customerData->getSectionData());
+    }
+
+    /**
+     * Test getSectionData() with customer
+     */
+    public function testGetSectionDataWithCustomer()
+    {
+        $this->currentCustomerMock->expects($this->any())->method('getCustomerId')->willReturn(1);
+        $customerMock = $this->createMock(CustomerInterface::class);
+        $customerMock->expects($this->any())->method('getFirstname')->willReturn('John');
+        $customerMock->expects($this->any())->method('getWebsiteId')->willReturn(1);
+        $this->currentCustomerMock->expects($this->any())->method('getCustomer')->willReturn($customerMock);
+        $this->customerViewHelperMock->expects($this->any())->method('getCustomerName')
+            ->with($customerMock)
+            ->willReturn('John Adam');
+
+        $this->assertEquals(
+            [
+                'fullname' => 'John Adam',
+                'firstname' => 'John',
+                'websiteId' => 1,
+            ],
+            $this->customerData->getSectionData()
+        );
+    }
+}

app/code/Magento/Customer/Test/Unit/CustomerData/JsLayoutDataProviderPoolTest.php

@@ -0,0 +1,84 @@
+<?php
+/**
+ * Copyright © Magento, Inc. All rights reserved.
+ * See COPYING.txt for license details.
+ */
+
+declare(strict_types=1);
+
+namespace Magento\Customer\Test\Unit\CustomerData;
+
+use Magento\Framework\TestFramework\Unit\Helper\ObjectManager as ObjectManagerHelper;
+use Magento\Tax\CustomerData\CheckoutTotalsJsLayoutDataProvider as CheckoutTotalsJs;
+use Magento\Customer\CustomerData\JsLayoutDataProviderPool;
+use PHPUnit\Framework\TestCase;
+
+class JsLayoutDataProviderPoolTest extends TestCase
+{
+    /**
+     * @var ObjectManagerHelper
+     */
+    private $objectManagerHelper;
+
+    /**
+     * @var CheckoutTotalsJs
+     */
+    private $checkoutTotalsJsLayoutDataProviderMock;
+
+    /**
+     * @var JsLayoutDataProviderPool
+     */
+    private $jsLayoutDataProviderPool;
+
+    /**
+     * Setup environment to test
+     */
+    protected function setUp()
+    {
+        $this->checkoutTotalsJsLayoutDataProviderMock = $this->createMock(CheckoutTotalsJs::class);
+        $this->objectManagerHelper = new ObjectManagerHelper($this);
+
+        $this->jsLayoutDataProviderPool = $this->objectManagerHelper->getObject(
+            JsLayoutDataProviderPool::class,
+            [
+                'jsLayoutDataProviders' => [
+                    'checkout_totals' => $this->checkoutTotalsJsLayoutDataProviderMock
+                ]
+            ]
+        );
+    }
+
+    /**
+     * Test getData() function
+     */
+    public function testGetData()
+    {
+        $checkoutTotalsJsData = [
+            'components' => [
+                'minicart_content' => [
+                    'children' => [
+                        'subtotal.container' => [
+                            'children' => [
+                                'subtotal' => [
+                                    'children' => [
+                                        'subtotal.totals' => [
+                                            'config' => [
+                                                'display_cart_subtotal_incl_tax' => 1,
+                                                'display_cart_subtotal_excl_tax' => 1
+                                            ]
+                                        ],
+                                    ],
+                                ],
+                            ],
+                        ],
+                    ],
+                ],
+            ]
+        ];
+        $this->checkoutTotalsJsLayoutDataProviderMock->expects($this->any())
+            ->method('getData')
+            ->willReturn($checkoutTotalsJsData);
+
+        $this->assertEquals($checkoutTotalsJsData, $this->jsLayoutDataProviderPool->getData());
+    }
+}