/**
* Check if the message is requeued or not correctly.
*
* @dataProvider processMessageProvider
*/
public function testProcessMessage($processFlag, $expectedMethod, $expectedRequeue = null)
{
$amqpConnection = $this->getMockBuilder('\\PhpAmqpLib\\Connection\\AMQPConnection')->disableOriginalConstructor()->getMock();
$amqpChannel = $this->getMockBuilder('\\PhpAmqpLib\\Channel\\AMQPChannel')->disableOriginalConstructor()->getMock();
$consumer = new MultipleConsumer($amqpConnection, $amqpChannel);
$callback = function ($msg) use(&$lastQueue, $processFlag) {
return $processFlag;
};
$consumer->setQueues(array('test-1' => array('callback' => $callback), 'test-2' => array('callback' => $callback)));
// Create a default message
$amqpMessage = new AMQPMessage('foo body');
$amqpMessage->delivery_info['channel'] = $amqpChannel;
$amqpMessage->delivery_info['delivery_tag'] = 0;
$amqpChannel->expects($this->any())->method('basic_reject')->will($this->returnCallback(function ($delivery_tag, $requeue) use($expectedMethod, $expectedRequeue) {
\PHPUnit_Framework_Assert::assertSame($expectedMethod, 'basic_reject');
// Check if this function should be called.
\PHPUnit_Framework_Assert::assertSame($requeue, $expectedRequeue);
// Check if the message should be requeued.
}));
$amqpChannel->expects($this->any())->method('basic_ack')->will($this->returnCallback(function ($delivery_tag) use($expectedMethod) {
\PHPUnit_Framework_Assert::assertSame($expectedMethod, 'basic_ack');
// Check if this function should be called.
}));
$consumer->processQueueMessage('test-1', $amqpMessage);
$consumer->processQueueMessage('test-2', $amqpMessage);
}
/**
* Check queues provider works well with static queues together
*
* @dataProvider processMessageProvider
*/
public function testQueuesProviderAndStaticQueuesTogether($processFlag, $expectedMethod, $expectedRequeue = null)
{
$callback = $this->prepareCallback($processFlag);
$this->multipleConsumer->setQueues(array('test-1' => array('callback' => $callback), 'test-2' => array('callback' => $callback)));
$queuesProvider = $this->prepareQueuesProvider();
$queuesProvider->expects($this->once())->method('getQueues')->will($this->returnValue(array('test-3' => array('callback' => $callback), 'test-4' => array('callback' => $callback))));
$this->multipleConsumer->setQueuesProvider($queuesProvider);
/**
* We don't test consume method, which merges queues by calling $this->setupConsumer();
* So we need to invoke it manually
*/
$reflectionClass = new \ReflectionClass(get_class($this->multipleConsumer));
$reflectionMethod = $reflectionClass->getMethod('mergeQueues');
$reflectionMethod->setAccessible(true);
$reflectionMethod->invoke($this->multipleConsumer);
$this->prepareAMQPChannelExpectations($expectedMethod, $expectedRequeue);
// Create a default message
$amqpMessage = new AMQPMessage('foo body');
$amqpMessage->delivery_info['channel'] = $this->amqpChannel;
$amqpMessage->delivery_info['delivery_tag'] = 0;
$this->multipleConsumer->processQueueMessage('test-1', $amqpMessage);
$this->multipleConsumer->processQueueMessage('test-2', $amqpMessage);
$this->multipleConsumer->processQueueMessage('test-3', $amqpMessage);
$this->multipleConsumer->processQueueMessage('test-4', $amqpMessage);
}
