Smart Endpoint Unit Tests

MichaelJ2324 · Mar 23, 2017 · e7ba8d1 · e7ba8d1
1 parent 2db83c8
commit e7ba8d1
Show file tree

Hide file tree

Showing 3 changed files with 31 additions and 2 deletions.
diff --git a/src/Endpoint/Abstracts/AbstractSmartEndpoint.php b/src/Endpoint/Abstracts/AbstractSmartEndpoint.php
@@ -3,7 +3,7 @@
 namespace MRussell\REST\Endpoint\Abstracts;
 
 use MRussell\REST\Endpoint\Data\AbstractEndpointData;
-use MRussell\REST\Exception\Endpoint\EndpointException;
+use MRussell\REST\Exception\Endpoint\InvalidDataType;
 
 abstract class AbstractSmartEndpoint extends AbstractEndpoint
 {
@@ -80,7 +80,7 @@ public function setData($data) {
             $this->data->reset();
             $this->data->update($data);
         } else {
-            throw new EndpointException("Invalid data passed to Endpoint");
+            throw new InvalidDataType(get_class($this));
         }
         return $this;
     }

diff --git a/src/Exception/Endpoint/InvalidDataType.php b/src/Exception/Endpoint/InvalidDataType.php
@@ -0,0 +1,9 @@
+<?php
+
+namespace MRussell\REST\Exception\Endpoint;
+
+
+class InvalidDataType extends EndpointException
+{
+    protected $message = 'Invalid data type passed to Endpoint [%s]';
+}
diff --git a/tests/Endpoint/AbstractSmartEndpointTest.php b/tests/Endpoint/AbstractSmartEndpointTest.php
@@ -2,6 +2,7 @@
 
 namespace MRussell\REST\Tests\Endpoint;
 
+use MRussell\REST\Endpoint\Data\EndpointData;
 use MRussell\REST\Tests\Stubs\Endpoint\SmartEndpoint;
 
 
@@ -117,11 +118,30 @@ public function testSetProperties(){
 
     /**
      * @covers ::setData
+     * @covers ::getData
      * @covers ::configureData
      */
     public function testSetData(){
         $Endpoint = new SmartEndpoint();
+        $Data = new EndpointData();
+        $this->assertEquals($Endpoint,$Endpoint->setData($Data));
+        $this->assertEquals($Data,$Endpoint->getData());
 
+        $this->assertEquals($Endpoint,$Endpoint->setData(array('foo' => 'bar')));
+        $this->assertEquals(array('foo' => 'bar'),$Endpoint->getData()->asArray());
+
+        $Class = new \ReflectionClass('MRussell\REST\Endpoint\Abstracts\AbstractSmartEndpoint');
+        $method = $Class->getMethod('configureData');
+        $method->setAccessible(TRUE);
+        $this->assertEquals(array('foo' => 'bar'),$method->invoke($Endpoint,$Endpoint->getData()));
+    }
+
+    /**
+     * @expectedException MRussell\REST\Exception\Endpoint\InvalidDataType
+     */
+    public function testInvalidDataType(){
+        $Endpoint = new SmartEndpoint();
+        $Endpoint->setData('test');
     }
 
 }