public function testDataSourceDoctrineCollection()
{
$grid = new Datagrid();
$this->assertFalse($grid->hasDataSource());
$coll = $this->getMock('Doctrine\\Common\\Collections\\ArrayCollection', array(), array(), '', false);
$em = $this->getMock('Doctrine\\ORM\\EntityManager', array(), array(), '', false);
$grid->setDataSource($coll, $em);
$this->assertTrue($grid->hasDataSource());
$this->assertInstanceOf('Zf2datatable\\DataSource\\Doctrine2Collection', $grid->getDataSource());
$this->setExpectedException('InvalidArgumentException', 'If providing a Collection, also the Doctrine\\ORM\\EntityManager is needed as a second parameter');
$grid->setDataSource($coll);
}
