doctrine2/tests/Doctrine/Tests/ORM/Functional/Ticket/DDC1335Test.php

<?php

namespace Doctrine\Tests\ORM\Functional\Ticket;

use DateTime;

require_once __DIR__ . '/../../../TestInit.php';

class DDC1135Test extends \Doctrine\Tests\OrmFunctionalTestCase
{
    protected function setUp()
    {
        parent::setUp();
        
        $classes = array(
            $this->_em->getClassMetadata(__NAMESPACE__ . '\DDC1135User'),
            $this->_em->getClassMetadata(__NAMESPACE__ . '\DDC1135Phone'),
        );
                
        try {
            $this->_schemaTool->dropSchema($classes);
            $this->_schemaTool->createSchema($classes);
        } catch(\Exception $e) {
            try {
                $this->_schemaTool->createSchema($classes);
            } catch (Exception $exc) {
                $this->fail($exc->getMessage());  
            }
        }
        $this->loadFixture();
    }
    
   
    public function testDql()
    {
        $dql      = 'SELECT u FROM ' . __NAMESPACE__ . '\DDC1135User u INDEX BY u.id';
        $query    = $this->_em->createQuery($dql);
        $result   = $query->getResult();
        
        $this->assertEquals(sizeof($result), 3);
        $this->assertArrayHasKey(1, $result);
        $this->assertArrayHasKey(2, $result);
        $this->assertArrayHasKey(3, $result);
        
        
        $dql      = 'SELECT u, p FROM '.__NAMESPACE__ . '\DDC1135User u INDEX BY u.email INNER JOIN u.phones p INDEX BY p.id';
        $query    = $this->_em->createQuery($dql);
        $result   = $query->getResult();
        
        $this->assertEquals(sizeof($result), 3);
        $this->assertArrayHasKey('foo@foo.com', $result);
        $this->assertArrayHasKey('bar@bar.com', $result);
        $this->assertArrayHasKey('foobar@foobar.com', $result);
        
        $this->assertEquals(sizeof($result['foo@foo.com']->phones), 3);
        $this->assertEquals(sizeof($result['bar@bar.com']->phones), 3);
        $this->assertEquals(sizeof($result['foobar@foobar.com']->phones), 3);
        
        $this->assertArrayHasKey(1, $result['foo@foo.com']->phones->toArray());
        $this->assertArrayHasKey(2, $result['foo@foo.com']->phones->toArray());
        $this->assertArrayHasKey(3, $result['foo@foo.com']->phones->toArray());
        
        $this->assertArrayHasKey(4, $result['bar@bar.com']->phones->toArray());
        $this->assertArrayHasKey(5, $result['bar@bar.com']->phones->toArray());
        $this->assertArrayHasKey(6, $result['bar@bar.com']->phones->toArray());
        
        $this->assertArrayHasKey(7, $result['foobar@foobar.com']->phones->toArray());
        $this->assertArrayHasKey(8, $result['foobar@foobar.com']->phones->toArray());
        $this->assertArrayHasKey(9, $result['foobar@foobar.com']->phones->toArray());
    }

    public function testTicket()
    {
        $builder = $this->_em->createQueryBuilder();
        $builder->select('u')->from(__NAMESPACE__ . '\DDC1135User', 'u', 'u.id');

        $dql    = $builder->getQuery()->getDQL();
        $result = $builder->getQuery()->getResult();
        
        $this->assertEquals(sizeof($result), 3);
        $this->assertArrayHasKey(1, $result);
        $this->assertArrayHasKey(2, $result);
        $this->assertArrayHasKey(3, $result);
        $this->assertEquals('SELECT u FROM Doctrine\Tests\ORM\Functional\Ticket\DDC1135User u INDEX BY u.id', $dql);
    }
    
    public function testIndexByUnique()
    {
        $builder = $this->_em->createQueryBuilder();
        $builder->select('u')->from(__NAMESPACE__ . '\DDC1135User', 'u', 'u.email');

        $dql    = $builder->getQuery()->getDQL();
        $result = $builder->getQuery()->getResult();
        
        $this->assertEquals(sizeof($result), 3);
        $this->assertArrayHasKey('foo@foo.com', $result);
        $this->assertArrayHasKey('bar@bar.com', $result);
        $this->assertArrayHasKey('foobar@foobar.com', $result);
        $this->assertEquals('SELECT u FROM Doctrine\Tests\ORM\Functional\Ticket\DDC1135User u INDEX BY u.email', $dql);
    }
    
