MC-13613 Product mass update

Author: vzabaznov

lib/internal/Magento/Framework/MessageQueue/Test/Unit/ConsumerTest.php

@@ -6,6 +6,8 @@
 
 namespace Magento\Framework\MessageQueue\Test\Unit;
 
+use Magento\Framework\MessageQueue\PoisonPill\PoisonPillCompareInterface;
+use Magento\Framework\MessageQueue\PoisonPill\PoisonPillReadInterface;
 use Magento\Framework\Phrase;
 
 /**
@@ -65,6 +67,16 @@ class ConsumerTest extends \PHPUnit\Framework\TestCase
      */
     private $consumer;
 
+    /**
+     * @var PoisonPillReadInterface|\PHPUnit_Framework_MockObject_MockObject
+     */
+    private $poisonPillRead;
+
+    /**
+     * @var PoisonPillCompareInterface|\PHPUnit_Framework_MockObject_MockObject
+     */
+    private $poisonPillCompare;
+
     /**
      * Set up.
      *
@@ -85,8 +97,15 @@ protected function setUp()
             ->disableOriginalConstructor()->getMock();
 
         $objectManager = new \Magento\Framework\TestFramework\Unit\Helper\ObjectManager($this);
+        $this->poisonPillCompare = $this->getMockBuilder(PoisonPillCompareInterface::class)
+            ->disableOriginalConstructor()->getMock();
+        $this->poisonPillRead = $this->getMockBuilder(PoisonPillReadInterface::class)
+            ->disableOriginalConstructor()->getMock();
         //Hard dependency used because CallbackInvoker invokes closure logic defined inside of Customer class.
-        $this->callbackInvoker = new \Magento\Framework\MessageQueue\CallbackInvoker();
+        $this->callbackInvoker = new \Magento\Framework\MessageQueue\CallbackInvoker(
+            $this->poisonPillRead,
+            $this->poisonPillCompare
+        );
         $this->consumer = $objectManager->getObject(
             \Magento\Framework\MessageQueue\Consumer::class,
             [
@@ -134,7 +153,8 @@ public function testProcessWithNotFoundException()
         $numberOfMessages = 1;
         $consumerName = 'consumer.name';
         $exceptionPhrase = new Phrase('Exception successfully thrown');
-
+        $this->poisonPillRead->expects($this->atLeastOnce())->method('getLatestVersion')->willReturn('version-1');
+        $this->poisonPillCompare->expects($this->atLeastOnce())->method('isLatestVersion')->willReturn(true);
         $queue = $this->getMockBuilder(\Magento\Framework\MessageQueue\QueueInterface::class)
             ->disableOriginalConstructor()->getMock();
         $this->configuration->expects($this->once())->method('getQueue')->willReturn($queue);