#10208: Fix getImage method when sending product alert email

 - Unit test updated

Author: ishakhsuvarov

app/code/Magento/ProductAlert/Test/Unit/Block/Email/StockTest.php

@@ -25,6 +25,16 @@ class StockTest extends \PHPUnit_Framework_TestCase
      */
     protected $imageBuilder;
 
+    /**
+     * @var \Magento\Store\Model\StoreManagerInterface|\PHPUnit_Framework_MockObject_MockObject
+     */
+    private $storeManagerMock;
+
+    /**
+     * @var \Magento\Store\Model\App\Emulation|\PHPUnit_Framework_MockObject_MockObject
+     */
+    private $appEmulationMock;
+
     protected function setUp()
     {
         $objectManager = new \Magento\Framework\TestFramework\Unit\Helper\ObjectManager($this);
@@ -39,12 +49,25 @@ protected function setUp()
         $this->imageBuilder = $this->getMockBuilder(\Magento\Catalog\Block\Product\ImageBuilder::class)
             ->disableOriginalConstructor()
             ->getMock();
+        $this->storeManagerMock = $this->getMockBuilder(\Magento\Store\Model\StoreManagerInterface::class)
+            ->disableOriginalConstructor()
+            ->getMock();
+        $this->appEmulationMock = $this->getMockBuilder(\Magento\Store\Model\App\Emulation::class)
+            ->disableOriginalConstructor()
+            ->getMock();
+
+        $contextMock = $this->getMockBuilder(\Magento\Framework\View\Element\Template\Context::class)
+            ->disableOriginalConstructor()
+            ->getMock();
+        $contextMock->expects($this->any())->method('getStoreManager')->willReturn($this->storeManagerMock);
 
         $this->_block = $objectManager->getObject(
             \Magento\ProductAlert\Block\Email\Stock::class,
             [
                 'maliciousCode' => $this->_filter,
                 'imageBuilder' => $this->imageBuilder,
+                'context' => $contextMock,
+                'appEmulation' => $this->appEmulationMock
             ]
         );
     }
@@ -82,6 +105,13 @@ public function testGetImage()
             ->disableOriginalConstructor()
             ->getMock();
 
+        $storeMock = $this->getMockBuilder(\Magento\Store\Api\Data\StoreInterface::class)
+            ->disableOriginalConstructor()
+            ->getMock();
+
+        $this->appEmulationMock->expects($this->once())->method('startEnvironmentEmulation');
+        $this->storeManagerMock->expects($this->atLeastOnce())->method('getStore')->willReturn($storeMock);
+        $storeMock->expects($this->atLeastOnce())->method('getId')->willReturn(42);
         $this->imageBuilder->expects($this->once())
             ->method('setProduct')
             ->with($productMock)
@@ -97,10 +127,38 @@ public function testGetImage()
         $this->imageBuilder->expects($this->once())
             ->method('create')
             ->willReturn($imageMock);
+        $this->appEmulationMock->expects($this->once())->method('stopEnvironmentEmulation');
 
         $this->assertInstanceOf(
             \Magento\Catalog\Block\Product\Image::class,
             $this->_block->getImage($productMock, $imageId, $attributes)
         );
     }
+
+    /**
+     * Test that app emulation stops when exception occurs.
+     *
+     * @expectedException \Exception
+     * @expectedExceptionMessage Image Builder Exception
+     */
+    public function testGetImageThrowsAnException()
+    {
+        $productMock = $this->getMockBuilder(\Magento\Catalog\Model\Product::class)
+            ->disableOriginalConstructor()
+            ->getMock();
+        $storeMock = $this->getMockBuilder(\Magento\Store\Api\Data\StoreInterface::class)
+            ->disableOriginalConstructor()
+            ->getMock();
+
+        $this->appEmulationMock->expects($this->once())->method('startEnvironmentEmulation');
+        $this->storeManagerMock->expects($this->atLeastOnce())->method('getStore')->willReturn($storeMock);
+        $storeMock->expects($this->atLeastOnce())->method('getId')->willReturn(42);
+
+        $this->imageBuilder->expects($this->once())
+            ->method('setProduct')
+            ->willThrowException(new \Exception("Image Builder Exception"));
+        $this->appEmulationMock->expects($this->once())->method('stopEnvironmentEmulation');
+
+        $this->_block->getImage($productMock, 1, []);
+    }
 }