/vendor/magento/module-eav/Test/Unit/Model/ResourceModel/Entity/Attribute/SetTest.php

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

namespace Magento\Eav\Test\Unit\Model\ResourceModel\Entity\Attribute;

use Magento\Eav\Model\ResourceModel\Entity\Attribute\Set;
 
class SetTest extends \PHPUnit_Framework_TestCase
{
    /**
     * @var \PHPUnit_Framework_MockObject_MockObject|Set
     */
    protected $model;

    /**
     * @var \PHPUnit_Framework_MockObject_MockObject
     */
    protected $eavConfigMock;

    /**
     * @var \PHPUnit_Framework_MockObject_MockObject
     */
    protected $objectMock;

    /**
     * @var \PHPUnit_Framework_MockObject_MockObject
     */
    protected $typeMock;

    /**
     * @var \PHPUnit_Framework_MockObject_MockObject
     */
    protected $transactionManagerMock;

    /**
     * @var \PHPUnit_Framework_MockObject_MockObject
     */
    protected $resourceMock;

    /**
     * @var \PHPUnit_Framework_MockObject_MockObject
     */
    protected $relationProcessor;

    /**
     * {@inheritdoc}
     */
    protected function setUp()
    {
        $this->resourceMock = $this->getMockBuilder('Magento\Framework\App\ResourceConnection')
            ->disableOriginalConstructor()
            ->setMethods(['getConnection', 'getTableName'])
            ->getMock();
        $this->transactionManagerMock = $this->getMock(
            '\Magento\Framework\Model\ResourceModel\Db\TransactionManagerInterface'
        );
        $this->relationProcessor = $this->getMock(
            '\Magento\Framework\Model\ResourceModel\Db\ObjectRelationProcessor',
            [],
            [],
            '',
            false
        );
        $contextMock = $this->getMock('Magento\Framework\Model\ResourceModel\Db\Context', [], [], '', false);
        $contextMock->expects($this->once())
            ->method('getTransactionManager')
            ->willReturn($this->transactionManagerMock);
        $contextMock->expects($this->once())
            ->method('getObjectRelationProcessor')
            ->willReturn($this->relationProcessor);
        $contextMock->expects($this->once())->method('getResources')->willReturn($this->resourceMock);

        $this->eavConfigMock = $this->getMockBuilder('Magento\Eav\Model\Config')
            ->setMethods(['isCacheEnabled', 'getEntityType', 'getCache'])
            ->disableOriginalConstructor()
            ->getMock();
        $this->model = $this->getMock(
            'Magento\Eav\Model\ResourceModel\Entity\Attribute\Set',
            [
                'beginTransaction',
                'getMainTable',
                'getIdFieldName',
                '_afterDelete',
                'commit',
                'rollBack',
                '__wakeup'
            ],
            [
                $contextMock,
                $this->getMock('Magento\Eav\Model\ResourceModel\Entity\Attribute\GroupFactory', [], [], '', false),
                $this->eavConfigMock
            ],
            '',
            true
        );
        $this->typeMock = $this->getMock('\Magento\Eav\Model\Entity\Type', [], [], '', false);
        $this->objectMock = $this->getMock(
            'Magento\Framework\Model\AbstractModel',
            [
                'getEntityTypeId',
                'getAttributeSetId',
                'beforeDelete',
                'getId',
                'isDeleted',
                'afterDelete',
                'afterDeleteCommit',
                '__wakeup'
            ],
            [],
            '',
            false
        );

    }

    /**
     * @expectedException \Magento\Framework\Exception\StateException
     * @expectedExceptionMessage Default attribute set can not be deleted
     * @return void
     */
    public function testBeforeDeleteStateException()
    {
        $this->resourceMock->expects($this->any())
            ->method('getConnection')
            ->willReturn($this->getMock('\Magento\Framework\DB\Adapter\AdapterInterface'));

        $this->transactionManagerMock->expects($this->once())
            ->method('start')
            ->with($this->getMock('\Magento\Framework\DB\Adapter\AdapterInterface'))
            ->willReturn($this->getMock('\Magento\Framework\DB\Adapter\AdapterInterface'));

        $this->objectMock->expects($this->once())->method('getEntityTypeId')->willReturn(665);
        $this->eavConfigMock->expects($this->once())->method('getEntityType')->with(665)->willReturn($this->typeMock);
        $this->typeMock->expects($this->once())->method('getDefaultAttributeSetId')->willReturn(4);
        $this->objectMock->expects($this->once())->method('getAttributeSetId')->willReturn(4);

        $this->model->delete($this->objectMock);
    }

