public function testGetImage()
{
$imageId = 'test_image_id';
$attributes = [];
$productMock = $this->getMockBuilder('Magento\\Catalog\\Model\\Product')->disableOriginalConstructor()->getMock();
$imageMock = $this->getMockBuilder('Magento\\Catalog\\Block\\Product\\Image')->disableOriginalConstructor()->getMock();
$this->imageBuilder->expects($this->once())->method('setProduct')->with($productMock)->willReturnSelf();
$this->imageBuilder->expects($this->once())->method('setImageId')->with($imageId)->willReturnSelf();
$this->imageBuilder->expects($this->once())->method('setAttributes')->with($attributes)->willReturnSelf();
$this->imageBuilder->expects($this->once())->method('create')->willReturn($imageMock);
$this->assertInstanceOf('Magento\\Catalog\\Block\\Product\\Image', $this->_block->getImage($productMock, $imageId, $attributes));
}
