Add QueueAwareTrait

Author: chadicus

src/QueueAwareTrait.php

@@ -0,0 +1,39 @@
+<?php
+
+namespace DominionEnterprises\Mongo;
+
+/**
+ * Trait for QueueAwareInterface implementation.
+ */
+trait QueueAwareTrait
+{
+    /**
+     * The mongoQueue instance.
+     *
+     * @var QueueInterface
+     */
+    private $mongoQueue;
+
+    /**
+     * Returns the Queue instance.
+     *
+     * @return QueueInterface
+     */
+    public function getQueue()
+    {
+        return $this->mongoQueue;
+    }
+
+    /**
+     * Sets the Queue instance.
+     *
+     * @param QueueInterface $mongoQueue
+     *
+     * @return QueueAwareInterface
+     */
+    public function setQueue(QueueInterface $mongoQueue)
+    {
+        $this->mongoQueue = $mongoQueue;
+        return $this;
+    }
+}

tests/QueueAwareTraitTest.php

@@ -0,0 +1,39 @@
+<?php
+namespace DominionEnterprisesTest\Mongo;
+
+class QueueAwareTraitTest extends \PHPUnit_Framework_TestCase
+{
+    /**
+     * Verify basic behavior of setQueue().
+     *
+     * @test
+     * @covers \DominionEnterprises\Mongo\QueueAwareTrait
+     *
+     * @return void
+     */
+    public function setQueue()
+    {
+        $mockQueue = $this->getObjectForTrait('\DominionEnterprises\Mongo\QueueAwareTrait');
+        $this->assertAttributeEquals(null, 'mongoQueue', $mockQueue);
+        $mongoQueue = $this->getMock('\\DominionEnterprises\\Mongo\\QueueInterface');
+        $mockQueue->setQueue($mongoQueue);
+        $this->assertAttributeEquals($mongoQueue, 'mongoQueue', $mockQueue);
+    }
+
+    /**
+     * Verify basic behavior of getQueue().
+     *
+     * @test
+     * @covers \DominionEnterprises\Mongo\QueueAwareTrait
+     *
+     * @return void
+     */
+    public function getQueue()
+    {
+        $mockQueue = $this->getObjectForTrait('\DominionEnterprises\Mongo\QueueAwareTrait');
+        $this->assertNull($mockQueue->getQueue());
+        $mongoQueue = $this->getMock('\\DominionEnterprises\\Mongo\\QueueInterface');
+        $mockQueue->setQueue($mongoQueue);
+        $this->assertEquals($mongoQueue, $mockQueue->getQueue());
+    }
+}