Your IP : 216.73.216.220
<?php
/**
* Copyright © Magento, Inc. All rights reserved.
* See COPYING.txt for license details.
*/
declare(strict_types=1);
namespace Magento\Eav\Test\Unit\Model;
use Magento\Eav\Api\Data\AttributeSetSearchResultsInterface;
use Magento\Eav\Api\Data\AttributeSetSearchResultsInterfaceFactory;
use Magento\Eav\Model\AttributeSetRepository;
use Magento\Eav\Model\Config;
use Magento\Eav\Model\Entity\Attribute\SetFactory;
use Magento\Eav\Model\ResourceModel\Entity\Attribute\Set;
use Magento\Eav\Model\ResourceModel\Entity\Attribute\Set\Collection;
use Magento\Eav\Model\ResourceModel\Entity\Attribute\Set\CollectionFactory;
use Magento\Framework\Api\ExtensionAttribute\JoinProcessor;
use Magento\Framework\Api\SearchCriteria\CollectionProcessorInterface;
use Magento\Framework\Api\SearchCriteriaInterface;
use Magento\Framework\Exception\CouldNotDeleteException;
use Magento\Framework\Exception\StateException;
use PHPUnit\Framework\MockObject\MockObject;
use PHPUnit\Framework\TestCase;
/**
* @SuppressWarnings(PHPMD.LongVariable)
* @SuppressWarnings(PHPMD.CouplingBetweenObjects)
*/
class AttributeSetRepositoryTest extends TestCase
{
/**
* @var AttributeSetRepository
*/
private $model;
/**
* @var MockObject
*/
private $resourceMock;
/**
* @var MockObject
*/
private $setFactoryMock;
/**
* @var MockObject
*/
private $collectionFactoryMock;
/**
* @var MockObject
*/
private $eavConfigMock;
/**
* @var MockObject
*/
private $resultFactoryMock;
/**
* @var MockObject
*/
private $extensionAttributesJoinProcessorMock;
/**
* @var CollectionProcessorInterface|MockObject
*/
private $collectionProcessor;
/**
* @return void
*/
protected function setUp(): void
{
$this->resourceMock = $this->createMock(Set::class);
$this->setFactoryMock = $this->createPartialMock(
SetFactory::class,
['create']
);
$this->collectionFactoryMock = $this->createPartialMock(
CollectionFactory::class,
['create']
);
$this->eavConfigMock = $this->createPartialMock(Config::class, ['getEntityType']);
$this->resultFactoryMock = $this->createPartialMock(
AttributeSetSearchResultsInterfaceFactory::class,
['create']
);
$this->extensionAttributesJoinProcessorMock = $this->createPartialMock(
JoinProcessor::class,
['process']
);
$this->collectionProcessor = $this->getMockBuilder(CollectionProcessorInterface::class)
->getMockForAbstractClass();
$this->model = new AttributeSetRepository(
$this->resourceMock,
$this->setFactoryMock,
$this->collectionFactoryMock,
$this->eavConfigMock,
$this->resultFactoryMock,
$this->extensionAttributesJoinProcessorMock,
$this->collectionProcessor
);
}
/**
* @return void
*/
public function testGet()
{
$attributeSetId = 1;
$attributeSetMock = $this->createMock(\Magento\Eav\Model\Entity\Attribute\Set::class);
$this->setFactoryMock->expects($this->once())->method('create')->willReturn($attributeSetMock);
$this->resourceMock->expects($this->once())->method('load')->with($attributeSetMock, $attributeSetId, null);
$attributeSetMock->expects($this->any())->method('getId')->willReturn($attributeSetId);
$this->assertEquals($attributeSetMock, $this->model->get($attributeSetId));
}
/**
* @return void
*/
public function testGetThrowsExceptionIfRequestedAttributeSetDoesNotExist()
{
$this->expectException('Magento\Framework\Exception\NoSuchEntityException');
$this->expectExceptionMessage('No such entity with id = 9999');
$attributeSetId = 9999;
$attributeSetMock = $this->createMock(\Magento\Eav\Model\Entity\Attribute\Set::class);
$this->setFactoryMock->expects($this->once())->method('create')->willReturn($attributeSetMock);
$this->resourceMock->expects($this->once())->method('load')->with($attributeSetMock, $attributeSetId, null);
$this->model->get($attributeSetId);
}
/**
* @return void
*/
public function testSave()
{
$attributeSetMock = $this->createMock(\Magento\Eav\Model\Entity\Attribute\Set::class);
$this->resourceMock->expects($this->once())->method('save')->with($attributeSetMock);
$this->assertEquals($attributeSetMock, $this->model->save($attributeSetMock));
}
/**
* @return void
*/
public function testSaveThrowsExceptionIfGivenEntityCannotBeSaved()
{
$this->expectException('Magento\Framework\Exception\CouldNotSaveException');
$attributeSetMock = $this->createMock(\Magento\Eav\Model\Entity\Attribute\Set::class);
$this->resourceMock->expects($this->once())->method('save')->with($attributeSetMock)->willThrowException(
new \Exception('Some internal exception message.')
);
$this->model->save($attributeSetMock);
$this->expectExceptionMessage(
"The attribute set couldn't be saved due to an error. Verify your information and try again. "
. "If the error persists, please try again later."
);
}
/**
* @return void
*/
public function testDelete()
{
$attributeSetMock = $this->createMock(\Magento\Eav\Model\Entity\Attribute\Set::class);
$this->resourceMock->expects($this->once())->method('delete')->with($attributeSetMock);
$this->assertTrue($this->model->delete($attributeSetMock));
}
/**
* @return void
*/
public function testDeleteThrowsExceptionIfGivenEntityCannotBeDeleted()
{
$this->expectException('Magento\Framework\Exception\CouldNotDeleteException');
$attributeSetMock = $this->createMock(\Magento\Eav\Model\Entity\Attribute\Set::class);
$this->resourceMock->expects($this->once())->method('delete')->with($attributeSetMock)->willThrowException(
new CouldNotDeleteException(__('Some internal exception message.'))
);
$this->model->delete($attributeSetMock);
$this->expectExceptionMessage(
"The attribute set couldn't be deleted due to an error. "
. "Try again — if the error persists, please try again later."
);
}
/**
* @return void
*/
public function testDeleteThrowsExceptionIfGivenAttributeSetIsDefault()
{
$this->expectException('Magento\Framework\Exception\CouldNotDeleteException');
$this->expectExceptionMessage('The default attribute set can\'t be deleted.');
$attributeSetMock = $this->createMock(\Magento\Eav\Model\Entity\Attribute\Set::class);
$this->resourceMock->expects($this->once())->method('delete')->with($attributeSetMock)->willThrowException(
new StateException(__('Some internal exception message.'))
);
$this->model->delete($attributeSetMock);
}
/**
* @return void
*/
public function testDeleteById()
{
$attributeSetId = 1;
$attributeSetMock = $this->createMock(\Magento\Eav\Model\Entity\Attribute\Set::class);
$attributeSetMock->expects($this->any())->method('getId')->willReturn($attributeSetId);
$this->setFactoryMock->expects($this->once())->method('create')->willReturn($attributeSetMock);
$this->resourceMock->expects($this->once())->method('load')->with($attributeSetMock, $attributeSetId, null);
$this->resourceMock->expects($this->once())->method('delete')->with($attributeSetMock);
$this->assertTrue($this->model->deleteById($attributeSetId));
}
/**
* @return void
*/
public function testGetList()
{
$attributeSetMock = $this->createMock(\Magento\Eav\Model\Entity\Attribute\Set::class);
$collectionMock = $this->getMockBuilder(Collection::class)
->disableOriginalConstructor()
->setMethods([
'getItems',
'getSize',
])
->getMock();
$collectionMock->expects($this->once())
->method('getItems')
->willReturn([$attributeSetMock]);
$collectionMock->expects($this->once())
->method('getSize')
->willReturn(1);
$this->collectionFactoryMock->expects($this->once())->method('create')->willReturn($collectionMock);
$searchCriteriaMock = $this->getMockForAbstractClass(SearchCriteriaInterface::class);
$resultMock = $this->getMockBuilder(AttributeSetSearchResultsInterface::class)
->getMockForAbstractClass();
$resultMock->expects($this->once())
->method('setSearchCriteria')
->with($searchCriteriaMock)
->willReturnSelf();
$resultMock->expects($this->once())
->method('setItems')
->with([$attributeSetMock])
->willReturnSelf();
$resultMock->expects($this->once())
->method('setTotalCount')
->with(1)
->willReturnSelf();
$this->resultFactoryMock->expects($this->once())
->method('create')
->willReturn($resultMock);
$this->collectionProcessor->expects($this->once())
->method('process')
->with($searchCriteriaMock, $collectionMock)
->willReturnSelf();
$this->model->getList($searchCriteriaMock);
}
/**
* @return void
*/
public function testGetListIfEntityTypeCodeIsNull()
{
$attributeSetMock = $this->createMock(\Magento\Eav\Model\Entity\Attribute\Set::class);
$collectionMock = $this->getMockBuilder(Collection::class)
->disableOriginalConstructor()
->setMethods([
'getItems',
'getSize',
])
->getMock();
$collectionMock->expects($this->once())
->method('getItems')
->willReturn([$attributeSetMock]);
$collectionMock->expects($this->once())
->method('getSize')
->willReturn(1);
$this->collectionFactoryMock->expects($this->once())
->method('create')
->willReturn($collectionMock);
$searchCriteriaMock = $this->getMockForAbstractClass(SearchCriteriaInterface::class);
$resultMock = $this->getMockBuilder(AttributeSetSearchResultsInterface::class)
->getMockForAbstractClass();
$resultMock->expects($this->once())
->method('setSearchCriteria')
->with($searchCriteriaMock)
->willReturnSelf();
$resultMock->expects($this->once())
->method('setItems')
->with([$attributeSetMock])
->willReturnSelf();
$resultMock->expects($this->once())
->method('setTotalCount')
->with(1)
->willReturnSelf();
$this->resultFactoryMock->expects($this->once())
->method('create')
->willReturn($resultMock);
$this->collectionProcessor->expects($this->once())
->method('process')
->with($searchCriteriaMock, $collectionMock)
->willReturnSelf();
$this->model->getList($searchCriteriaMock);
}
}