public function testGetCount() { $websitesMock = $this->getMock('\\Magento\\Store\\Model\\ResourceModel\\Website\\Collection', [], [], '', false); $this->websitesFactoryMock->expects($this->once())->method('create')->willReturn($websitesMock); $websitesMock->expects($this->once())->method('getSize')->willReturn('expected'); $this->assertEquals('expected', $this->model->getCount()); }
/** * @expectedException \DomainException * @expectedExceptionMessage Default website is not defined */ public function testGetDefaultIsZero() { $collectionMock = $this->getMockBuilder('Magento\\Store\\Model\\ResourceModel\\Website\\Collection')->disableOriginalConstructor()->setMethods([])->getMock(); $this->websiteCollectionFactoryMock->expects($this->any())->method('create')->willReturn($collectionMock); $collectionMock->expects($this->any())->method('addFieldToFilter'); $collectionMock->expects($this->any())->method('getItems')->willReturn([]); $this->model->getDefault(); }