/**
* @param \FSi\Bundle\AdminBundle\Admin\CRUD\ListElement $element
* @param \FSi\Component\DataSource\DataSource $datasource
* @param \FSi\Component\DataGrid\DataGrid $datagrid
* @param \FSi\Bundle\AdminBundle\Admin\Context\Request\HandlerInterface $handler
*/
function let($element, $datasource, $datagrid, $handler)
{
$this->beConstructedWith(array($handler));
$element->createDataGrid()->willReturn($datagrid);
$element->createDataSource()->willReturn($datasource);
$this->setElement($element);
}
/**
* {@inheritdoc}
*/
public function setElement(Element $element)
{
$this->element = $element;
$this->dataSource = $this->element->createDataSource();
$this->dataGrid = $this->element->createDataGrid();
}
/**
* @param \FSi\Bundle\AdminBundle\Admin\CRUD\ListElement $adminElement
* @return \FSi\Component\DataGrid\DataGrid
*/
protected function getDataGrid(ListElement $adminElement)
{
if (!array_key_exists($adminElement->getId(), $this->datagrids)) {
$this->datagrids[$adminElement->getId()] = $adminElement->createDataGrid();
}
return $this->datagrids[$adminElement->getId()];
}
