Issue magento#35796 Update Unit test

artskirk · Jul 27, 2022 · 75a5913 · 75a5913
1 parent c296e63
commit 75a5913
Showing 1 changed file with 10 additions and 2 deletions.
diff --git a/...le/Test/Unit/Model/ResourceModel/Indexer/Price/DisabledProductOptionPriceModifierTest.php b/...le/Test/Unit/Model/ResourceModel/Indexer/Price/DisabledProductOptionPriceModifierTest.php
@@ -71,14 +71,22 @@ protected function setUp(): void
     public function testGetWebsiteIdsOfProduct(): void
     {
         $selectMock = $this->createMock(Select::class);
+        $statementMock = $this->createMock(\Zend_Db_Statement_Pdo::class);
+        $statementMock->method('fetchColumn')->willReturn(1);
         $selectMock->method('from')->willReturnSelf();
         $selectMock->method('where')->willReturnSelf();
-        $selectMock->method('query')->willReturnSelf();
+        $selectMock->method('query')->willReturn($statementMock);
+
         $this->connectionMock->method('select')
             ->willReturn($selectMock);
 
+        $this->connectionMock->expects($this->once())
+            ->method('fetchCol')
+            ->with($selectMock)
+            ->willReturn([]);
+
         /** @var IndexTableStructure|MockObject $indexTableStructure */
         $indexTableStructure = $this->createMock(IndexTableStructure::class);
-        $this->priceModifier->modifyPrice($indexTableStructure, []);
+        $this->assertEmpty($this->priceModifier->modifyPrice($indexTableStructure, [1]));
     }
 }