Fix Magento\Sales\Service\V1\ShipmentGetTest::testShipmentGet

magento · Jun 10, 2018 · 9e5757d · 9e5757d
1 parent 33393e2
commit 9e5757d
Showing 1 changed file with 14 additions and 1 deletion.
diff --git a/dev/tests/api-functional/testsuite/Magento/Sales/Service/V1/ShipmentGetTest.php b/dev/tests/api-functional/testsuite/Magento/Sales/Service/V1/ShipmentGetTest.php
@@ -5,6 +5,8 @@
  */
 namespace Magento\Sales\Service\V1;
 
+use Magento\Framework\Api\ExtensibleDataInterface;
+use Magento\Framework\Api\SimpleDataObjectConverter;
 use Magento\TestFramework\TestCase\WebapiAbstract;
 
 /**
@@ -57,7 +59,18 @@ public function testShipmentGet()
         unset($data['tracks']);
         foreach ($data as $key => $value) {
             if (!empty($value)) {
-                $this->assertEquals($shipment->getData($key), $value, $key);
+                if ($key === ExtensibleDataInterface::EXTENSION_ATTRIBUTES_KEY) {
+                    foreach ($value as $extensionAttributeKey => $extensionAttributeValue) {
+                        $methodName = 'get' .
+                            SimpleDataObjectConverter::snakeCaseToUpperCamelCase($extensionAttributeKey);
+                        $this->assertEquals(
+                            $shipment->getExtensionAttributes()->$methodName(),
+                            $extensionAttributeValue
+                        );
+                    }
+                } else {
+                    $this->assertEquals($shipment->getData($key), $value, $key);
+                }
             }
         }
         $shipmentItem = $this->objectManager->get(\Magento\Sales\Model\Order\Shipment\Item::class);