--- /dev/null
+<?php
+
+namespace Doctrine\Tests\ORM\Hydration;
+
+use Doctrine\ORM\Query\ResultSetMapping;
+use Doctrine\ORM\Mapping\ClassMetadata;
+
+require_once __DIR__ . '/../../TestInit.php';
+
+/**
+ * Description of ResultSetMappingTest
+ *
+ * @author robo
+ */
+class ResultSetMappingTest extends \Doctrine\Tests\OrmTestCase
+{
+ /**
+ * @var ResultSetMapping
+ */
+ private $_rsm;
+
+ /**
+ * @var \Doctrine\ORM\EntityManager
+ */
+ private $_em;
+
+ protected function setUp() {
+ parent::setUp();
+ $this->_rsm = new ResultSetMapping;
+ $this->_em = $this->_getTestEntityManager();
+ }
+
+ /**
+ * For SQL: SELECT id, status, username, name FROM cms_users
+ */
+ public function testBasicResultSetMapping()
+ {
+ $this->_rsm->addEntityResult(
+ 'Doctrine\Tests\Models\CMS\CmsUser',
+ 'u'
+ );
+ $this->_rsm->addFieldResult('u', 'id', 'id');
+ $this->_rsm->addFieldResult('u', 'status', 'status');
+ $this->_rsm->addFieldResult('u', 'username', 'username');
+ $this->_rsm->addFieldResult('u', 'name', 'name');
+
+ $this->assertFalse($this->_rsm->isScalarResult('id'));
+ $this->assertFalse($this->_rsm->isScalarResult('status'));
+ $this->assertFalse($this->_rsm->isScalarResult('username'));
+ $this->assertFalse($this->_rsm->isScalarResult('name'));
+
+ $this->assertTrue($this->_rsm->getClassName('u') == 'Doctrine\Tests\Models\CMS\CmsUser');
+ $class = $this->_rsm->getDeclaringClass('id');
+ $this->assertTrue($class == 'Doctrine\Tests\Models\CMS\CmsUser');
+
+ $this->assertEquals('u', $this->_rsm->getEntityAlias('id'));
+ $this->assertEquals('u', $this->_rsm->getEntityAlias('status'));
+ $this->assertEquals('u', $this->_rsm->getEntityAlias('username'));
+ $this->assertEquals('u', $this->_rsm->getEntityAlias('name'));
+
+ $this->assertEquals('id', $this->_rsm->getFieldName('id'));
+ $this->assertEquals('status', $this->_rsm->getFieldName('status'));
+ $this->assertEquals('username', $this->_rsm->getFieldName('username'));
+ $this->assertEquals('name', $this->_rsm->getFieldName('name'));
+ }
+
+ /**
+ * @group DDC-1057
+ *
+ * Fluent interface test, not a real result set mapping
+ */
+ public function testFluentInterface()
+ {
+ $rms = $this->_rsm;
+
+ $rms->addEntityResult('Doctrine\Tests\Models\CMS\CmsUser','u')
+ ->addJoinedEntityResult('Doctrine\Tests\Models\CMS\CmsPhonenumber','p','u','phonenumbers')
+ ->addFieldResult('u', 'id', 'id')
+ ->addFieldResult('u', 'name', 'name')
+ ->setDiscriminatorColumn('name', 'name')
+ ->addIndexByColumn('id', 'id')
+ ->addIndexBy('username', 'username')
+ ->addIndexByScalar('sclr0')
+ ->addScalarResult('sclr0', 'numPhones')
+ ->addMetaResult('a', 'user_id', 'user_id');
+
+
+ $this->assertTrue($rms->hasIndexBy('id'));
+ $this->assertTrue($rms->isFieldResult('id'));
+ $this->assertTrue($rms->isFieldResult('name'));
+ $this->assertTrue($rms->isScalarResult('sclr0'));
+ $this->assertTrue($rms->isRelation('p'));
+ $this->assertTrue($rms->hasParentAlias('p'));
+ $this->assertTrue($rms->isMixedResult());
+ }
+
+ /**
+ * @group DDC-1663
+ */
+ public function testAddNamedNativeQueryResultSetMapping()
+ {
+ $cm = new ClassMetadata('Doctrine\Tests\Models\CMS\CmsUser');
+ $cm->initializeReflection(new \Doctrine\Common\Persistence\Mapping\RuntimeReflectionService);
+
+ $cm->mapOneToOne(array(
+ 'fieldName' => 'email',
+ 'targetEntity' => 'Doctrine\Tests\Models\CMS\CmsEmail',
+ 'cascade' => array('persist'),
+ 'inversedBy' => 'user',
+ 'orphanRemoval' => false,
+ 'joinColumns' => array(array(
+ 'nullable' => true,
+ 'referencedColumnName' => 'id',
+ )
+ )
+ ));
+
+ $cm->addNamedNativeQuery(array(
+ 'name' => 'find-all',
+ 'query' => 'SELECT u.id AS user_id, e.id AS email_id, u.name, e.email, u.id + e.id AS scalarColumn FROM cms_users u INNER JOIN cms_emails e ON e.id = u.email_id',
+ 'resultSetMapping' => 'find-all',
+ ));
+ $cm->addSqlResultSetMapping(array(
+ 'name' => 'find-all',
+ 'entities' => array(
+ array(
+ 'entityClass' => '__CLASS__',
+ 'fields' => array(
+ array(
+ 'name' => 'id',
+ 'column'=> 'user_id'
+ ),
+ array(
+ 'name' => 'name',
+ 'column'=> 'name'
+ )
+ )
+ ),
+ array(
+ 'entityClass' => 'CmsEmail',
+ 'fields' => array(
+ array(
+ 'name' => 'id',
+ 'column'=> 'email_id'
+ ),
+ array(
+ 'name' => 'email',
+ 'column'=> 'email'
+ )
+ )
+ )
+ ),
+ 'columns' => array(
+ array(
+ 'name' => 'scalarColumn'
+ )
+ )
+ ));
+
+
+ $queryMapping = $cm->getNamedNativeQuery('find-all');
+
+ $rsm = new \Doctrine\ORM\Query\ResultSetMappingBuilder($this->_em);
+ $rsm->addNamedNativeQueryMapping($cm, $queryMapping);
+
+ $this->assertEquals('scalarColumn', $rsm->getScalarAlias('scalarColumn'));
+
+ $this->assertEquals('c0', $rsm->getEntityAlias('user_id'));
+ $this->assertEquals('c0', $rsm->getEntityAlias('name'));
+ $this->assertEquals('Doctrine\Tests\Models\CMS\CmsUser', $rsm->getClassName('c0'));
+ $this->assertEquals('Doctrine\Tests\Models\CMS\CmsUser', $rsm->getDeclaringClass('name'));
+ $this->assertEquals('Doctrine\Tests\Models\CMS\CmsUser', $rsm->getDeclaringClass('user_id'));
+
+
+ $this->assertEquals('c1', $rsm->getEntityAlias('email_id'));
+ $this->assertEquals('c1', $rsm->getEntityAlias('email'));
+ $this->assertEquals('Doctrine\Tests\Models\CMS\CmsEmail', $rsm->getClassName('c1'));
+ $this->assertEquals('Doctrine\Tests\Models\CMS\CmsEmail', $rsm->getDeclaringClass('email'));
+ $this->assertEquals('Doctrine\Tests\Models\CMS\CmsEmail', $rsm->getDeclaringClass('email_id'));
+ }
+
+ /**
+ * @group DDC-1663
+ */
+ public function testAddNamedNativeQueryResultSetMappingWithoutFields()
+ {
+ $cm = new ClassMetadata('Doctrine\Tests\Models\CMS\CmsUser');
+ $cm->initializeReflection(new \Doctrine\Common\Persistence\Mapping\RuntimeReflectionService);
+
+ $cm->addNamedNativeQuery(array(
+ 'name' => 'find-all',
+ 'query' => 'SELECT u.id AS user_id, e.id AS email_id, u.name, e.email, u.id + e.id AS scalarColumn FROM cms_users u INNER JOIN cms_emails e ON e.id = u.email_id',
+ 'resultSetMapping' => 'find-all',
+ ));
+ $cm->addSqlResultSetMapping(array(
+ 'name' => 'find-all',
+ 'entities' => array(
+ array(
+ 'entityClass' => '__CLASS__',
+ )
+ ),
+ 'columns' => array(
+ array(
+ 'name' => 'scalarColumn'
+ )
+ )
+ ));
+
+
+ $queryMapping = $cm->getNamedNativeQuery('find-all');
+
+ $rsm = new \Doctrine\ORM\Query\ResultSetMappingBuilder($this->_em);
+ $rsm->addNamedNativeQueryMapping($cm, $queryMapping);
+
+ $this->assertEquals('scalarColumn', $rsm->getScalarAlias('scalarColumn'));
+
+ $this->assertEquals('c0', $rsm->getEntityAlias('id'));
+ $this->assertEquals('c0', $rsm->getEntityAlias('name'));
+ $this->assertEquals('c0', $rsm->getEntityAlias('status'));
+ $this->assertEquals('c0', $rsm->getEntityAlias('username'));
+ $this->assertEquals('Doctrine\Tests\Models\CMS\CmsUser', $rsm->getClassName('c0'));
+ $this->assertEquals('Doctrine\Tests\Models\CMS\CmsUser', $rsm->getDeclaringClass('id'));
+ $this->assertEquals('Doctrine\Tests\Models\CMS\CmsUser', $rsm->getDeclaringClass('name'));
+ $this->assertEquals('Doctrine\Tests\Models\CMS\CmsUser', $rsm->getDeclaringClass('status'));
+ $this->assertEquals('Doctrine\Tests\Models\CMS\CmsUser', $rsm->getDeclaringClass('username'));
+ }
+
+ /**
+ * @group DDC-1663
+ */
+ public function testAddNamedNativeQueryResultClass()
+ {
+ $cm = new ClassMetadata('Doctrine\Tests\Models\CMS\CmsUser');
+ $cm->initializeReflection(new \Doctrine\Common\Persistence\Mapping\RuntimeReflectionService);
+
+ $cm->addNamedNativeQuery(array(
+ 'name' => 'find-all',
+ 'resultClass' => '__CLASS__',
+ 'query' => 'SELECT * FROM cms_users',
+ ));
+
+ $queryMapping = $cm->getNamedNativeQuery('find-all');
+
+ $rsm = new \Doctrine\ORM\Query\ResultSetMappingBuilder($this->_em);
+ $rsm->addNamedNativeQueryMapping($cm, $queryMapping);
+
+
+ $this->assertEquals('c0', $rsm->getEntityAlias('id'));
+ $this->assertEquals('c0', $rsm->getEntityAlias('name'));
+ $this->assertEquals('c0', $rsm->getEntityAlias('status'));
+ $this->assertEquals('c0', $rsm->getEntityAlias('username'));
+ $this->assertEquals('Doctrine\Tests\Models\CMS\CmsUser', $rsm->getClassName('c0'));
+ $this->assertEquals('Doctrine\Tests\Models\CMS\CmsUser', $rsm->getDeclaringClass('id'));
+ $this->assertEquals('Doctrine\Tests\Models\CMS\CmsUser', $rsm->getDeclaringClass('name'));
+ $this->assertEquals('Doctrine\Tests\Models\CMS\CmsUser', $rsm->getDeclaringClass('status'));
+ $this->assertEquals('Doctrine\Tests\Models\CMS\CmsUser', $rsm->getDeclaringClass('username'));
+ }
+}
+