1 files changed, 92 insertions, 0 deletions
diff --git a/tests/PHPUnit/Unit/Category/CategoryListTest.php b/tests/PHPUnit/Unit/Category/CategoryListTest.php
new file mode 100644
index 0000000000..53f5d9c42c
--- /dev/null
+++ b/tests/PHPUnit/Unit/Category/CategoryListTest.php
@@ -0,0 +1,92 @@
+<?php
+/**
+ * Piwik - free/libre analytics platform
+ *
+ * @link http://piwik.org
+ * @license http://www.gnu.org/licenses/gpl-3.0.html GPL v3 or later
+ */
+
+namespace Piwik\Tests\Unit\Category;
+
+use Piwik\Category\CategoryList;
+use Piwik\Category\Category;
+
+/**
+ * @group Category
+ * @group CategoryList
+ * @group CategoryListTest
+ */
+class CategoryListTest extends \PHPUnit_Framework_TestCase
+{
+    /**
+     * @var CategoryList
+     */
+    private $categoryList;
+
+    public function setUp()
+    {
+        $this->categoryList = new CategoryList();
+    }
+
+    public function test_getCategories_IsEmptyByDefault()
+    {
+        $this->assertSame(array(), $this->categoryList->getCategories());
+    }
+
+    public function test_addCategory_ShouldAddCategory_and_getCategories_shouldBeIndexedById()
+    {
+        $category = $this->addCategory('myTest');
+
+        $this->assertSame(array('myTest' => $category), $this->categoryList->getCategories());
+    }
+
+    /**
+     * @expectedException \Exception
+     * @expectedExceptionMessage Category myTest already exists
+     */
+    public function test_addCategory_shouldThrowException_IfAddingSameCategoryIdTwice()
+    {
+        $this->addCategory('myTest');
+        $this->addCategory('myTest');
+    }
+
+    public function test_hasCategory_ShouldDetectIfCategoryExists()
+    {
+        $this->assertFalse($this->categoryList->hasCategory('myTest'));
+
+        $this->addCategory('myTest');
+
+        $this->assertTrue($this->categoryList->hasCategory('myTest'));
+
+        $this->assertFalse($this->categoryList->hasCategory('myTest2'));
+        $this->assertFalse($this->categoryList->hasCategory('General_Visits'));
+    }
+
+    public function test_getCategory_ShouldReturnExistingCategoryIfPossible()
+    {
+        $this->assertNull($this->categoryList->getCategory('myTest'));
+
+        $category = $this->addCategory('myTest');
+
+        $this->assertSame($category, $this->categoryList->getCategory('myTest'));
+
+        $this->assertNull($this->categoryList->getCategory('myTest2'));
+        $this->assertNull($this->categoryList->getCategory('General_Visits'));
+    }
+
+    private function addCategory($id)
+    {
+        $category = $this->createCategory($id);
+        $this->categoryList->addCategory($category);
+
+        return $category;
+    }
+
+    private function createCategory($categoryId)
+    {
+        $config = new Category();
+        $config->setId($categoryId);
+
+        return $config;
+    }
+}