public function testSetProduct() { /** @var \Magento\Catalog\Model\Product|\PHPUnit_Framework_MockObject_MockObject $product */ $product = $this->getMock('Magento\\Catalog\\Model\\Product', [], [], '', false); $product->expects($this->any())->method('getId')->will($this->returnValue('5')); $productStore = new \Magento\Framework\DataObject(['id' => 33]); $product->expects($this->any())->method('getStore')->will($this->returnValue($productStore)); $this->collection->setProduct($product); $this->assertEquals(33, $this->collection->getStoreId()); }