doctrine2/tests/Doctrine/Tests/ORM/Functional/HydrationCacheTest.php

<?php
namespace Doctrine\Tests\ORM\Functional;

use Doctrine\Tests\OrmFunctionalTestCase;
use Doctrine\Tests\Models\Cms\CmsUser;
use Doctrine\DBAL\Cache\QueryCacheProfile;
use Doctrine\Common\Cache\ArrayCache;

/**
 * @group DDC-1766
 */
class HydrationCacheTest extends OrmFunctionalTestCase
{
    public function setUp()
    {
        $this->useModelSet('cms');
        parent::setUp();
    }

    public function testHydrationCache()
    {
        $cache = new ArrayCache();

        $user = new CmsUser;
        $user->name = "Benjamin";
        $user->username = "beberlei";
        $user->status = 'active';

        $this->_em->persist($user);
        $this->_em->flush();
        $this->_em->clear();


        $dql = "SELECT u FROM Doctrine\Tests\Models\Cms\CmsUser u";

        $users = $this->_em->createQuery($dql)
                      ->setHydrationCacheProfile(new QueryCacheProfile(null, null, $cache))
                      ->getResult();

        $c = $this->getCurrentQueryCount();
        $users = $this->_em->createQuery($dql)
                      ->setHydrationCacheProfile(new QueryCacheProfile(null, null, $cache))
                      ->getResult();

        $this->assertEquals($c, $this->getCurrentQueryCount(), "Should not execute query. Its cached!");

        $users = $this->_em->createQuery($dql)
                      ->setHydrationCacheProfile(new QueryCacheProfile(null, null, $cache))
                      ->getArrayResult();

        $this->assertEquals($c + 1, $this->getCurrentQueryCount(), "Hydration is part of cache key.");

        $users = $this->_em->createQuery($dql)
                      ->setHydrationCacheProfile(new QueryCacheProfile(null, null, $cache))
                      ->getArrayResult();

        $this->assertEquals($c + 1, $this->getCurrentQueryCount(), "Hydration now cached");

        $users = $this->_em->createQuery($dql)
                      ->setHydrationCacheProfile(new QueryCacheProfile(null, 'cachekey', $cache))
                      ->getArrayResult();

        $this->assertTrue($cache->contains('cachekey'), 'Explicit cache key');

        $users = $this->_em->createQuery($dql)
                      ->setHydrationCacheProfile(new QueryCacheProfile(null, 'cachekey', $cache))
                      ->getArrayResult();
        $this->assertEquals($c + 2, $this->getCurrentQueryCount(), "Hydration now cached");

    }
}