pageFactory = $this->getMockBuilder(\Magento\Cms\Model\PageFactory::class) ->disableOriginalConstructor(true) ->setMethods(['create']) ->getMock(); $this->pageResource = $this->getMockBuilder(\Magento\Cms\Model\ResourceModel\Page::class) ->disableOriginalConstructor(true) ->getMock(); $this->page = $this->getMockBuilder(\Magento\Cms\Model\Page::class) ->disableOriginalConstructor() ->setMethods(['setStoreId', 'getId']) ->getMock(); $this->getPageByIdentifierCommand = new GetPageByIdentifier($this->pageFactory, $this->pageResource); } /** * Test for getByIdentifier method */ public function testGetByIdentifier() { $identifier = 'home'; $storeId = 0; $this->pageFactory->expects($this->once()) ->method('create') ->willReturn($this->page); $this->page->expects($this->once()) ->method('setStoreId') ->willReturn($this->page); $this->page->expects($this->once()) ->method('getId') ->willReturn(1); $this->pageResource->expects($this->once()) ->method('load') ->with($this->page, $identifier) ->willReturn($this->page); $this->getPageByIdentifierCommand->execute($identifier, $storeId); } }