Add a method reject and ask for a revision (#68)

worldia · May 30, 2016 · 3b511c9 · 3b511c9
1 parent 39ca62c
commit 3b511c9
Show file tree

Hide file tree

Showing 3 changed files with 46 additions and 1 deletion.
diff --git a/lib/Textmaster/Model/Document.php b/lib/Textmaster/Model/Document.php
@@ -264,6 +264,17 @@ final public function complete($satisfaction = null, $message = null)
         return $this;
     }
 
+    /**
+     * {@inheritdoc}
+     */
+    public function reject($message)
+    {
+        $this->getApi()->supportMessages()->create($this->getProject()->getId(), $this->getId(), $message);
+        $this->dispatchEvent(DocumentInterface::STATUS_INCOMPLETE, $this->data);
+
+        return $this;
+    }
+
     /**
      * {@inheritdoc}
      */

diff --git a/lib/Textmaster/Model/DocumentInterface.php b/lib/Textmaster/Model/DocumentInterface.php
@@ -180,6 +180,15 @@ public function setCustomData($customData, $key = null);
      */
     public function complete($satisfaction = null, $message = null);
 
+    /**
+     * Ask for a revision on the document.
+     *
+     * @param string $message
+     *
+     * @return DocumentInterface
+     */
+    public function reject($message);
+
     /**
      * Get allowed status.
      *

diff --git a/test/Textmaster/Unit/Model/DocumentTest.php b/test/Textmaster/Unit/Model/DocumentTest.php
@@ -62,7 +62,8 @@ public function setUp()
 
         $clientMock = $this->getMockBuilder('Textmaster\Client')->setMethods(array('projects'))->disableOriginalConstructor()->getMock();
         $projectApiMock = $this->getMock('Textmaster\Api\Project', array('documents', 'show'), array($clientMock));
-        $documentApiMock = $this->getMock('Textmaster\Api\Document', array('show', 'update', 'complete'), array($clientMock, $projectId), '', false);
+        $documentApiMock = $this->getMock('Textmaster\Api\Document', array('show', 'update', 'complete', 'supportMessages'), array($clientMock, $projectId), '', false);
+        $supportMessageApiMock = $this->getMock('Textmaster\Api\Project\Document\SupportMessage', array('create'), array($clientMock), '', false);
 
         $clientMock->method('projects')
             ->willReturn($projectApiMock);
@@ -82,6 +83,12 @@ public function setUp()
         $documentApiMock->method('complete')
             ->willReturn($completeValues);
 
+        $documentApiMock->method('supportMessages')
+            ->willReturn($supportMessageApiMock);
+
+        $supportMessageApiMock->method('create')
+            ->willReturn(null);
+
         $this->clientMock = $clientMock;
     }
 
@@ -324,6 +331,24 @@ public function shouldComplete()
         $this->assertSame(DocumentInterface::STATUS_COMPLETED, $document->getStatus());
     }
 
+    /**
+     * @test
+     */
+    public function shouldReject()
+    {
+        $projectId = '654321';
+        $valuesToCreate = array(
+            'id' => '123456',
+            'project_id' => $projectId,
+        );
+
+        $document = new Document($this->clientMock, $valuesToCreate);
+        $document->save();
+        $document->reject('Bad job!');
+
+        $this->assertSame('123456', $document->getId());
+    }
+
     /**
      * @test
      * @expectedException \Textmaster\Exception\InvalidArgumentException