public function testGetStorageModel() { $storageModelMock = $this->getMockBuilder('Magento\\Framework\\Model\\AbstractModel')->disableOriginalConstructor()->getMock(); $this->storageMock->expects($this->once())->method('getStorageModel')->will($this->returnValue($storageModelMock)); $this->assertSame($storageModelMock, $this->helper->getStorageModel()); }