protected function setUp()
{
parent::setUp();
$this->buildFactoryMock = $this->getMockBuilder(BuildFactory::class)->setMethods(['create'])->disableOriginalConstructor()->getMock();
$this->filterManagerMock = $this->getMockBuilder(FilterManager::class)->disableOriginalConstructor()->getMock();
$this->productHelperMock = $this->getMockBuilder(ProductHelper::class)->disableOriginalConstructor()->getMock();
$this->attributeFactoryMock = $this->getMockBuilder(AttributeFactory::class)->setMethods(['create'])->disableOriginalConstructor()->getMock();
$this->validatorFactoryMock = $this->getMockBuilder(ValidatorFactory::class)->setMethods(['create'])->disableOriginalConstructor()->getMock();
$this->groupCollectionFactoryMock = $this->getMockBuilder(CollectionFactory::class)->setMethods(['create'])->disableOriginalConstructor()->getMock();
$this->layoutFactoryMock = $this->getMockBuilder(LayoutFactory::class)->disableOriginalConstructor()->getMock();
$this->redirectMock = $this->getMockBuilder(ResultRedirect::class)->disableOriginalConstructor()->getMock();
$this->attributeSetMock = $this->getMockBuilder(AttributeSetInterface::class)->disableOriginalConstructor()->getMock();
$this->builderMock = $this->getMockBuilder(Build::class)->disableOriginalConstructor()->getMock();
$this->inputTypeValidatorMock = $this->getMockBuilder(InputTypeValidator::class)->disableOriginalConstructor()->getMock();
$this->buildFactoryMock->expects($this->any())->method('create')->willReturn($this->builderMock);
$this->validatorFactoryMock->expects($this->any())->method('create')->willReturn($this->inputTypeValidatorMock);
}
/**
* @return \Magento\Backend\Model\View\Result\Redirect
* @SuppressWarnings(PHPMD.CyclomaticComplexity)
* @SuppressWarnings(PHPMD.NPathComplexity)
* @SuppressWarnings(PHPMD.ExcessiveMethodLength)
*/
public function execute()
{
$data = $this->getRequest()->getPostValue();
$resultRedirect = $this->resultRedirectFactory->create();
if ($data) {
$setId = $this->getRequest()->getParam('set');
$attributeSet = null;
if (!empty($data['new_attribute_set_name'])) {
$name = $this->filterManager->stripTags($data['new_attribute_set_name']);
$name = trim($name);
try {
/** @var $attributeSet \Magento\Eav\Model\Entity\Attribute\Set */
$attributeSet = $this->buildFactory->create()->setEntityTypeId($this->_entityTypeId)->setSkeletonId($setId)->setName($name)->getAttributeSet();
} catch (AlreadyExistsException $alreadyExists) {
$this->messageManager->addError(__('An attribute set named \'%1\' already exists.', $name));
$this->messageManager->setAttributeData($data);
return $resultRedirect->setPath('catalog/*/edit', ['_current' => true]);
} catch (\Magento\Framework\Exception\LocalizedException $e) {
$this->messageManager->addError($e->getMessage());
} catch (\Exception $e) {
$this->messageManager->addException($e, __('Something went wrong while saving the attribute.'));
}
}
$redirectBack = $this->getRequest()->getParam('back', false);
/* @var $model \Magento\Catalog\Model\ResourceModel\Eav\Attribute */
$model = $this->attributeFactory->create();
$attributeId = $this->getRequest()->getParam('attribute_id');
$attributeCode = $this->getRequest()->getParam('attribute_code');
$frontendLabel = $this->getRequest()->getParam('frontend_label');
$attributeCode = $attributeCode ?: $this->generateCode($frontendLabel[0]);
if (strlen($this->getRequest()->getParam('attribute_code')) > 0) {
$validatorAttrCode = new \Zend_Validate_Regex(['pattern' => '/^[a-z][a-z_0-9]{0,30}$/']);
if (!$validatorAttrCode->isValid($attributeCode)) {
$this->messageManager->addError(__('Attribute code "%1" is invalid. Please use only letters (a-z), ' . 'numbers (0-9) or underscore(_) in this field, first character should be a letter.', $attributeCode));
return $resultRedirect->setPath('catalog/*/edit', ['attribute_id' => $attributeId, '_current' => true]);
}
}
$data['attribute_code'] = $attributeCode;
//validate frontend_input
if (isset($data['frontend_input'])) {
/** @var $inputType \Magento\Eav\Model\Adminhtml\System\Config\Source\Inputtype\Validator */
$inputType = $this->validatorFactory->create();
if (!$inputType->isValid($data['frontend_input'])) {
foreach ($inputType->getMessages() as $message) {
$this->messageManager->addError($message);
}
return $resultRedirect->setPath('catalog/*/edit', ['attribute_id' => $attributeId, '_current' => true]);
}
}
if ($attributeId) {
$model->load($attributeId);
if (!$model->getId()) {
$this->messageManager->addError(__('This attribute no longer exists.'));
return $resultRedirect->setPath('catalog/*/');
}
// entity type check
if ($model->getEntityTypeId() != $this->_entityTypeId) {
$this->messageManager->addError(__('We can\'t update the attribute.'));
$this->_session->setAttributeData($data);
return $resultRedirect->setPath('catalog/*/');
}
$data['attribute_code'] = $model->getAttributeCode();
$data['is_user_defined'] = $model->getIsUserDefined();
$data['frontend_input'] = $model->getFrontendInput();
} else {
/**
* @todo add to helper and specify all relations for properties
*/
$data['source_model'] = $this->productHelper->getAttributeSourceModelByInputType($data['frontend_input']);
$data['backend_model'] = $this->productHelper->getAttributeBackendModelByInputType($data['frontend_input']);
}
$data += ['is_filterable' => 0, 'is_filterable_in_search' => 0, 'apply_to' => []];
if (is_null($model->getIsUserDefined()) || $model->getIsUserDefined() != 0) {
$data['backend_type'] = $model->getBackendTypeByInput($data['frontend_input']);
}
$defaultValueField = $model->getDefaultValueByInput($data['frontend_input']);
if ($defaultValueField) {
$data['default_value'] = $this->getRequest()->getParam($defaultValueField);
}
if (!$model->getIsUserDefined() && $model->getId()) {
// Unset attribute field for system attributes
unset($data['apply_to']);
}
$model->addData($data);
if (!$attributeId) {
$model->setEntityTypeId($this->_entityTypeId);
$model->setIsUserDefined(1);
}
$groupCode = $this->getRequest()->getParam('group');
if ($setId && $groupCode) {
// For creating product attribute on product page we need specify attribute set and group
$attributeSetId = $attributeSet ? $attributeSet->getId() : $setId;
$groupCollection = $attributeSet ? $attributeSet->getGroups() : $this->groupCollectionFactory->create()->setAttributeSetFilter($attributeSetId)->load();
foreach ($groupCollection as $group) {
if ($group->getAttributeGroupCode() == $groupCode) {
$attributeGroupId = $group->getAttributeGroupId();
break;
}
}
$model->setAttributeSetId($attributeSetId);
$model->setAttributeGroupId($attributeGroupId);
}
try {
$model->save();
$this->messageManager->addSuccess(__('You saved the product attribute.'));
$this->_attributeLabelCache->clean();
$this->_session->setAttributeData(false);
if ($this->getRequest()->getParam('popup')) {
$requestParams = ['attributeId' => $this->getRequest()->getParam('product'), 'attribute' => $model->getId(), '_current' => true, 'product_tab' => $this->getRequest()->getParam('product_tab')];
if (!is_null($attributeSet)) {
$requestParams['new_attribute_set_id'] = $attributeSet->getId();
}
$resultRedirect->setPath('catalog/product/addAttribute', $requestParams);
} elseif ($redirectBack) {
$resultRedirect->setPath('catalog/*/edit', ['attribute_id' => $model->getId(), '_current' => true]);
} else {
$resultRedirect->setPath('catalog/*/');
}
return $resultRedirect;
} catch (\Exception $e) {
$this->messageManager->addError($e->getMessage());
$this->_session->setAttributeData($data);
return $resultRedirect->setPath('catalog/*/edit', ['attribute_id' => $attributeId, '_current' => true]);
}
}
return $resultRedirect->setPath('catalog/*/');
}