public function testSave() { $rule = $this->getMock('\\Magento\\SalesRule\\Model\\Data\\Rule', [], [], '', false); $model = $this->getMock('\\Magento\\SalesRule\\Model\\Rule', [], [], '', false); $this->toModelConverter->expects($this->once())->method('toModel')->with($rule)->willReturn($model); $model->expects($this->once())->method('save'); $model->expects($this->once())->method('getId')->willReturn(10); $model->expects($this->once())->method('load')->with(10); $model->expects($this->once())->method('getStoreLabels'); $this->toDataModelConverter->expects($this->once())->method('toDataModel')->with($model)->willReturn($rule); $this->assertEquals($rule, $this->ruleRepository->save($rule)); }