AssertExportCustomerAddresses.php

Go to the documentation of this file.

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

use Magento\Customer\Test\Fixture\Address;
use Magento\Customer\Test\Fixture\Customer;
use Magento\Mtf\Constraint\AbstractConstraint;
use Magento\Mtf\Util\Command\File\Export;
use Magento\Mtf\Util\Command\File\Export\Data;

class AssertExportCustomerAddresses extends AbstractConstraint
{
    public function processAssert(
        Export $export,
        array $exportedFields,
        Customer $customer
    ) {
        $exportData = $export->getLatest();

        foreach ($customer->getDataFieldConfig('address')['source']->getAddresses() as $address) {
            \PHPUnit\Framework\Assert::assertTrue(
                $this->isAddressDataInFile(
                    $exportedFields,
                    $customer,
                    $address,
                    $exportData
                ),
                'Customer address was not found in exported file.'
            );
        }
    }

    public function toString()
    {
        return 'Customer address exists in exported file.';
    }

    private function isAddressDataInFile(
        array $fields,
        Customer $customer,
        Address $address,
        Data $exportData
    ) {
        $expectedFields = [];
        $result = true;

        foreach ($fields as $field) {
            $fixture = ($field == 'email') ? $customer : $address;
            $expectedFields[] = $fixture->getData($field);
        }

        $content = $exportData->getContent();

        foreach ($expectedFields as $expectedField) {
            if (strpos($content, $expectedField) === false) {
                $result = false;
                break;
            }
        }

        return $result;
    }
}