Merge pull request #3 from robcaw/feature-max-priority

Add max priority to queue

src/Broker/Queue.php

@@ -20,6 +20,11 @@ class Queue implements QueueInterface
      */
     protected $deadLetterExchange;
 
+    /**
+     * @var int
+     */
+    protected $maxPriority;
+
     public function __construct(Connection $connection, $name, $durable = true, array $bindings)
     {
         $this->connection = $connection;
@@ -48,6 +53,16 @@ public function setDeadLetterExchange($deadLetterExchange)
         $this->deadLetterExchange = $deadLetterExchange;
     }
 
+    public function getMaxPriority()
+    {
+        return $this->maxPriority;
+    }
+
+    public function setMaxPriority($maxPriority)
+    {
+        $this->maxPriority = $maxPriority;
+    }
+
     public function declareQueue()
     {
         $this->getAMQPQueue()->declareQueue();
@@ -105,6 +120,10 @@ private function getAMQPQueue()
             if ($this->getDeadLetterExchange() !== null) {
                 $this->queue->setArgument('x-dead-letter-exchange', $this->getDeadLetterExchange());
             }
+
+            if ($this->getMaxPriority() !== null) {
+                $this->queue->setArgument('x-max-priority', $this->getMaxPriority());
+            }
         }
 
         return $this->queue;

src/Broker/QueueRegistry.php

@@ -52,6 +52,11 @@ private function create($name)
             $queue->setDeadLetterExchange($config['dead_letter_exchange']);
         }
 
+        // Set max priority
+        if (isset($config['max_priority'])) {
+            $queue->setMaxPriority($config['max_priority']);
+        }
+
         return $queue;
     }
 

tests/Broker/QueueTest.php

@@ -0,0 +1,60 @@
+<?php
+
+namespace Radish\Broker;
+
+use AMQPQueue;
+use Mockery;
+use Mockery\Mock;
+use PHPUnit_Framework_TestCase;
+
+class QueueTest extends PHPUnit_Framework_TestCase
+{
+    /**
+     * @var Queue
+     */
+    private $queue;
+    /**
+     * @var Mock|Connection
+     */
+    private $connection;
+    /**
+     * @var Mock|AMQPQueue
+     */
+    private $amqpQueue;
+
+    public function setUp()
+    {
+        $this->amqpQueue = Mockery::mock('AMQPQueue', [
+            'declareQueue' => null,
+            'setName' => null,
+            'setFlags' => null,
+        ]);
+
+        $this->connection = Mockery::mock('Radish\Broker\Connection', [
+            'createQueue' => $this->amqpQueue,
+        ]);
+
+        $this->queue = new Queue($this->connection, 'test_queue', true, []);
+    }
+
+    public function testDeclareQueueSetsMaxPriorityArgOnAmqpQueue()
+    {
+        $this->queue->setMaxPriority(100);
+
+        $this->amqpQueue->shouldReceive('setArgument')
+            ->with('x-max-priority', 100)
+            ->once();
+
+        $this->queue->declareQueue();
+    }
+
+    public function testDeclareQueueDoesntSetMaxPriorityOnAmqpWhenNull()
+    {
+        $this->queue->setMaxPriority(null);
+
+        $this->amqpQueue->shouldReceive('setArgument')
+            ->never();
+
+        $this->queue->declareQueue();
+    }
+}