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();
}
