MAGETWO-91694: Product does not get removed from flat product table after getting disabled

- Fix units

Author: Yauhen_Lyskavets

app/code/Magento/Catalog/Test/Unit/Model/Indexer/Product/Flat/Action/EraserTest.php

@@ -54,6 +54,7 @@ public function testRemoveDeletedProducts()
         $productsToDeleteIds = [1, 2];
         $select = $this->createMock(\Magento\Framework\DB\Select::class);
         $select->expects($this->once())->method('from')->with('catalog_product_entity')->will($this->returnSelf());
+        $select->expects($this->once())->method('columns')->with('entity_id')->will($this->returnSelf());
         $select->expects($this->once())->method('where')->with('entity_id IN(?)', $productsToDeleteIds)
             ->will($this->returnSelf());
         $products = [['entity_id' => 2]];