From 748fd787c71362600d187a82e5e698428750f731 Mon Sep 17 00:00:00 2001
From: Maksym Novik <novik.kor@gmail.com>
Date: Sat, 18 May 2019 19:06:22 +0300
Subject: [PATCH] Source Entity expansion with Pickup Location attributes
 #2182. Improved tests coverage

---
 .../InventorySourceExtensionTest.php          | 31 ++++++++++++-------
 1 file changed, 20 insertions(+), 11 deletions(-)

diff --git a/app/code/Magento/InventoryInStorePickup/Test/Integration/Extension/InventorySourceExtensionTest.php b/app/code/Magento/InventoryInStorePickup/Test/Integration/Extension/InventorySourceExtensionTest.php
index af2ef3f42817..d6adcab7b002 100644
--- a/app/code/Magento/InventoryInStorePickup/Test/Integration/Extension/InventorySourceExtensionTest.php
+++ b/app/code/Magento/InventoryInStorePickup/Test/Integration/Extension/InventorySourceExtensionTest.php
@@ -34,21 +34,22 @@ protected function setUp()
     public function testGetListOfSourcesWithPickupLocationExtensionAfterSave()
     {
         $pickupLocationConfig = [
-            'default' => false,
-            'eu-1' => true,
-            'eu-2' => true,
-            'eu-3' => false,
-            'eu-disabled' => false,
-            'us-1' => true,
+            'default' => ['active' => false, 'name' => 'default', 'desc' => 'default'],
+            'eu-1' => ['active' => true, 'name' => '', 'desc' => ''],
+            'eu-2' => ['active' => true, 'name' => 'zzz', 'desc' => ''],
+            'eu-3' => ['active' => false, 'name' => '', 'desc' => 'zzz1'],
+            'eu-disabled' => ['active' => false, 'name' => '', 'desc' => ''],
+            'us-1' => ['active' => true, 'name' => '666', 'desc' => ''],
         ];
 
         $searchResult = $this->sourceRepository->getList();
 
         /** @var SourceInterface $item */
         foreach ($searchResult->getItems() as $item) {
-            $item->getExtensionAttributes()->setIsPickupLocationActive(
-                $pickupLocationConfig[$item->getSourceCode()]
-            );
+            $item->getExtensionAttributes()
+                 ->setIsPickupLocationActive($pickupLocationConfig[$item->getSourceCode()]['active'])
+                 ->setFrontendDescription($pickupLocationConfig[$item->getSourceCode()]['desc'])
+                 ->setFrontendName($pickupLocationConfig[$item->getSourceCode()]['name']);
             $this->sourceRepository->save($item);
         }
 
@@ -57,7 +58,10 @@ public function testGetListOfSourcesWithPickupLocationExtensionAfterSave()
         $pickupLocationsStatus = [];
 
         foreach ($searchResult->getItems() as $item) {
-            $pickupLocationsStatus[$item->getSourceCode()] = $item->getExtensionAttributes()->getIsPickupLocationActive();
+            $extension = $item->getExtensionAttributes();
+            $pickupLocationsStatus[$item->getSourceCode()]['active'] = $extension->getIsPickupLocationActive();
+            $pickupLocationsStatus[$item->getSourceCode()]['name'] = $extension->getFrontendName();
+            $pickupLocationsStatus[$item->getSourceCode()]['desc'] = $extension->getFrontendDescription();
         }
 
         $this->assertEquals($pickupLocationConfig, $pickupLocationsStatus);
@@ -71,10 +75,15 @@ public function testGetSourceWithPickupLocationExtensionAfterSave()
         $sourceCode = 'source-code-1';
 
         $source = $this->sourceRepository->get($sourceCode);
-        $source->getExtensionAttributes()->setIsPickupLocationActive(true);
+        $source->getExtensionAttributes()
+               ->setIsPickupLocationActive(true)
+               ->setFrontendName('zzz')
+               ->setFrontendDescription('666');
         $this->sourceRepository->save($source);
 
         $source = $this->sourceRepository->get($sourceCode);
         $this->assertEquals(true, $source->getExtensionAttributes()->getIsPickupLocationActive());
+        $this->assertEquals('zzz', $source->getExtensionAttributes()->getFrontendName());
+        $this->assertEquals('666', $source->getExtensionAttributes()->getFrontendDescription());
     }
 }