public function testGetDefaultWebsiteId() { $id = 1; $websiteMock = $this->getMockBuilder('Magento\\Store\\Model\\Website')->disableOriginalConstructor()->getMock(); $websiteMock->expects($this->once())->method('getId')->willReturn($id); $this->storeManagerMock->expects($this->once())->method('getWebsite')->with(true)->willReturn($websiteMock); $this->assertEquals($id, $this->model->getDefaultWebsiteId()); }