newsupernova/dev/tests/integration/testsuite/Magento/Eav/Model/ResourceModel/AttributeLoaderTest.php

<?php
/**
 * Copyright © Magento, Inc. All rights reserved.
 * See COPYING.txt for license details.
 */

namespace Magento\Eav\Model\ResourceModel;

use Magento\Framework\EntityManager\MetadataPool;
use Magento\TestFramework\Helper\Bootstrap;

/**
 * @magentoAppIsolation enabled
 * @magentoDbIsolation enabled
 * @magentoDataFixture Magento/Eav/_files/attribute_for_search.php
 */
class AttributeLoaderTest extends \Magento\TestFramework\Indexer\TestCase
{
    /**
     * @var AttributeLoader
     */
    private $attributeLoader;

    protected function setUp()
    {
        $objectManager = Bootstrap::getObjectManager();
        $metadataPool = $objectManager->create(
            MetadataPool::class,
            [
                'metadata' => [
                    'Test\Entity\Type' => [
                        'entityTableName' => 'test_entity',
                        'eavEntityType' => 'test',
                        'identifierField' => 'entity_id',
                    ]
                ]
            ]
        );
        $this->attributeLoader = $objectManager->create(AttributeLoader::class, ['metadataPool' => $metadataPool]);
    }

    /**
     * @param string[] $expectedAttributeCodes
     * @param int|null $attributeSetId
     * @dataProvider getAttributesDataProvider
     */
    public function testGetAttributes($expectedAttributeCodes, $attributeSetId = null)
    {
        $attributes = $this->attributeLoader->getAttributes('Test\Entity\Type', $attributeSetId);
        $this->assertEquals(count($expectedAttributeCodes), count($attributes));
        $attributeCodes = [];
        foreach ($attributes as $attribute) {
            $attributeCodes[] = $attribute->getAttributeCode();
        }
        $this->assertEquals($expectedAttributeCodes, $attributeCodes);
        $attributes2 = $this->attributeLoader->getAttributes('Test\Entity\Type', $attributeSetId);
        $this->assertEquals($attributes, $attributes2);
    }

    public function getAttributesDataProvider()
    {
        $objectManager = \Magento\TestFramework\Helper\Bootstrap::getObjectManager();
        $entityType = $objectManager->create(\Magento\Eav\Model\Entity\Type::class)
            ->loadByCode('order');
        $attributeSetId = $entityType->getDefaultAttributeSetId();

        return [
            'all' => [
                [
                    'attribute_for_search_1',
                    'attribute_for_search_2',
                    'attribute_for_search_3',
                ]
            ],
            "$attributeSetId" => [
                [
                    'attribute_for_search_1',
                    'attribute_for_search_2',
                ],
                $attributeSetId,
            ]
        ];
    }
}