1 files changed, 190 insertions, 0 deletions
diff --git a/tests/units/Model/FileTest.php b/tests/units/Model/FileTest.php
new file mode 100644
index 00000000..da00917d
--- /dev/null
+++ b/tests/units/Model/FileTest.php
@@ -0,0 +1,190 @@
+<?php
+
+require_once __DIR__.'/../Base.php';
+
+use Model\Task;
+use Model\File;
+use Model\TaskCreation;
+use Model\Project;
+
+class FileTest extends Base
+{
+    public function testCreation()
+    {
+        $p = new Project($this->container);
+        $f = new File($this->container);
+        $tc = new TaskCreation($this->container);
+
+        $this->assertEquals(1, $p->create(array('name' => 'test')));
+        $this->assertEquals(1, $tc->create(array('project_id' => 1, 'title' => 'test')));
+
+        $this->assertEquals(1, $f->create(1, 'test', '/tmp/foo', 10));
+
+        $file = $f->getById(1);
+        $this->assertNotEmpty($file);
+        $this->assertEquals('test', $file['name']);
+        $this->assertEquals('/tmp/foo', $file['path']);
+        $this->assertEquals(0, $file['is_image']);
+        $this->assertEquals(1, $file['task_id']);
+        $this->assertEquals(time(), $file['date'], '', 2);
+        $this->assertEquals(0, $file['user_id']);
+        $this->assertEquals(10, $file['size']);
+
+        $this->assertEquals(2, $f->create(1, 'test2.png', '/tmp/foobar', 10));
+
+        $file = $f->getById(2);
+        $this->assertNotEmpty($file);
+        $this->assertEquals('test2.png', $file['name']);
+        $this->assertEquals('/tmp/foobar', $file['path']);
+        $this->assertEquals(1, $file['is_image']);
+    }
+
+    public function testCreationFileNameTooLong()
+    {
+        $p = new Project($this->container);
+        $f = new File($this->container);
+        $tc = new TaskCreation($this->container);
+
+        $this->assertEquals(1, $p->create(array('name' => 'test')));
+        $this->assertEquals(1, $tc->create(array('project_id' => 1, 'title' => 'test')));
+
+        $this->assertNotFalse($f->create(1, 'test', '/tmp/foo', 10));
+        $this->assertNotFalse($f->create(1, str_repeat('a', 1000), '/tmp/foo', 10));
+
+        $files = $f->getAll(1);
+        $this->assertNotEmpty($files);
+        $this->assertCount(2, $files);
+
+        $this->assertEquals(str_repeat('a', 255), $files[0]['name']);
+        $this->assertEquals('test', $files[1]['name']);
+    }
+
+    public function testIsImage()
+    {
+        $f = new File($this->container);
+
+        $this->assertTrue($f->isImage('test.png'));
+        $this->assertTrue($f->isImage('test.jpeg'));
+        $this->assertTrue($f->isImage('test.gif'));
+        $this->assertTrue($f->isImage('test.jpg'));
+        $this->assertTrue($f->isImage('test.JPG'));
+
+        $this->assertFalse($f->isImage('test.bmp'));
+        $this->assertFalse($f->isImage('test'));
+        $this->assertFalse($f->isImage('test.pdf'));
+    }
+
+    public function testGeneratePath()
+    {
+        $f = new File($this->container);
+
+        $this->assertStringStartsWith('12/34/', $f->generatePath(12, 34, 'test.png'));
+        $this->assertNotEquals($f->generatePath(12, 34, 'test1.png'), $f->generatePath(12, 34, 'test2.png'));
+    }
+
+    public function testUploadScreenshot()
+    {
+        $p = new Project($this->container);
+        $f = new File($this->container);
+        $tc = new TaskCreation($this->container);
+
+        $this->assertEquals(1, $p->create(array('name' => 'test')));
+        $this->assertEquals(1, $tc->create(array('project_id' => 1, 'title' => 'test')));
+
+        $this->assertEquals(1, $f->uploadScreenshot(1, 1, base64_encode('image data')));
+
+        $file = $f->getById(1);
+        $this->assertNotEmpty($file);
+        $this->assertStringStartsWith('Screenshot taken ', $file['name']);
+        $this->assertStringStartsWith('1/1/', $file['path']);
+        $this->assertEquals(1, $file['is_image']);
+        $this->assertEquals(1, $file['task_id']);
+        $this->assertEquals(time(), $file['date'], '', 2);
+        $this->assertEquals(0, $file['user_id']);
+        $this->assertEquals(10, $file['size']);
+    }
+
+    public function testUploadFileContent()
+    {
+        $p = new Project($this->container);
+        $f = new File($this->container);
+        $tc = new TaskCreation($this->container);
+
+        $this->assertEquals(1, $p->create(array('name' => 'test')));
+        $this->assertEquals(1, $tc->create(array('project_id' => 1, 'title' => 'test')));
+
+        $this->assertEquals(1, $f->uploadContent(1, 1, 'my file.pdf', base64_encode('file data')));
+
+        $file = $f->getById(1);
+        $this->assertNotEmpty($file);
+        $this->assertEquals('my file.pdf', $file['name']);
+        $this->assertStringStartsWith('1/1/', $file['path']);
+        $this->assertEquals(0, $file['is_image']);
+        $this->assertEquals(1, $file['task_id']);
+        $this->assertEquals(time(), $file['date'], '', 2);
+        $this->assertEquals(0, $file['user_id']);
+        $this->assertEquals(9, $file['size']);
+    }
+
+    public function testGetAll()
+    {
+        $p = new Project($this->container);
+        $f = new File($this->container);
+        $tc = new TaskCreation($this->container);
+
+        $this->assertEquals(1, $p->create(array('name' => 'test')));
+        $this->assertEquals(1, $tc->create(array('project_id' => 1, 'title' => 'test')));
+
+        $this->assertEquals(1, $f->create(1, 'B.pdf', '/tmp/foo', 10));
+        $this->assertEquals(2, $f->create(1, 'A.png', '/tmp/foo', 10));
+        $this->assertEquals(3, $f->create(1, 'D.doc', '/tmp/foo', 10));
+        $this->assertEquals(4, $f->create(1, 'C.JPG', '/tmp/foo', 10));
+
+        $files = $f->getAll(1);
+        $this->assertNotEmpty($files);
+        $this->assertCount(4, $files);
+        $this->assertEquals('A.png', $files[0]['name']);
+        $this->assertEquals('B.pdf', $files[1]['name']);
+        $this->assertEquals('C.JPG', $files[2]['name']);
+        $this->assertEquals('D.doc', $files[3]['name']);
+
+        $files = $f->getAllImages(1);
+        $this->assertNotEmpty($files);
+        $this->assertCount(2, $files);
+        $this->assertEquals('A.png', $files[0]['name']);
+        $this->assertEquals('C.JPG', $files[1]['name']);
+
+        $files = $f->getAllDocuments(1);
+        $this->assertNotEmpty($files);
+        $this->assertCount(2, $files);
+        $this->assertEquals('B.pdf', $files[0]['name']);
+        $this->assertEquals('D.doc', $files[1]['name']);
+    }
+
+    public function testRemove()
+    {
+        $p = new Project($this->container);
+        $f = new File($this->container);
+        $tc = new TaskCreation($this->container);
+
+        $this->assertEquals(1, $p->create(array('name' => 'test')));
+        $this->assertEquals(1, $tc->create(array('project_id' => 1, 'title' => 'test')));
+
+        $this->assertEquals(1, $f->create(1, 'B.pdf', '/tmp/foo', 10));
+        $this->assertEquals(2, $f->create(1, 'A.png', '/tmp/foo', 10));
+        $this->assertEquals(3, $f->create(1, 'D.doc', '/tmp/foo', 10));
+
+        $this->assertTrue($f->remove(2));
+
+        $files = $f->getAll(1);
+        $this->assertNotEmpty($files);
+        $this->assertCount(2, $files);
+        $this->assertEquals('B.pdf', $files[0]['name']);
+        $this->assertEquals('D.doc', $files[1]['name']);
+
+        $this->assertTrue($f->removeAll(1));
+
+        $files = $f->getAll(1);
+        $this->assertEmpty($files);
+    }
+}