public function testProcessRelations() { $this->relationProcessorMock->expects($this->once())->method('processRelation')->with($this->modelMock); $this->modelMock->expects($this->once())->method('getEventPrefix')->willReturn('custom_event_prefix'); $this->eventManagerMock->expects($this->once())->method('dispatch')->with('custom_event_prefix_process_relation', ['object' => $this->modelMock]); $this->entityRelationComposite->processRelations($this->modelMock); }
/** * @inheritdoc */ public function save(\Magento\Framework\Model\AbstractModel $object) { /** * Direct deleted items to delete method */ if ($object->isDeleted()) { return $this->delete($object); } $this->beginTransaction(); try { if (!$this->isModified($object)) { $this->entityRelationComposite->processRelations($object); $this->commit(); return $this; } $object->validateBeforeSave(); $object->beforeSave(); if ($object->isSaveAllowed()) { if (!$this->isPartialSave()) { $this->loadAllAttributes($object); } if ($this->getEntityTable() == \Magento\Eav\Model\Entity::DEFAULT_ENTITY_TABLE && !$object->getEntityTypeId()) { $object->setEntityTypeId($this->getTypeId()); } $object->setParentId((int) $object->getParentId()); $this->objectRelationProcessor->validateDataIntegrity($this->getEntityTable(), $object->getData()); $this->_beforeSave($object); $this->_processSaveData($this->_collectSaveData($object)); $this->_afterSave($object); $this->entitySnapshot->registerSnapshot($object); $object->afterSave(); $this->entityRelationComposite->processRelations($object); } $this->addCommitCallback([$object, 'afterCommitCallback'])->commit(); $object->setHasDataChanges(false); } catch (\Exception $e) { $this->rollBack(); $object->setHasDataChanges(true); throw $e; } return $this; }