ACP2E-3387: [Cloud] Magento: queue message has been deleted

Author: aplapana

app/code/Magento/MysqlMq/Test/Unit/Model/ResourceModel/QueueTest.php

@@ -261,7 +261,6 @@ function ($arg1, $arg2 = null) use ($queueName, $select) {
      */
     public function testDeleteMarkedMessages()
     {
-        $messageIds = [1, 2];
         $tableNames = ['queue_message_status', 'queue_message'];
         $connection = $this->getMockBuilder(AdapterInterface::class)
             ->disableOriginalConstructor()
@@ -285,12 +284,17 @@ function ($arg1, $arg2) use ($tableNames) {
         $select->expects($this->once())
             ->method('from')->with(['queue_message_status' => $tableNames[0]], ['message_id'])->willReturnSelf();
         $select->expects($this->once())->method('where')
-            ->with('status <> ?', QueueManagement::MESSAGE_STATUS_TO_BE_DELETED)
+            ->with('status = ?', QueueManagement::MESSAGE_STATUS_TO_BE_DELETED)
             ->willReturnSelf();
-        $select->expects($this->once())->method('distinct')->willReturnSelf();
-        $connection->expects($this->once())->method('fetchCol')->with($select)->willReturn($messageIds);
+        $select->expects($this->once())
+            ->method('assemble')
+            ->willReturn('SELECT message_id FROM queue_message_status WHERE status = 7');
+
         $connection->expects($this->once())->method('delete')
-            ->with($tableNames[1], ['id NOT IN (?)' => $messageIds])->willReturn(2);
+            ->with(
+                $tableNames[1],
+                'id IN (SELECT message_id FROM queue_message_status WHERE status = 7)'
+            )->willReturn(2);
         $this->queue->deleteMarkedMessages();
     }