AssertImportCustomerAddresses.php

Go to the documentation of this file.

<?php
namespace Magento\CustomerImportExport\Test\Constraint;

use Magento\Customer\Test\Page\Adminhtml\CustomerIndexEdit;
use Magento\ImportExport\Test\Fixture\ImportData;
use Magento\Mtf\Constraint\AbstractConstraint;
use Magento\Mtf\Fixture\FixtureFactory;

class AssertImportCustomerAddresses extends AbstractConstraint
{
    private $mappingKeys = [
        'region' => 'region_id'
    ];

    private $importedFieldsKeys = [
        'firstname',
        'lastname',
        'postcode',
        'region_id',
        'city',
        'company',
        'country_id',
        'street',
        'telephone'
    ];

    private $mappingCountries = [
        'US' => 'United States'
    ];

    private $customerIndexEdit;

    private $fixtureFactory;

    private $import;

    public function processAssert(
        CustomerIndexEdit $customerIndexEdit,
        FixtureFactory $fixtureFactory,
        ImportData $import
    ) {
        $this->customerIndexEdit = $customerIndexEdit;
        $this->fixtureFactory = $fixtureFactory;
        $this->import = $import;

        $resultArrays = $this->getPrepareAddresses();

        \PHPUnit\Framework\Assert::assertEquals(
            $resultArrays['pageData'],
            $resultArrays['csvData'],
            'Addresses from page and csv are not match.'
        );
    }

    private function getPrepareAddresses()
    {
        $addressTemplate = ($this->import->getBehavior() !== 'Delete Entities')
            ? $this->fixtureFactory->createByCode('address', ['dataset' => 'US_address_1_without_email'])
            : null;
        $customers = $this->import->getDataFieldConfig('import_file')['source']->getEntities();
        $customerForm = $this->customerIndexEdit->getCustomerForm();

        // Prepare customer address data from page form.
        $resultAddressesArray = [];
        foreach ($customers as $customer) {
            $this->customerIndexEdit->open(['id' => $customer->getId()]);
            $customerForm->openTab('addresses');
            $addresses = $customerForm->getTab('addresses')->getDataAddresses($addressTemplate);
            $address = array_shift($addresses);
            if (!empty($address)) {
                $resultAddressesArray[] = $address;
            }
        }

        // Prepare customer address data from csv file.
        $resultCsvArray = [];
        if ($this->import->getBehavior() !== 'Delete Entities') {
            $resultCsvArray = $this->getResultCsv();
        }
        return ['pageData' => $resultAddressesArray, 'csvData' => $resultCsvArray];
    }

    private function getResultCsv()
    {
        $csvData = $this->import->getDataFieldConfig('import_file')['source']->getCsv();

        $csvKeys = [];
        foreach (array_shift($csvData) as $csvKey) {
            $csvKeys[] = isset($this->mappingKeys[$csvKey]) ? $this->mappingKeys[$csvKey] : $csvKey;
        }

        $resultCsvData = [];
        foreach ($csvData as $csvRowData) {
            $csvRowData = array_combine($csvKeys, $csvRowData);
            $csvRowData = $this->deleteWasteData($csvRowData);
            if (isset($this->mappingCountries[$csvRowData['country_id']])) {
                $csvRowData['country_id'] = $this->mappingCountries[$csvRowData['country_id']];
            }
            $resultCsvData[] = $csvRowData;
        }
        return $resultCsvData;
    }

    private function deleteWasteData(array $csvData)
    {
        $necessaryData = array_flip($this->importedFieldsKeys);

        return array_intersect_key($csvData, $necessaryData);
    }

    public function toString()
    {
        return 'Imported customer addresses are correct.';
    }
}