123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158 |
- <?php
-
-
- namespace Doctrine\Tests\ORM\Functional;
-
- use Doctrine\ORM\Query;
-
- require_once __DIR__ . '/../../TestInit.php';
-
-
- class CustomTreeWalkersTest extends \Doctrine\Tests\OrmFunctionalTestCase
- {
- protected function setUp() {
- $this->useModelSet('cms');
- parent::setUp();
- }
-
- public function assertSqlGeneration($dqlToBeTested, $sqlToBeConfirmed)
- {
- try {
- $query = $this->_em->createQuery($dqlToBeTested);
- $query->setHint(Query::HINT_CUSTOM_TREE_WALKERS, array('Doctrine\Tests\ORM\Functional\CustomTreeWalker'))
- ->useQueryCache(false);
-
- $this->assertEquals($sqlToBeConfirmed, $query->getSql());
- $query->free();
- } catch (\Exception $e) {
- $this->fail($e->getMessage() . ' at "' . $e->getFile() . '" on line ' . $e->getLine());
-
- }
- }
-
- public function testSupportsQueriesWithoutWhere()
- {
- $this->assertSqlGeneration(
- 'select u from Doctrine\Tests\Models\CMS\CmsUser u',
- "SELECT c0_.id AS id0, c0_.status AS status1, c0_.username AS username2, c0_.name AS name3 FROM cms_users c0_ WHERE c0_.id = 1"
- );
- }
-
- public function testSupportsQueriesWithMultipleConditionalExpressions()
- {
- $this->assertSqlGeneration(
- 'select u from Doctrine\Tests\Models\CMS\CmsUser u where u.name = :name or u.name = :otherName',
- "SELECT c0_.id AS id0, c0_.status AS status1, c0_.username AS username2, c0_.name AS name3 FROM cms_users c0_ WHERE (c0_.name = ? OR c0_.name = ?) AND c0_.id = 1"
- );
- }
-
- public function testSupportsQueriesWithSimpleConditionalExpression()
- {
- $this->assertSqlGeneration(
- 'select u from Doctrine\Tests\Models\CMS\CmsUser u where u.name = :name',
- "SELECT c0_.id AS id0, c0_.status AS status1, c0_.username AS username2, c0_.name AS name3 FROM cms_users c0_ WHERE c0_.name = ? AND c0_.id = 1"
- );
- }
- }
-
- class CustomTreeWalker extends Query\TreeWalkerAdapter
- {
- public function walkSelectStatement(Query\AST\SelectStatement $selectStatement)
- {
-
- $dqlAliases = array();
-
- foreach ($this->_getQueryComponents() as $dqlAlias => $comp) {
-
-
- if ($comp['metadata']->name == 'Doctrine\Tests\Models\CMS\CmsUser') {
- $dqlAliases[] = $dqlAlias;
- }
- }
-
-
- $factors = array();
- foreach ($dqlAliases as $alias) {
- $pathExpr = new Query\AST\PathExpression(Query\AST\PathExpression::TYPE_STATE_FIELD, $alias, 'id');
- $pathExpr->type = Query\AST\PathExpression::TYPE_STATE_FIELD;
- $comparisonExpr = new Query\AST\ComparisonExpression($pathExpr, '=', 1);
-
- $condPrimary = new Query\AST\ConditionalPrimary;
- $condPrimary->simpleConditionalExpression = $comparisonExpr;
-
- $factor = new Query\AST\ConditionalFactor($condPrimary);
- $factors[] = $factor;
- }
-
- if (($whereClause = $selectStatement->whereClause) !== null) {
-
- $condExpr = $whereClause->conditionalExpression;
-
-
- if ( ! ($condExpr instanceof Query\AST\ConditionalExpression)) {
- $condExpr = new Query\AST\ConditionalExpression(array($condExpr));
-
- $whereClause->conditionalExpression = $condExpr;
- }
-
- $existingTerms = $whereClause->conditionalExpression->conditionalTerms;
-
- if (count($existingTerms) > 1) {
-
-
-
- $primary = new Query\AST\ConditionalPrimary;
- $primary->conditionalExpression = new Query\AST\ConditionalExpression($existingTerms);
- $existingFactor = new Query\AST\ConditionalFactor($primary);
- $term = new Query\AST\ConditionalTerm(array_merge(array($existingFactor), $factors));
-
- $selectStatement->whereClause->conditionalExpression->conditionalTerms = array($term);
- } else {
-
- $singleTerm = $selectStatement->whereClause->conditionalExpression->conditionalTerms[0];
-
-
- if ( ! ($singleTerm instanceof Query\AST\ConditionalTerm)) {
- $singleTerm = new Query\AST\ConditionalTerm(array($singleTerm));
-
- $selectStatement->whereClause->conditionalExpression->conditionalTerms[0] = $singleTerm;
- }
-
- $singleTerm->conditionalFactors = array_merge($singleTerm->conditionalFactors, $factors);
- $selectStatement->whereClause->conditionalExpression->conditionalTerms = array($singleTerm);
- }
- } else {
-
- $term = new Query\AST\ConditionalTerm($factors);
- $condExpr = new Query\AST\ConditionalExpression(array($term));
- $whereClause = new Query\AST\WhereClause($condExpr);
- $selectStatement->whereClause = $whereClause;
- }
- }
- }
|