Added unit test for Category model

1 files changed, 186 insertions, 34 deletions

diff --git a/tests/units/Model/CategoryTest.php b/tests/units/Model/CategoryTest.php
index 3a64e26c..600007d0 100644
--- a/tests/units/Model/CategoryTest.php
+++ b/tests/units/Model/CategoryTest.php
@@ -2,6 +2,7 @@
 
 require_once __DIR__.'/../Base.php';
 
+use Kanboard\Model\Config;
 use Kanboard\Model\TaskCreation;
 use Kanboard\Model\TaskFinder;
 use Kanboard\Model\Project;
@@ -11,55 +12,206 @@ class CategoryTest extends Base
 {
     public function testCreation()
     {
-        $tc = new TaskCreation($this->container);
-        $tf = new TaskFinder($this->container);
-        $p = new Project($this->container);
-        $c = new Category($this->container);
-
-        $this->assertEquals(1, $p->create(array('name' => 'Project #1')));
-        $this->assertEquals(1, $c->create(array('name' => 'Category #1', 'project_id' => 1)));
-        $this->assertEquals(2, $c->create(array('name' => 'Category #2', 'project_id' => 1)));
-        $this->assertEquals(1, $tc->create(array('title' => 'Task #1', 'project_id' => 1, 'category_id' => 2)));
-
-        $task = $tf->getById(1);
-        $this->assertTrue(is_array($task));
+        $taskCreationModel = new TaskCreation($this->container);
+        $taskFinderModel = new TaskFinder($this->container);
+        $projectModel = new Project($this->container);
+        $categoryModel = new Category($this->container);
+
+        $this->assertEquals(1, $projectModel->create(array('name' => 'Project #1')));
+        $this->assertEquals(1, $categoryModel->create(array('name' => 'Category #1', 'project_id' => 1)));
+        $this->assertEquals(2, $categoryModel->create(array('name' => 'Category #2', 'project_id' => 1)));
+        $this->assertEquals(1, $taskCreationModel->create(array('title' => 'Task #1', 'project_id' => 1, 'category_id' => 2)));
+
+        $task = $taskFinderModel->getById(1);
         $this->assertEquals(2, $task['category_id']);
 
-        $category = $c->getById(2);
-        $this->assertTrue(is_array($category));
+        $category = $categoryModel->getById(2);
         $this->assertEquals(2, $category['id']);
         $this->assertEquals('Category #2', $category['name']);
         $this->assertEquals(1, $category['project_id']);
+    }
+
+    public function testExists()
+    {
+        $projectModel = new Project($this->container);
+        $categoryModel = new Category($this->container);
+
+        $this->assertEquals(1, $projectModel->create(array('name' => 'Project #1')));
+        $this->assertEquals(1, $categoryModel->create(array('name' => 'Category #1', 'project_id' => 1)));
+        $this->assertTrue($categoryModel->exists(1));
+        $this->assertFalse($categoryModel->exists(2));
+    }
+
+    public function testGetById()
+    {
+        $projectModel = new Project($this->container);
+        $categoryModel = new Category($this->container);
+
+        $this->assertEquals(1, $projectModel->create(array('name' => 'Project #1')));
+        $this->assertEquals(1, $categoryModel->create(array('name' => 'Category #1', 'project_id' => 1, 'description' => 'test')));
+
+        $category = $categoryModel->getById(1);
+        $this->assertEquals(1, $category['id']);
+        $this->assertEquals('Category #1', $category['name']);
+        $this->assertEquals(1, $category['project_id']);
+        $this->assertEquals('test', $category['description']);
+    }
+
+    public function testGetNameById()
+    {
+        $projectModel = new Project($this->container);
+        $categoryModel = new Category($this->container);
+
+        $this->assertEquals(1, $projectModel->create(array('name' => 'Project #1')));
+        $this->assertEquals(1, $categoryModel->create(array('name' => 'Category #1', 'project_id' => 1, 'description' => 'test')));
+
+        $this->assertEquals('Category #1', $categoryModel->getNameById(1));
+        $this->assertEquals('', $categoryModel->getNameById(2));
+    }
+
+    public function testGetIdByName()
+    {
+        $projectModel = new Project($this->container);
+        $categoryModel = new Category($this->container);
+
+        $this->assertEquals(1, $projectModel->create(array('name' => 'Project #1')));
+        $this->assertEquals(1, $categoryModel->create(array('name' => 'Category #1', 'project_id' => 1, 'description' => 'test')));
+
+        $this->assertSame(1, $categoryModel->getIdByName(1, 'Category #1'));
+        $this->assertSame(0, $categoryModel->getIdByName(1, 'Category #2'));
+    }
+
+    public function testGetList()
+    {
+        $projectModel = new Project($this->container);
+        $categoryModel = new Category($this->container);
+
+        $this->assertEquals(1, $projectModel->create(array('name' => 'Project #1')));
+        $this->assertEquals(1, $categoryModel->create(array('name' => 'Category #1', 'project_id' => 1, 'description' => 'test')));
+        $this->assertEquals(2, $categoryModel->create(array('name' => 'Category #2', 'project_id' => 1)));
+
+        $categories = $categoryModel->getList(1, false, false);
+        $this->assertCount(2, $categories);
+        $this->assertEquals('Category #1', $categories[1]);
+        $this->assertEquals('Category #2', $categories[2]);
+
+        $categories = $categoryModel->getList(1, true, false);
+        $this->assertCount(3, $categories);
+        $this->assertEquals('No category', $categories[0]);
+        $this->assertEquals('Category #1', $categories[1]);
+        $this->assertEquals('Category #2', $categories[2]);
 
-        $this->assertEquals(2, $c->getIdByName(1, 'Category #2'));
-        $this->assertEquals(0, $c->getIdByName(2, 'Category #2'));
+        $categories = $categoryModel->getList(1, false, true);
+        $this->assertCount(3, $categories);
+        $this->assertEquals('All categories', $categories[-1]);
+        $this->assertEquals('Category #1', $categories[1]);
+        $this->assertEquals('Category #2', $categories[2]);
 
-        $this->assertEquals('Category #2', $c->getNameById(2));
-        $this->assertEquals('', $c->getNameById(23));
+        $categories = $categoryModel->getList(1, true, true);
+        $this->assertCount(4, $categories);
+        $this->assertEquals('All categories', $categories[-1]);
+        $this->assertEquals('No category', $categories[0]);
+        $this->assertEquals('Category #1', $categories[1]);
+        $this->assertEquals('Category #2', $categories[2]);
+    }
+
+    public function testGetAll()
+    {
+        $projectModel = new Project($this->container);
+        $categoryModel = new Category($this->container);
+
+        $this->assertEquals(1, $projectModel->create(array('name' => 'Project #1')));
+        $this->assertEquals(1, $categoryModel->create(array('name' => 'Category #1', 'project_id' => 1, 'description' => 'test')));
+        $this->assertEquals(2, $categoryModel->create(array('name' => 'Category #2', 'project_id' => 1)));
+
+        $categories = $categoryModel->getAll(1);
+        $this->assertCount(2, $categories);
+
+        $this->assertEquals('Category #1', $categories[0]['name']);
+        $this->assertEquals('test', $categories[0]['description']);
+        $this->assertEquals(1, $categories[0]['project_id']);
+        $this->assertEquals(1, $categories[0]['id']);
+
+        $this->assertEquals('Category #2', $categories[1]['name']);
+        $this->assertEquals('', $categories[1]['description']);
+        $this->assertEquals(1, $categories[1]['project_id']);
+        $this->assertEquals(2, $categories[1]['id']);
+    }
+
+    public function testCreateDefaultCategories()
+    {
+        $projectModel = new Project($this->container);
+        $categoryModel = new Category($this->container);
+        $configModel = new Config($this->container);
+
+        $this->assertTrue($configModel->save(array('project_categories' => 'C1, C2, C3')));
+        $this->assertEquals(1, $projectModel->create(array('name' => 'Project #1')));
+        $this->assertTrue($categoryModel->createDefaultCategories(1));
+
+        $categories = $categoryModel->getAll(1);
+        $this->assertCount(3, $categories);
+        $this->assertEquals('C1', $categories[0]['name']);
+        $this->assertEquals('C2', $categories[1]['name']);
+        $this->assertEquals('C3', $categories[2]['name']);
+    }
+
+    public function testUpdate()
+    {
+        $projectModel = new Project($this->container);
+        $categoryModel = new Category($this->container);
+
+        $this->assertEquals(1, $projectModel->create(array('name' => 'Project #1')));
+        $this->assertEquals(1, $categoryModel->create(array('name' => 'Category #1', 'project_id' => 1)));
+        $this->assertTrue($categoryModel->update(array('id' => 1, 'description' => 'test')));
+
+        $category = $categoryModel->getById(1);
+        $this->assertEquals('Category #1', $category['name']);
+        $this->assertEquals(1, $category['project_id']);
+        $this->assertEquals('test', $category['description']);
     }
 
     public function testRemove()
     {
-        $tc = new TaskCreation($this->container);
-        $tf = new TaskFinder($this->container);
-        $p = new Project($this->container);
-        $c = new Category($this->container);
-
-        $this->assertEquals(1, $p->create(array('name' => 'Project #1')));
-        $this->assertEquals(1, $c->create(array('name' => 'Category #1', 'project_id' => 1)));
-        $this->assertEquals(2, $c->create(array('name' => 'Category #2', 'project_id' => 1)));
-        $this->assertEquals(1, $tc->create(array('title' => 'Task #1', 'project_id' => 1, 'category_id' => 2)));
-
-        $task = $tf->getById(1);
-        $this->assertTrue(is_array($task));
+        $taskCreationModel = new TaskCreation($this->container);
+        $taskFinderModel = new TaskFinder($this->container);
+        $projectModel = new Project($this->container);
+        $categoryModel = new Category($this->container);
+
+        $this->assertEquals(1, $projectModel->create(array('name' => 'Project #1')));
+        $this->assertEquals(1, $categoryModel->create(array('name' => 'Category #1', 'project_id' => 1)));
+        $this->assertEquals(2, $categoryModel->create(array('name' => 'Category #2', 'project_id' => 1)));
+        $this->assertEquals(1, $taskCreationModel->create(array('title' => 'Task #1', 'project_id' => 1, 'category_id' => 2)));
+
+        $task = $taskFinderModel->getById(1);
         $this->assertEquals(2, $task['category_id']);
 
-        $this->assertTrue($c->remove(1));
-        $this->assertTrue($c->remove(2));
+        $this->assertTrue($categoryModel->remove(1));
+        $this->assertTrue($categoryModel->remove(2));
 
         // Make sure tasks assigned with that category are reseted
-        $task = $tf->getById(1);
-        $this->assertTrue(is_array($task));
+        $task = $taskFinderModel->getById(1);
         $this->assertEquals(0, $task['category_id']);
     }
+
+    public function testDuplicate()
+    {
+        $projectModel = new Project($this->container);
+        $categoryModel = new Category($this->container);
+
+        $this->assertEquals(1, $projectModel->create(array('name' => 'Project #1')));
+        $this->assertEquals(2, $projectModel->create(array('name' => 'Project #2')));
+        $this->assertEquals(1, $categoryModel->create(array('name' => 'Category #1', 'project_id' => 1, 'description' => 'test')));
+
+        $this->assertTrue($categoryModel->duplicate(1, 2));
+
+        $category = $categoryModel->getById(1);
+        $this->assertEquals('Category #1', $category['name']);
+        $this->assertEquals(1, $category['project_id']);
+        $this->assertEquals('test', $category['description']);
+
+        $category = $categoryModel->getById(2);
+        $this->assertEquals('Category #1', $category['name']);
+        $this->assertEquals(2, $category['project_id']);
+        $this->assertEquals('test', $category['description']);
+    }
 }