    /**
     * @expectedException \Exception
     * @expectedExceptionMessage test exception
     * @return void
     */
    public function testBeforeDelete()
    {
        $this->resourceMock->expects($this->any())
            ->method('getConnection')
            ->willReturn($this->getMock('\Magento\Framework\DB\Adapter\AdapterInterface'));

        $this->transactionManagerMock->expects($this->once())
            ->method('start')
            ->with($this->getMock('\Magento\Framework\DB\Adapter\AdapterInterface'))
            ->willReturn($this->getMock('\Magento\Framework\DB\Adapter\AdapterInterface'));

        $this->objectMock->expects($this->once())->method('getEntityTypeId')->willReturn(665);
        $this->eavConfigMock->expects($this->once())->method('getEntityType')->with(665)->willReturn($this->typeMock);
        $this->typeMock->expects($this->once())->method('getDefaultAttributeSetId')->willReturn(4);
        $this->objectMock->expects($this->once())->method('getAttributeSetId')->willReturn(5);
        $this->relationProcessor->expects($this->once())
            ->method('delete')
            ->willThrowException(new \Exception('test exception'));

        $this->model->delete($this->objectMock);
    }

    /**
     * @return void
     */
    public function testGetSetInfoCacheMiss()
    {
        $cacheMock = $this->getMockBuilder('Magento\Framework\App\CacheInterface')
            ->disableOriginalConstructor()
            ->setMethods(['load', 'save', 'getFrontend', 'remove', 'clean'])
            ->getMock();
        $cacheKey = Set::ATTRIBUTES_CACHE_ID . 1;
        $cacheMock
            ->expects($this->once())
            ->method('load')
            ->with($cacheKey)
            ->willReturn(false);
        $cacheMock
            ->expects($this->once())
            ->method('save')
            ->with(
                serialize(
                    [
                        1 => [
                            10000 => [
                                'group_id' =>  10,
                                'group_sort' =>  100,
                                'sort' =>  1000
                            ]
                        ]
                    ]
                ),
                $cacheKey,
                [\Magento\Eav\Model\Cache\Type::CACHE_TAG, \Magento\Eav\Model\Entity\Attribute::CACHE_TAG]
            );

        $this->eavConfigMock->expects($this->any())->method('isCacheEnabled')->willReturn(true);
        $this->eavConfigMock->expects($this->any())->method('getCache')->willReturn($cacheMock);

        $fetchResult = [
            [
                'attribute_id' => 1,
                'attribute_group_id' => 10,
                'group_sort_order' => 100,
                'sort_order' => 1000,
                'attribute_set_id' => 10000
            ]
        ];

        $selectMock = $this->getMockBuilder('Magento\Framework\DB\Select')
            ->disableOriginalConstructor()
            ->setMethods(['from', 'joinLeft', 'where'])
            ->getMock();
        $selectMock->expects($this->once())->method('from')->will($this->returnSelf());
        $selectMock->expects($this->once())->method('joinLeft')->will($this->returnSelf());
        $selectMock->expects($this->atLeastOnce())->method('where')->will($this->returnSelf());

        $connectionMock = $this->getMockBuilder('Magento\Framework\DB\Adapter\Pdo\Mysql')
            ->disableOriginalConstructor()
            ->setMethods(['select', 'fetchAll'])
            ->getMock();
        $connectionMock->expects($this->atLeastOnce())->method('select')->willReturn($selectMock);
        $connectionMock->expects($this->atLeastOnce())->method('fetchAll')->willReturn($fetchResult);

        $this->resourceMock->expects($this->any())->method('getConnection')->willReturn($connectionMock);
        $this->resourceMock->expects($this->any())->method('getTableName')->willReturn('_TABLE_');
        $this->assertEquals(
            [
                1 => [
                    10000 => [
                        'group_id' =>  10,
                        'group_sort' =>  100,
                        'sort' =>  1000
                    ]
                ],
                2 => [],
                3 => []
            ],
            $this->model->getSetInfo([1, 2, 3], 1)
        );
    }

    /**
     * @return void
     */
    public function testGetSetInfoCacheHit()
    {
        $cached = [
            1 => [
                10000 => [
                    'group_id' => 10,
                    'group_sort' => 100,
                    'sort' => 1000
                ]
            ]
        ];

        $this->resourceMock->expects($this->never())->method('getConnection');
        $this->eavConfigMock->expects($this->any())->method('isCacheEnabled')->willReturn(true);
        $cacheMock = $this->getMockBuilder('Magento\Framework\App\CacheInterface')
            ->disableOriginalConstructor()
            ->setMethods(['load', 'save', 'getFrontend', 'remove', 'clean'])
            ->getMock();
        $cacheMock
            ->expects($this->once())
            ->method('load')
            ->with(Set::ATTRIBUTES_CACHE_ID . 1)
            ->willReturn(serialize($cached));

        $this->eavConfigMock->expects($this->any())->method('getCache')->willReturn($cacheMock);

        $this->assertEquals(
            [
                1 => [
                    10000 => [
                        'group_id' =>  10,
                        'group_sort' =>  100,
                        'sort' =>  1000
                    ]
                ],
                2 => [],
                3 => []
            ],
            $this->model->getSetInfo([1, 2, 3], 1)
        );
    }
}