/** * Test default address attribute mapping array */ public function testGetDefaultAddressAttributeMapping() { $attributeMapping = $this->_model->getDefaultAddressAttributeMapping(); $this->assertInternalType('array', $attributeMapping, 'Default address attribute mapping must be an array.'); $this->assertArrayHasKey(Address::COLUMN_DEFAULT_BILLING, $attributeMapping, 'Default address attribute mapping array must have a default billing column.'); $this->assertArrayHasKey(Address::COLUMN_DEFAULT_SHIPPING, $attributeMapping, 'Default address attribute mapping array must have a default shipping column.'); }
/** * Test export method */ public function testExport() { $websiteCode = Address::COLUMN_WEBSITE; $emailCode = Address::COLUMN_EMAIL; $entityIdCode = Address::COLUMN_ADDRESS_ID; $expectedAttributes = array(); /** @var $collection \Magento\Customer\Model\Resource\Address\Attribute\Collection */ $collection = \Magento\TestFramework\Helper\Bootstrap::getObjectManager()->create('Magento\\Customer\\Model\\Resource\\Address\\Attribute\\Collection'); /** @var $attribute \Magento\Customer\Model\Attribute */ foreach ($collection as $attribute) { $expectedAttributes[] = $attribute->getAttributeCode(); } // Get customer default addresses column name to customer attribute mapping array. $defaultAddressMap = ImportAddress::getDefaultAddressAttributeMapping(); $this->_model->setWriter(\Magento\TestFramework\Helper\Bootstrap::getObjectManager()->create('Magento\\ImportExport\\Model\\Export\\Adapter\\Csv')); $this->_model->setParameters(array()); $data = $this->_csvToArray($this->_model->export(), $entityIdCode); $this->assertEquals(count($expectedAttributes), count(array_intersect($expectedAttributes, $data['header'])), 'Expected attribute codes were not exported'); $this->assertNotEmpty($data['data'], 'No data was exported'); /** @var $objectManager \Magento\TestFramework\ObjectManager */ $objectManager = \Magento\TestFramework\Helper\Bootstrap::getObjectManager(); // Get addresses /** @var $customers \Magento\Customer\Model\Customer[] */ $customers = $objectManager->get('Magento\\Framework\\Registry')->registry('_fixture/Magento_ImportExport_Customers_Array'); foreach ($customers as $customer) { /** @var $address \Magento\Customer\Model\Address */ foreach ($customer->getAddresses() as $address) { // Check unique key $data['data'][$address->getId()][$websiteCode] = $this->_websites[$customer->getWebsiteId()]; $data['data'][$address->getId()][$emailCode] = $customer->getEmail(); $data['data'][$address->getId()][$entityIdCode] = $address->getId(); // Check by expected attributes foreach ($expectedAttributes as $code) { if (!in_array($code, $this->_model->getDisabledAttributes())) { $this->assertEquals($address->getData($code), $data['data'][$address->getId()][$code], 'Attribute "' . $code . '" is not equal'); } } // Check customer default addresses column name to customer attribute mapping array foreach ($defaultAddressMap as $exportCode => $code) { $this->assertEquals($address->getData($code), (int) $data['data'][$address->getId()][$exportCode], 'Attribute "' . $code . '" is not equal'); } } } }
/** * Test _saveCustomerDefaults * * @magentoDataFixture Magento/Customer/_files/import_export/customer_with_addresses.php */ public function testSaveCustomerDefaults() { /** @var $objectManager \Magento\TestFramework\ObjectManager */ $objectManager = \Magento\TestFramework\Helper\Bootstrap::getObjectManager(); // get not default address $customers = $objectManager->get('Magento\\Framework\\Registry')->registry($this->_fixtureKey); /** @var $notDefaultAddress \Magento\Customer\Model\Address */ $notDefaultAddress = null; /** @var $addressCustomer \Magento\Customer\Model\Customer */ $addressCustomer = null; /** @var $customer \Magento\Customer\Model\Customer */ foreach ($customers as $customer) { /** @var $address \Magento\Customer\Model\Address */ foreach ($customer->getAddressesCollection() as $address) { if (!$customer->getDefaultBillingAddress() && !$customer->getDefaultShippingAddress()) { $notDefaultAddress = $address; $addressCustomer = $customer; break; } if ($notDefaultAddress) { break; } } } $this->assertNotNull($notDefaultAddress, 'Not default address must exists.'); $this->assertNotNull($addressCustomer, 'Not default address customer must exists.'); $addressId = $notDefaultAddress->getId(); $customerId = $addressCustomer->getId(); // set customer defaults $defaults = []; foreach (Address::getDefaultAddressAttributeMapping() as $attributeCode) { /** @var $attribute \Magento\Eav\Model\Entity\Attribute\AbstractAttribute */ $attribute = $addressCustomer->getAttribute($attributeCode); $attributeTable = $attribute->getBackend()->getTable(); $attributeId = $attribute->getId(); $defaults[$attributeTable][$customerId][$attributeId] = $addressId; } // invoke _saveCustomerDefaults $saveDefaults = new \ReflectionMethod($this->_testClassName, '_saveCustomerDefaults'); $saveDefaults->setAccessible(true); $saveDefaults->invoke($this->_entityAdapter, $defaults); // check DB /** @var $testCustomer \Magento\Customer\Model\Customer */ $testCustomer = \Magento\TestFramework\Helper\Bootstrap::getObjectManager()->create('Magento\\Customer\\Model\\Customer'); $testCustomer->load($customerId); $this->assertEquals($customerId, $testCustomer->getId(), 'Customer must exists.'); $this->assertNotNull($testCustomer->getDefaultBillingAddress(), 'Default billing address must exists.'); $this->assertNotNull($testCustomer->getDefaultShippingAddress(), 'Default shipping address must exists.'); $this->assertEquals($addressId, $testCustomer->getDefaultBillingAddress()->getId(), 'Incorrect default billing address.'); $this->assertEquals($addressId, $testCustomer->getDefaultShippingAddress()->getId(), 'Incorrect default shipping address.'); }