public function testDeleteMessage() { $uniq = mt_rand(); $queue = 'test_queue'; $message1 = $this->getTestMessage($uniq); // Store a second message for a good time, and make sure we delete the // right one. $message2 = $this->getTestMessage($uniq); $this->db->storeMessage($message1, $queue); // store message 2 with a $this->db->storeMessage($message2, $queue); // Confirm work without using the API. $pdo = $this->db->getDatabase(); $result = $pdo->query("\n\t\t\tSELECT * FROM damaged\n\t\t\tWHERE gateway='test'\n\t\t\t\tAND order_id = '{$message1['order_id']}'"); $rows = $result->fetchAll(PDO::FETCH_ASSOC); $this->assertEquals(2, count($rows), 'Both records were stored.'); $this->assertNotNull($rows[0]['id'], 'Record includes a primary row id'); $this->assertNotEquals($rows[0]['id'], $rows[1]['id'], 'Records have unique primary ids'); $message2['damaged_id'] = $rows[1]['id']; $this->db->deleteMessage($message2); // Confirm work without using the API. $pdo = $this->db->getDatabase(); $result = $pdo->query("\n\t\t\tSELECT * FROM damaged\n\t\t\tWHERE gateway='test'\n\t\t\t\tAND order_id = '{$message1['order_id']}'"); $rowsAfter = $result->fetchAll(PDO::FETCH_ASSOC); $this->assertEquals(1, count($rowsAfter), 'Not only one row deleted.'); $this->assertEquals($rowsAfter[0]['id'], $rows[0]['id'], 'Deleted the wrong row.'); }
/** * Do the actual work of the script. */ public function execute() { $this->damagedDatabase = DamagedDatabase::get(); $messages = $this->damagedDatabase->fetchRetryMessages($this->getOption('max-messages')); $stats = array(); $config = Configuration::getDefaultConfig(); foreach ($messages as $message) { $queueName = $message['original_queue']; // FIXME: getting it by alias, this will be annoying cos -new $queue = BaseQueueConsumer::getQueue($queueName); unset($message['original_queue']); $queue->push($message); $this->damagedDatabase->deleteMessage($message); if (isset($stats[$queueName])) { $stats[$queueName]++; } else { $stats[$queueName] = 1; } } foreach ($stats as $queueName => $count) { Logger::info("Requeued {$count} messages to {$queueName}."); } }