public function testRetrievingDataHandler() { $dataGrid = new DataGrid($env = $this->getMockEnvironment(), array('bar'), array()); $env->shouldReceive('getDataHandlerMappings')->andReturn(array('DataHandlerStub' => function (array $data) { return count($data) == 1; })); $this->assertInstanceOf('DataHandlerStub', $dataGrid->createDataHandler()); $this->assertInstanceOf('DataHandlerStub', $dataGrid->getDataHandler()); }