Allow Queue to be constructed with MongoCollection

Author: chadicus

src/Queue.php

@@ -25,16 +25,21 @@ final class Queue implements QueueInterface
     /**
      * Construct queue.
      *
-     * @param string $url the mongo url
+     * @param \MongoCollection|string $collectionOrUrl A MongoCollection instance or the mongo connection url.
      * @param string $db the mongo db name
      * @param string $collection the collection name to use for the queue
      *
-     * @throws \InvalidArgumentException $url, $db or $collection was not a string
+     * @throws \InvalidArgumentException $collectionOrUrl, $db or $collection was not a string
      */
-    public function __construct($url, $db, $collection)
+    public function __construct($collectionOrUrl, $db = null, $collection = null)
     {
-        if (!is_string($url)) {
-            throw new \InvalidArgumentException('$url was not a string');
+        if ($collectionOrUrl instanceof \MongoCollection) {
+            $this->collection = $collectionOrUrl;
+            return;
+        }
+
+        if (!is_string($collectionOrUrl)) {
+            throw new \InvalidArgumentException('$collectionOrUrl was not a string');
         }
 
         if (!is_string($db)) {
@@ -45,7 +50,7 @@ public function __construct($url, $db, $collection)
             throw new \InvalidArgumentException('$collection was not a string');
         }
 
-        $mongo = new \MongoClient($url);
+        $mongo = new \MongoClient($collectionOrUrl);
         $mongoDb = $mongo->selectDB($db);
         $this->collection = $mongoDb->selectCollection($collection);
     }

tests/QueueTest.php

@@ -852,4 +852,44 @@ public function sendWithLowEarliestGet()
 
         $this->assertSame($expected, $message);
     }
+
+    /**
+     * Verify Queue can be constructed with \MongoCollection
+     *
+     * @test
+     * @covers ::__construct
+     *
+     * @return void
+     */
+    public function constructWithCollection()
+    {
+        $mongo = new \MongoClient($this->mongoUrl);
+        $collection = $mongo->selectDB('testing')->selectCollection('custom_collection');
+        $collection->drop();
+        $queue = new Queue($collection);
+
+        $payload = ['key1' => 0, 'key2' => true];
+        $queue->send($payload, 34, 0.8);
+
+        $expected = [
+            'payload' => $payload,
+            'running' => false,
+            'resetTimestamp' => Queue::MONGO_INT32_MAX,
+            'earliestGet' => 34,
+            'priority' => 0.8,
+        ];
+
+        $this->assertSame(1, $collection->count());
+
+        $message = $collection->findOne();
+
+        $this->assertLessThanOrEqual(time(), $message['created']->sec);
+        $this->assertGreaterThan(time() - 10, $message['created']->sec);
+
+        unset($message['_id'], $message['created']);
+        $message['resetTimestamp'] = $message['resetTimestamp']->sec;
+        $message['earliestGet'] = $message['earliestGet']->sec;
+
+        $this->assertSame($expected, $message);
+    }
 }