[InstantPurchase] Cover Ui/CustomerAddressesFormatter and Ui/ShippingMethodFormatter by Unit Test

Author: edenduong

app/code/Magento/InstantPurchase/Test/Unit/Model/Ui/CustomerAddressesFormatterTest.php

@@ -0,0 +1,56 @@
+<?php
+/**
+ * Copyright © Magento, Inc. All rights reserved.
+ * See COPYING.txt for license details.
+ */
+declare(strict_types=1);
+
+namespace Magento\InstantPurchase\Test\Unit\Model\Ui;
+
+use Magento\InstantPurchase\Model\Ui\CustomerAddressesFormatter;
+use Magento\Customer\Model\Address;
+use Magento\Framework\TestFramework\Unit\Helper\ObjectManager as ObjectManagerHelper;
+use Magento\Directory\Model\Country;
+use PHPUnit\Framework\TestCase;
+
+class CustomerAddressesFormatterTest extends TestCase
+{
+    /**
+     * @var CustomerAddressesFormatter|\PHPUnit_Framework_MockObject_MockObject
+     */
+    private $customerAddressesFormatter;
+
+    /**
+     * Setup environment for testing
+     */
+    protected function setUp()
+    {
+        $objectManager = new ObjectManagerHelper($this);
+        $this->customerAddressesFormatter = $objectManager->getObject(CustomerAddressesFormatter::class);
+    }
+
+    /**
+     * Test format()
+     */
+    public function testFormat()
+    {
+        $addressMock = $this->createPartialMock(
+            Address::class,
+            ['getName', 'getStreetFull', 'getCity', 'getRegion', 'getPostcode', 'getCountryModel']
+        );
+        $countryMock = $this->createMock(Country::class);
+
+        $countryMock->expects($this->any())->method('getName')->willReturn('USA');
+        $addressMock->expects($this->any())->method('getName')->willReturn('Address Name');
+        $addressMock->expects($this->any())->method('getStreetFull')->willReturn('Address Street Full');
+        $addressMock->expects($this->any())->method('getCity')->willReturn('Address City');
+        $addressMock->expects($this->any())->method('getRegion')->willReturn('Address Region');
+        $addressMock->expects($this->any())->method('getPostcode')->willReturn('Address Postcode');
+        $addressMock->expects($this->any())->method('getCountryModel')->willReturn($countryMock);
+
+        $this->assertEquals(
+            'Address Name, Address Street Full, Address City, Address Region Address Postcode, USA',
+            $this->customerAddressesFormatter->format($addressMock)
+        );
+    }
+}

app/code/Magento/InstantPurchase/Test/Unit/Model/Ui/ShippingMethodFormatterTest.php

@@ -0,0 +1,46 @@
+<?php
+/**
+ * Copyright © Magento, Inc. All rights reserved.
+ * See COPYING.txt for license details.
+ */
+declare(strict_types=1);
+
+namespace Magento\InstantPurchase\Test\Unit\Model\Ui;
+
+use Magento\InstantPurchase\Model\Ui\ShippingMethodFormatter;
+use Magento\Quote\Api\Data\ShippingMethodInterface;
+use Magento\Framework\TestFramework\Unit\Helper\ObjectManager as ObjectManagerHelper;
+use PHPUnit\Framework\TestCase;
+
+class ShippingMethodFormatterTest extends TestCase
+{
+    /**
+     * @var ShippingMethodFormatter|\PHPUnit_Framework_MockObject_MockObject
+     */
+    private $shippingMethodFormatter;
+
+    /**
+     * Setup environment for testing
+     */
+    protected function setUp()
+    {
+        $objectManager = new ObjectManagerHelper($this);
+        $this->shippingMethodFormatter = $objectManager->getObject(ShippingMethodFormatter::class);
+    }
+
+    /**
+     * Test format()
+     */
+    public function testFormat()
+    {
+        $shippingMethodMock = $this->createMock(ShippingMethodInterface::class, ['getCarrierTitle', 'getMethodTitle']);
+
+        $shippingMethodMock->expects($this->any())->method('getCarrierTitle')->willReturn('flatrate');
+        $shippingMethodMock->expects($this->any())->method('getMethodTitle')->willReturn('flatrate');
+
+        $this->assertEquals(
+            'flatrate - flatrate',
+            $this->shippingMethodFormatter->format($shippingMethodMock)
+        );
+    }
+}