ProductAttributesCleanUpTest.php

Go to the documentation of this file.

<?php
namespace Magento\Catalog\Console\Command;

use Symfony\Component\Console\Tester\CommandTester;

class ProductAttributesCleanUpTest extends \PHPUnit\Framework\TestCase
{
    private $tester;

    private $command;

    private $objectManager;

    private $attributeResource;

    public function setUp()
    {
        $this->objectManager = \Magento\TestFramework\Helper\Bootstrap::getObjectManager();
        $this->command = $this->objectManager->create(\Magento\Catalog\Console\Command\ProductAttributesCleanUp::class);
        $this->attributeResource = $this->objectManager->create(\Magento\Catalog\Model\ResourceModel\Attribute::class);
        $this->tester = new CommandTester($this->command);

        // Prepare data fixtures for test
        $store = $this->prepareAdditionalStore();
        $productRepository = $this->objectManager->create(\Magento\Catalog\Api\ProductRepositoryInterface::class);
        $product = $productRepository->get('simple');
        $product->setName('Simple fixture store');
        $product->setStoreId($store->getId());
        $product->save();
    }

    public function testExecute()
    {
        // Verify that unused attribute was created
        $attribute = $this->getUnusedProductAttribute();

        $this->assertArrayHasKey('value', $attribute);
        $this->assertArrayHasKey('value_id', $attribute);
        $this->assertEquals($attribute['value'], 'Simple fixture store');

        // Execute command
        $this->tester->execute([]);

        // Verify that unused attribute was removed
        $this->assertContains('Unused product attributes successfully cleaned up', $this->tester->getDisplay());
        $attribute = $this->getUnusedProductAttribute();
        $this->assertFalse($attribute);
    }

    private function getUnusedProductAttribute()
    {
        $connection = $this->attributeResource->getConnection();
        $select = $connection->select();
        $select->from($this->attributeResource->getTable('catalog_product_entity_varchar'));
        $select->where('value = ?', 'Simple fixture store');
        return $connection->fetchRow($select);
    }

    private function prepareAdditionalStore()
    {
        $website = $this->objectManager->create(\Magento\Store\Model\Website::class);
        $website->load('test');

        $store = $this->objectManager->create(\Magento\Store\Model\Store::class);
        $store->load('fixturestore');

        $storeGroup = $this->objectManager->create(\Magento\Store\Model\Group::class);
        $storeGroup->setWebsiteId($website->getId());
        $storeGroup->setName('Fixture Store Group');
        $storeGroup->setRootCategoryId(2);
        $storeGroup->setDefaultStoreId($store->getId());
        $storeGroup->save();

        $store->setWebsiteId($website->getId())
            ->setGroupId($storeGroup->getId())
            ->save();

        return $store;
    }
}