/**
* Test _beforeSaveMethod via save() with failed validation
*
* @expectedException \Magento\Framework\Exception\LocalizedException
* @expectedExceptionMessage Cannot save comment:
*/
public function testSaveValidationFailed()
{
$this->entitySnapshotMock->expects($this->once())->method('isModified')->with($this->commentModelMock)->willReturn(true);
$this->validatorMock->expects($this->once())->method('validate')->with($this->equalTo($this->commentModelMock))->will($this->returnValue(['warning message']));
$this->commentResource->save($this->commentModelMock);
$this->assertTrue(true);
}
public function testProcessRelations()
{
$this->shipmentMock->expects($this->once())->method('getId')->willReturn('shipment-id-value');
$this->shipmentMock->expects($this->exactly(2))->method('getItems')->willReturn([$this->itemMock]);
$this->shipmentMock->expects($this->exactly(2))->method('getComments')->willReturn([$this->commentMock]);
$this->shipmentMock->expects($this->exactly(2))->method('getTracks')->willReturn([$this->trackMock]);
$this->itemMock->expects($this->once())->method('setParentId')->with('shipment-id-value')->willReturnSelf();
$this->itemResourceMock->expects($this->once())->method('save')->with($this->itemMock)->willReturnSelf();
$this->commentResourceMock->expects($this->once())->method('save')->with($this->commentMock)->willReturnSelf();
$this->trackResourceMock->expects($this->once())->method('save')->with($this->trackMock)->willReturnSelf();
$this->relationProcessor->processRelation($this->shipmentMock);
}
/**
* Process relations for Shipment
*
* @param \Magento\Framework\Model\AbstractModel $object
* @return void
* @throws \Exception
*/
public function processRelation(\Magento\Framework\Model\AbstractModel $object)
{
/** @var \Magento\Sales\Model\Order\Shipment $object */
if (null !== $object->getItems()) {
foreach ($object->getItems() as $item) {
$item->setParentId($object->getId());
$this->shipmentItemResource->save($item);
}
}
if (null !== $object->getTracks()) {
foreach ($object->getTracks() as $track) {
$this->shipmentTrackResource->save($track);
}
}
if (null !== $object->getComments()) {
foreach ($object->getComments() as $comment) {
$this->shipmentCommentResource->save($comment);
}
}
}