/** * @test * * @expectedException \Magento\Framework\Exception\CouldNotSaveException */ public function testSaveException() { $this->pageResource->expects($this->once())->method('save')->with($this->page)->willThrowException(new \Exception()); $this->repository->save($this->page); }
/** * Run test save method * * @return void */ public function testSave() { $pageMock = $this->getMock('Magento\\Cms\\Model\\Page', [], [], '', false); $this->resourceMock->expects($this->once())->method('save')->with($pageMock); $this->assertEquals($pageMock, $this->pageRepository->save($pageMock)); }