Merge pull request #16 from maxwellhealth/mongo-adapter

Adds MongoDB Storage Adapter

Author: richardfullmer

.travis.yml

@@ -1,5 +1,6 @@
 language: php
-
+services:
+  - mongodb
 php:
   - 5.6
   - 7.0
@@ -11,5 +12,5 @@ matrix:
   allow_failures:
     - php: nightly
     - php: hhvm
-
+before_script: echo "extension = mongo.so" >> ~/.phpenv/versions/$(phpenv version-name)/etc/php.ini
 install: composer install -n

README.md

@@ -43,10 +43,12 @@ There are a number of different storage implementations for where the configurat
 * DoctrineCacheStorageAdapter - requires [doctrine/cache][doctrine-cache]
 * PDOStorageAdapter - persistent using [PDO][pdo]
 * RedisStorageAdapter - persistent using [Redis][redis]
+* MongoDBStorageAdapter - persistent using [Mongo][mongo]
 
 [doctrine-cache]: https://packagist.org/packages/doctrine/cache
 [pdo]: http://php.net/pdo
 [redis]: http://redis.io
+[mongo]: http://mongodb.org
 
 All storage adapters must implement `Opensoft\Rollout\Storage\StorageInterface`.
 

src/Storage/MongoDBStorageAdapter.php

@@ -0,0 +1,65 @@
+<?php
+
+namespace Opensoft\Rollout\Storage;
+
+/**
+ * Storage adapter using MongoDB
+ *
+ * @author James Hrisho <@securingsincity>
+ */
+class MongoDBStorageAdapter implements StorageInterface
+{
+
+    /**
+     * @var object
+     */
+    private $mongo;
+
+    /**
+     * @var string
+     */
+    private $collection;
+
+    public function __construct($mongo, $collection = "rollout_feature")
+    {
+        $this->mongo = $mongo;
+        $this->collection = $collection;
+    }
+    public function getCollectionName()
+    {
+        return $this->collection;
+    }
+    /**
+     * @inheritdoc
+     */
+    public function get($key)
+    {
+        $collection = $this->getCollectionName();
+        $result = $this->mongo->$collection->findOne(['name' => $key]);
+
+        if (!$result) {
+            return null;
+        }
+
+        return $result['value'];
+    }
+
+    /**
+     * @inheritdoc
+     */
+    public function set($key, $value)
+    {
+        $collection = $this->getCollectionName();
+        $this->mongo->$collection->update(['name' => $key], ['$set' => ['value' => $value]], ['upsert' => true]);
+    }
+
+    /**
+     * @inheritdoc
+     */
+    public function remove($key)
+    {
+        $collection = $this->getCollectionName();
+        $this->mongo->$collection->remove(['name' => $key]);
+    }
+
+}

tests/Storage/MongoDBStorageAdapterTest.php

@@ -0,0 +1,91 @@
+<?php
+
+namespace Opensoft\Tests\Storage;
+
+use Opensoft\Rollout\Storage\MongoDBStorageAdapter;
+
+class MongoDBStorageAdapterTest extends \PHPUnit_Framework_TestCase
+{
+    private $mongo;
+
+    public function setUp()
+    {
+        $this->mongo = new mockMongo();
+        $collection = $this->mockCollection();
+        $collection->method('findOne')->will($this->returnValue(['name' => 'key', 'value' => true]));
+        $collection->method('update')->will($this->returnValue(null));
+
+        $this->mongo->setCollection($collection);
+    }
+
+    public function testGet()
+    {
+
+        $adapter = new MongoDBStorageAdapter($this->mongo);
+
+        $result = $adapter->get('key');
+        $this->assertSame(true, $result);
+    }
+
+    public function testGetNoValue()
+    {
+
+        $adapter = new MongoDBStorageAdapter($this->mongo);
+        $this->mongo->coll->method('findOne')->will($this->returnValue(null));
+        $result = $adapter->get('key');
+        $this->assertSame(true, $result);
+    }
+
+    public function testSet()
+    {
+
+        $adapter = new MongoDBStorageAdapter($this->mongo);
+
+        $adapter->set('key', 'value');
+
+    }
+
+    public function testRemove()
+    {
+
+        $adapter = new MongoDBStorageAdapter($this->mongo);
+
+        $adapter->remove('key');
+
+    }
+
+    public function testGetCollectionName()
+    {
+
+        $adapter = new MongoDBStorageAdapter($this->mongo, 'feature_test');
+
+        $result = $adapter->getCollectionName();
+        $this->assertSame('feature_test', $result);
+
+    }
+
+    public function mockCollection()
+    {
+        return $this->createMock('MongoCollection', ['find', 'findOne', 'update', 'remove'], [], '', false);
+
+    }
+}
+
+class mockMongo
+{
+    public $collection;
+    public function __construct()
+    {
+
+    }
+
+    public function setCollection($mongoCollection)
+    {
+        $this->collection = $mongoCollection;
+    }
+
+    public function __get($name)
+    {
+        return $this->collection;
+    }
+}