public function testGetType() { $imageId = 'test_image_id'; $attributes = []; $data = ['type' => 'image']; $productMock = $this->getMockBuilder('Magento\\Catalog\\Model\\Product')->disableOriginalConstructor()->getMock(); $this->prepareAttributes($data, $imageId); $this->helper->init($productMock, $imageId, $attributes); $this->assertEquals($data['type'], $this->helper->getType()); }