1 files changed, 41 insertions, 1 deletions

diff --git a/tests/functionals/ApiTest.php b/tests/functionals/ApiTest.php
index b5039759..92c752bb 100644
--- a/tests/functionals/ApiTest.php
+++ b/tests/functionals/ApiTest.php
@@ -937,7 +937,7 @@ class Api extends PHPUnit_Framework_TestCase
 
     public function testCreateFile()
     {
-        $this->assertTrue($this->client->createFile(1, 1, 'My file', false, base64_encode('plain text file')));
+        $this->assertEquals(1, $this->client->createFile(1, 1, 'My file', base64_encode('plain text file')));
     }
 
     public function testGetAllFiles()
@@ -962,4 +962,44 @@ class Api extends PHPUnit_Framework_TestCase
         $this->assertTrue($this->client->removeFile($file['id']));
         $this->assertEmpty($this->client->getAllFiles(1));
     }
+
+    public function testRemoveAllFiles()
+    {
+        $this->assertEquals(1, $this->client->createFile(1, 1, 'My file 1', base64_encode('plain text file')));
+        $this->assertEquals(2, $this->client->createFile(1, 1, 'My file 2', base64_encode('plain text file')));
+
+        $files = $this->client->getAllFiles(array('task_id' => 1));
+        $this->assertNotEmpty($files);
+        $this->assertCount(2, $files);
+
+        $this->assertTrue($this->client->removeAllFiles(array('task_id' => 1)));
+
+        $files = $this->client->getAllFiles(array('task_id' => 1));
+        $this->assertEmpty($files);
+    }
+
+    public function testCreateTaskWithReference()
+    {
+        $task = array(
+            'title' => 'Task with external ticket number',
+            'reference' => 'TICKET-1234',
+            'project_id' => 1,
+            'description' => '[Link to my ticket](http://my-ticketing-system/1234)',
+        );
+
+        $task_id = $this->client->createTask($task);
+
+        $this->assertNotFalse($task_id);
+        $this->assertInternalType('int', $task_id);
+        $this->assertTrue($task_id > 0);
+    }
+
+    public function testGetTaskByReference()
+    {
+        $task = $this->client->getTaskByReference(array('project_id' => 1, 'reference' => 'TICKET-1234'));
+
+        $this->assertNotEmpty($task);
+        $this->assertEquals('Task with external ticket number', $task['title']);
+        $this->assertEquals('TICKET-1234', $task['reference']);
+    }
 }
\ No newline at end of file