public function testParseOnTableColumns()
{
$column1 = m::mock('Frozennode\\Administrator\\DataTable\\Columns\\Column');
$column1->shouldReceive('renderOutput')->once()->andReturn('rendered');
$column2 = m::mock('Frozennode\\Administrator\\DataTable\\Columns\\Column');
$column2->shouldReceive('renderOutput')->once()->andReturn('rendered');
$columns = array('column1' => $column1, 'column2' => $column2);
$this->columnFactory->shouldReceive('getColumns')->once()->andReturn($columns);
$this->columnFactory->shouldReceive('getIncludedColumns')->once()->andReturn(array('column1' => 'foo', 'key_column' => 'flerp'));
$this->columnFactory->shouldReceive('getRelatedColumns')->once()->andReturn(array('column2' => 'bar'));
$this->fieldFactory->shouldReceive('getEditFields')->once()->andReturn(array());
$model = m::mock('Illuminate\\Database\\Eloquent\\Model');
$model->shouldReceive('getAttribute')->times(3)->andReturn('raw');
$outputRow = array();
$regColumnOutput = array('raw' => 'raw', 'rendered' => 'rendered');
$testOutput = array('column1' => $regColumnOutput, 'column2' => $regColumnOutput, 'key_column' => array('raw' => 'raw', 'rendered' => 'raw'));
$this->dataTable->parseOnTableColumns($model, $outputRow);
$this->assertEquals($outputRow, $testOutput);
}