Merge pull request #118 from lastzero/develop

Added getData() method to Basho\Riak\Search\Doc

Author: christophermancini

src/Riak/Search/Doc.php

@@ -44,13 +44,34 @@ public function __construct(\stdClass $data)
         $this->data = $data;
     }
 
+    /**
+     * Returns object location
+     *
+     * @return Location
+     */
     public function getLocation()
     {
         return new Location($this->_yz_rk, new Bucket($this->_yz_rb, $this->_yz_rt));
     }
 
+    /**
+     * Returns a single value from Solr result document
+     *
+     * @param string $name
+     * @return mixed
+     */
     public function __get($name)
     {
         return $this->data->{$name};
     }
+
+    /**
+     * Returns all values as array from Solr result document
+     *
+     * @return array
+     */
+    public function getData()
+    {
+        return (array)$this->data;
+    }
 }

tests/unit/Riak/Search/DocTest.php

@@ -0,0 +1,55 @@
+<?php
+
+namespace Basho\Tests\Riak\Search;
+
+use Basho\Riak\Search\Doc;
+use PHPUnit_Framework_TestCase as TestCase;
+
+/**
+ * Search result document test
+ *
+ * @author Michael Mayer <[email protected]>
+ */
+class DocTest extends TestCase
+{
+    /**
+     * @var Doc
+     */
+    protected $doc;
+
+    public function setUp()
+    {
+        $data = new \stdClass();
+        $data->_yz_id = '1*tests*test*5*39';
+        $data->_yz_rk = '5';
+        $data->_yz_rt = 'tests';
+        $data->_yz_rb = 'test';
+        $data->foo = 'bar';
+        $data->_status = 200;
+        $this->doc = new Doc($data);
+    }
+
+    public function testGetLocation()
+    {
+        $result = $this->doc->getLocation();
+        $this->assertInstanceOf('\Basho\Riak\Location', $result);
+        $this->assertInstanceOf('\Basho\Riak\Bucket', $result->getBucket());
+        $this->assertEquals('tests', $result->getBucket()->getType());
+        $this->assertEquals('test', $result->getBucket()->getName());
+        $this->assertEquals('5', $result->getKey());
+    }
+
+    public function testGetData()
+    {
+        $result = $this->doc->getData();
+        $this->assertInternalType('array', $result);
+        $this->assertEquals('bar', $result['foo']);
+        $this->assertEquals(200, $result['_status']);
+    }
+
+    public function testMagicGetter()
+    {
+        $this->assertEquals('bar', $this->doc->foo);
+        $this->assertEquals(200, $this->doc->_status);
+    }
+}