Added tests for ColumnCollection

Author: pierredup

src/CS/DataGridBundle/Tests/Grid/Column/ColumnCollectionTest.php

@@ -0,0 +1,51 @@
+<?php
+
+/*
+ * This file is part of the CSDataGridBundle package.
+ *
+ * (c) Pierre du Plessis <[email protected]>
+ *
+ * For the full copyright and license information, please view the LICENSE
+ * file that was distributed with this source code.
+ */
+
+namespace CS\DataGridBundle\Tests\Grid\Column;
+
+use CS\DataGridBundle\Grid\Column\Column;
+use CS\DataGridBundle\Grid\Column\ColumnCollection;
+
+class ColumnCollectionTest extends \PHPUnit_Framework_TestCase {
+
+	protected $stack;
+
+	public function setUp()
+	{
+		$this->collection = new ColumnCollection;
+
+		$this->collection->addRecursive(array('foo', 'bar', 'baz'));
+	}
+
+	public function testCollection()
+	{
+		$elements = $this->collection->all();
+
+		$this->assertCount(3, $elements);
+
+		unset($elements);
+
+		$this->assertTrue($this->collection->has('foo'));
+		$this->assertTrue($this->collection->has('bar'));
+		$this->assertTrue($this->collection->has('baz'));
+
+		$this->assertInstanceOf('CS\DataGridBundle\Grid\Column\ColumnInterface', $this->collection->get('foo'));
+		$this->assertInstanceOf('CS\DataGridBundle\Grid\Column\ColumnInterface', $this->collection->get('bar'));
+		$this->assertInstanceOf('CS\DataGridBundle\Grid\Column\ColumnInterface', $this->collection->get('baz'));
+
+		$this->collection->remove('bar');
+
+		$elements = $this->collection->all();
+
+		$this->assertCount(2, $elements);
+		$this->assertFalse($this->collection->has('bar'));
+	}
+}