    public function  testIndexWithJoin()
    {
        $builder = $this->_em->createQueryBuilder();
        $builder->select('u','p')
                ->from(__NAMESPACE__ . '\DDC1135User', 'u', 'u.email')
                ->join('u.phones', 'p', null, null, 'p.id');

        
        $dql    = $builder->getQuery()->getDQL();
        $result = $builder->getQuery()->getResult();
        
        $this->assertEquals(sizeof($result), 3);
        $this->assertArrayHasKey('foo@foo.com', $result);
        $this->assertArrayHasKey('bar@bar.com', $result);
        $this->assertArrayHasKey('foobar@foobar.com', $result);
        
        $this->assertEquals(sizeof($result['foo@foo.com']->phones), 3);
        $this->assertEquals(sizeof($result['bar@bar.com']->phones), 3);
        $this->assertEquals(sizeof($result['foobar@foobar.com']->phones), 3);
        
        $this->assertArrayHasKey(1, $result['foo@foo.com']->phones->toArray());
        $this->assertArrayHasKey(2, $result['foo@foo.com']->phones->toArray());
        $this->assertArrayHasKey(3, $result['foo@foo.com']->phones->toArray());
        
        $this->assertArrayHasKey(4, $result['bar@bar.com']->phones->toArray());
        $this->assertArrayHasKey(5, $result['bar@bar.com']->phones->toArray());
        $this->assertArrayHasKey(6, $result['bar@bar.com']->phones->toArray());
        
        $this->assertArrayHasKey(7, $result['foobar@foobar.com']->phones->toArray());
        $this->assertArrayHasKey(8, $result['foobar@foobar.com']->phones->toArray());
        $this->assertArrayHasKey(9, $result['foobar@foobar.com']->phones->toArray());
        
        $this->assertEquals('SELECT u, p FROM '.__NAMESPACE__ . '\DDC1135User u INDEX BY u.email INNER JOIN u.phones p INDEX BY p.id', $dql);
    }
    
    private function loadFixture()
    {
        $p1 = array('11 xxxx-xxxx','11 yyyy-yyyy','11 zzzz-zzzz');
        $p2 = array('22 xxxx-xxxx','22 yyyy-yyyy','22 zzzz-zzzz');
        $p3 = array('33 xxxx-xxxx','33 yyyy-yyyy','33 zzzz-zzzz');
        
        $u1 = new DDC1135User("foo@foo.com", "Foo",$p1);
        $u2 = new DDC1135User("bar@bar.com", "Bar",$p2);
        $u3 = new DDC1135User("foobar@foobar.com", "Foo Bar",$p3);
        
        $this->_em->persist($u1);
        $this->_em->persist($u2);
        $this->_em->persist($u3);
        $this->_em->flush();
        $this->_em->clear();
    }

}

/**
 * @Entity
 */
class DDC1135User
{
    /**
     * @Id @Column(type="integer")
     * @GeneratedValue
     */
    public $id;

    /**
     * @Column(type="string", unique=true)
     */
    public $email;
    
    /**
     * @Column(type="string")
     */
    public $name;
    
    /**
     * @OneToMany(targetEntity="DDC1135Phone", mappedBy="user", cascade={"persist", "remove"})
     */
    public $phones;
    
    public function __construct($email, $name,array $numbers)
    {
        $this->name   = $name;
        $this->email  = $email;
        $this->phones = new \Doctrine\Common\Collections\ArrayCollection();
        
        foreach ($numbers as $number) {
            $this->phones->add(new DDC1135Phone($this,$number));
        }
    }
}

/**
 * @Entity
 */
class DDC1135Phone
{
    /**
     * @Id
     * @Column(name="id", type="integer")
     * @GeneratedValue(strategy="AUTO")
     */
    public $id;

    /**
     * @Column(name="number", type="string", nullable = false)
     */
    public $number;

    /**
     * @ManyToOne(targetEntity="DDC1135User", inversedBy="phones")
     * @JoinColumn(name="user_id", referencedColumnName="id", nullable = false)
     */
    public $user;

    public function __construct($user, $number)
    {
        $this->user     = $user;
        $this->number   = $number;
    }
}