/**
* Gets mock object for query builder
*
* @param array $dqlParts
* @return \PHPUnit_Framework_MockObject_MockObject
*/
protected function getQueryBuilder(array $dqlParts = [])
{
$qb = $this->getMockBuilder('Doctrine\\ORM\\QueryBuilder')->disableOriginalConstructor()->disableOriginalClone()->getMock();
$select = new Select();
$select->add('t0.test as c1');
$dqlParts[] = ['select', [$select]];
$qb->expects($this->any())->method('getDQLPart')->will($this->returnValueMap($dqlParts));
return $qb;
}
/**
* @param string $alias
* @return Expr\Select|null
*/
public function getSelect($alias)
{
$expr = new Expr\Select();
foreach ($this->fits as $fit) {
if ($fit instanceof SelectFitInterface) {
$expr->add($fit->getSelect($alias));
}
}
return $expr->count() != 0 ? $expr : null;
}
