/**
* @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();
}
/**
* @param \Magento\Store\Model\WebsiteRepository $subject
* @param \Magento\Store\Model\Website $website
* @return \Magento\Store\Model\Website
*/
public function afterGetDefault(WebsiteRepository $subject, Website $website)
{
if (!$this->isNeededProcess()) {
return $website;
}
if ($this->geoWebsite->getId() != $website->getId()) {
$website = $subject->getById($this->geoWebsite->getId());
}
return $website;
}
