public function testSetColumn() { $grid = new Datagrid(); $this->assertEquals(array(), $grid->getColumns()); $col = $this->getMockForAbstractClass('Zf2datatable\\Column\\AbstractColumn'); $col->setUniqueId('myUniqueId'); $col2 = $this->getMockForAbstractClass('Zf2datatable\\Column\\AbstractColumn'); $col2->setUniqueId('myUniqueId2'); $grid->setColumns(array($col, $col2)); $this->assertCount(2, $grid->getColumns()); $this->assertEquals($col, $grid->getColumnByUniqueId('myUniqueId')); $this->assertEquals($col2, $grid->getColumnByUniqueId('myUniqueId2')); }