MAGETWO-90358: Auto-generate ExtensionAttributes object

Author: Alex Paliarush

dev/tests/integration/testsuite/Magento/Customer/Api/AddressRepositoryTest.php

@@ -161,9 +161,11 @@ public function testSaveNewAddress()
         $expectedNewAddress->setId($savedAddress->getId());
         $expectedNewAddress->setRegion($this->_expectedAddresses[1]->getRegion());
 
-        // Call getters on extension attributes to initialize them
-        $savedAddress->getExtensionAttributes();
-        $savedAddress->getRegion()->getExtensionAttributes();
+        $this->assertEquals($expectedNewAddress->getExtensionAttributes(), $savedAddress->getExtensionAttributes());
+        $this->assertEquals(
+            $expectedNewAddress->getRegion()->getExtensionAttributes(),
+            $savedAddress->getRegion()->getExtensionAttributes()
+        );
 
         $this->assertEquals($expectedNewAddress, $savedAddress);
     }

dev/tests/integration/testsuite/Magento/Customer/Block/Adminhtml/Edit/Tab/View/PersonalInfoTest.php

@@ -121,8 +121,7 @@ public function testGetCustomerEmpty()
     {
         $expectedCustomer = $this->createCustomerAndAddToBackendSession();
         $actualCustomer = $this->_block->getCustomer();
-        // Trigger extension attributes generation
-        $actualCustomer->getExtensionAttributes();
+        $this->assertEquals($expectedCustomer->getExtensionAttributes(), $actualCustomer->getExtensionAttributes());
         $this->assertEquals($expectedCustomer, $actualCustomer);
     }
 

dev/tests/integration/testsuite/Magento/Customer/Model/ResourceModel/AddressRepositoryTest.php

@@ -167,9 +167,11 @@ public function testSaveNewAddress()
         $expectedNewAddress->setId($savedAddress->getId());
         $expectedNewAddress->setRegion($this->_expectedAddresses[1]->getRegion());
 
-        // Trigger extension attributes object generation
-        $savedAddress->getExtensionAttributes();
-        $savedAddress->getRegion()->getExtensionAttributes();
+        $this->assertEquals($expectedNewAddress->getExtensionAttributes(), $savedAddress->getExtensionAttributes());
+        $this->assertEquals(
+            $expectedNewAddress->getRegion()->getExtensionAttributes(),
+            $savedAddress->getRegion()->getExtensionAttributes()
+        );
 
         $this->assertEquals($expectedNewAddress, $savedAddress);
     }