Merge pull request #179 from TheKvist/time-tracking-features

Time tracking features

Author: m4tthumphrey

lib/Gitlab/Api/Issues.php

@@ -127,6 +127,44 @@ public function removeComment($project_id, $issue_id, $note_id)
         return $this->delete($this->getProjectPath($project_id, 'issues/'.$this->encodePath($issue_id).'/notes/'.$this->encodePath($note_id)));
     }
 
+    /**
+     * @param int $project_id
+     * @param int $issue_id
+     * @param string $duration
+     */
+    public function setTimeEstimate($project_id, $issue_id, $duration)
+    {
+        return $this->post($this->getProjectPath($project_id, 'issues/'.$this->encodePath($issue_id).'/time_estimate'), array('duration' => $duration));
+    }
+
+    /**
+     * @param int $project_id
+     * @param int $issue_id
+     */
+    public function resetTimeEstimate($project_id, $issue_id)
+    {
+        return $this->post($this->getProjectPath($project_id, 'issues/'.$this->encodePath($issue_id).'/reset_time_estimate'));
+    }
+
+    /**
+     * @param int $project_id
+     * @param int $issue_id
+     * @param string $duration
+     */
+    public function addSpentTime($project_id, $issue_id, $duration)
+    {
+        return $this->post($this->getProjectPath($project_id, 'issues/'.$this->encodePath($issue_id).'/add_spent_time'), array('duration' => $duration));
+    }
+
+    /**
+     * @param int $project_id
+     * @param int $issue_id
+     */
+    public function resetSpentTime($project_id, $issue_id)
+    {
+        return $this->post($this->getProjectPath($project_id, 'issues/'.$this->encodePath($issue_id).'/reset_spent_time'));
+    }
+
     /**
      * @param int $project_id
      * @param int $issue_iid

test/Gitlab/Tests/Api/IssuesTest.php

@@ -187,9 +187,72 @@ public function shouldUpdateComment()
         $this->assertEquals($expectedArray, $api->updateComment(1, 2, 3, 'An edited comment'));
     }
 
-    protected function getApiClass()
+    /**
+     * @test
+     */
+    public function shouldSetTimeEstimate()
     {
-        return 'Gitlab\Api\Issues';
+        $expectedArray = array('time_estimate' => 14400, 'total_time_spent' => 0, 'human_time_estimate' => '4h', 'human_total_time_spent' => null);
+
+        $api = $this->getApiMock();
+        $api->expects($this->once())
+            ->method('post')
+            ->with('projects/1/issues/2/time_estimate', array('duration' => '4h'))
+            ->will($this->returnValue($expectedArray))
+        ;
+
+        $this->assertEquals($expectedArray, $api->setTimeEstimate(1, 2, '4h'));
+    }
+
+    /**
+     * @test
+     */
+    public function shouldResetTimeEstimate()
+    {
+        $expectedArray = array('time_estimate' => 0, 'total_time_spent' => 0, 'human_time_estimate' => null, 'human_total_time_spent' => null);
+
+        $api = $this->getApiMock();
+        $api->expects($this->once())
+            ->method('post')
+            ->with('projects/1/issues/2/reset_time_estimate')
+            ->will($this->returnValue($expectedArray))
+        ;
+
+        $this->assertEquals($expectedArray, $api->resetTimeEstimate(1, 2));
+    }
+
+    /**
+     * @test
+     */
+    public function shouldAddSpentTime()
+    {
+        $expectedArray = array('time_estimate' => 0, 'total_time_spent' => 14400, 'human_time_estimate' => null, 'human_total_time_spent' => '4h');
+
+        $api = $this->getApiMock();
+        $api->expects($this->once())
+            ->method('post')
+            ->with('projects/1/issues/2/add_spent_time', array('duration' => '4h'))
+            ->will($this->returnValue($expectedArray))
+        ;
+
+        $this->assertEquals($expectedArray, $api->addSpentTime(1, 2, '4h'));
+    }
+
+    /**
+     * @test
+     */
+    public function shouldResetSpentTime()
+    {
+        $expectedArray = array('time_estimate' => 0, 'total_time_spent' => 0, 'human_time_estimate' => null, 'human_total_time_spent' => null);
+
+        $api = $this->getApiMock();
+        $api->expects($this->once())
+            ->method('post')
+            ->with('projects/1/issues/2/reset_spent_time')
+            ->will($this->returnValue($expectedArray))
+        ;
+
+        $this->assertEquals($expectedArray, $api->resetSpentTime(1, 2));
     }
 
     /**
@@ -208,4 +271,9 @@ public function shouldGetIssueTimeStats()
 
         $this->assertEquals($expectedArray, $api->getTimeStats(1, 2));
     }
+
+    protected function getApiClass()
+    {
+        return 'Gitlab\Api\Issues';
+    }
 }