Phraseanet/tests/Alchemy/Tests/Phrasea/Feed/AggregateLinkGeneratorTest.php

<?php

namespace Alchemy\Tests\Phrasea\Feed;

use Alchemy\Phrasea\Feed\Aggregate;
use Alchemy\Phrasea\Feed\Link\AggregateLinkGenerator;
use Entities\Feed;
use Symfony\Component\Routing\Generator\UrlGenerator;

class AggregateLinkGeneratorTest extends \PhraseanetPHPUnitAbstract
{
    /**
     * @dataProvider provideGenerationData
     */
    public function testGenerate($expected, $format, $user, $page, $renew, $alreadyCreated)
    {
        $feed = new Feed();
        $feed->setTitle("title");

        $another_feed = new Feed($user);
        $another_feed->setTitle("another_title");

        $feeds = array($feed, $another_feed);

        $aggregate = new Aggregate(self::$DI['app']['EM'], $feeds);

        $generator = $this->getMockBuilder('Symfony\Component\Routing\Generator\UrlGenerator')
            ->disableOriginalConstructor()
            ->getMock();

        if ($alreadyCreated) {
            $token = $this->insertOneAggregateToken($user);
            $tokenValue = $token->getValue();
        }

        $capture = null;
        $generator->expects($this->once())
            ->method('generate')
            ->with('feed_user_aggregated', $this->isType('array'), UrlGenerator::ABSOLUTE_URL)
            ->will($this->returnCallback(function ($name, $data, $option) use (&$capture, $expected) {
                $capture = $data;

                return $expected;
            }));

        $random = self::$DI['app']['tokens'];

        $linkGenerator = new AggregateLinkGenerator($generator, self::$DI['app']['EM'], $random);

        $link = $linkGenerator->generate($aggregate, $user, $format, $page, $renew);

        if ($format == "atom") {
            $this->assertSame("application/atom+xml", $link->getMimetype());
            $this->assertSame("AGGREGATE - Atom", $link->getTitle());
        } elseif ($format == "rss") {
            $this->assertSame("application/rss+xml", $link->getMimetype());
            $this->assertSame("AGGREGATE - RSS", $link->getTitle());
        }

        if ($alreadyCreated) {
            if ($renew) {
                $this->assertEquals($format, $capture['format']);
                if (null !== $page) {
                    $this->assertEquals($page, $capture['page']);
                }
                $this->assertNotEquals($tokenValue, $capture['token']);

                $this->assertCount(0, self::$DI['app']['EM']
                    ->getRepository('Entities\AggregateToken')
                    ->findBy(array('value' => $tokenValue)));
                $this->assertCount(1, self::$DI['app']['EM']
                    ->getRepository('Entities\AggregateToken')
                    ->findBy(array('value' => $capture['token'])));
            } else {
                $expectedParams = array(
                    'token'  => $tokenValue,
                    'format' => $format,
                );

                if ($page !== null) {
                    $expectedParams['page'] = $page;
                }

                $this->assertEquals($expectedParams, $capture);

                $this->assertCount(1, self::$DI['app']['EM']
                    ->getRepository('Entities\AggregateToken')
                    ->findBy(array('value' => $tokenValue)));
            }
        } else {
            if (null !== $page) {
                    $this->assertEquals($page, $capture['page']);
            }
            $this->assertEquals($format, $capture['format']);
            $this->assertEquals(12, strlen($capture['token']));

            $this->assertCount(1, self::$DI['app']['EM']
                ->getRepository('Entities\AggregateToken')
                ->findBy(array('value' => $capture['token'])));
        }
        $token = self::$DI['app']['EM']
            ->getRepository('Entities\AggregateToken')
            ->findOneBy(array('usr_id' => $user->get_id()));
        self::$DI['app']['EM']->remove($token);
        self::$DI['app']['EM']->flush();
    }

    public function provideGenerationData()
    {
        $user = $this->getMockBuilder('User_Adapter')
            ->disableOriginalConstructor()
            ->getMock();

        $user->expects($this->any())
            ->method('get_id')
            ->will($this->returnValue(42));

        return array(
            array('doliprane', 'atom', $user, null, false, false),
            array('doliprane', 'atom', $user, null, false, true),
            array('doliprane', 'atom', $user, null, true, false),
            array('doliprane', 'atom', $user, null, true, true),
            array('doliprane', 'atom', $user, 1, false, false),
            array('doliprane', 'atom', $user, 1, false, true),
            array('doliprane', 'atom', $user, 1, true, false),
            array('doliprane', 'atom', $user, 1, true, true),
            array('doliprane', 'rss', $user, null, false, false),
            array('doliprane', 'rss', $user, null, false, true),
            array('doliprane', 'rss', $user, null, true, false),
            array('doliprane', 'rss', $user, null, true, true),
            array('doliprane', 'rss', $user, 1, false, false),
            array('doliprane', 'rss', $user, 1, false, true),
            array('doliprane', 'rss', $user, 1, true, false),
            array('doliprane', 'rss', $user, 1, true, true),
        );
    }

    public function provideGenerationDataPublic()
    {
        return array(
            array('doliprane', 'atom', null),
            array('doliprane', 'atom', 1),
            array('doliprane', 'rss', null),
            array('doliprane', 'rss', 1)
        );
    }
}