<?php

/*
 * This file is part of Twig.
 *
 * (c) Fabien Potencier
 *
 * For the full copyright and license information, please view the LICENSE
 * file that was distributed with this source code.
 */
require_once __DIR__ . '/../../../../lib/Twig/Extensions/Extension/Date.php';

/**
 * @author Robin van der Vleuten <robinvdvleuten@gmail.com>
 */
class Twig_Tests_Extension_DateTest extends PHPUnit_Framework_TestCase
{
    /**
     * @var TwigEnvironment
     */
    private $env;

    public static function setUpBeforeClass()
    {
        if (!class_exists('Twig_Extensions_Extension_Date')) {
            self::markTestSkipped('Unable to find class Twig_Extensions_Extension_Date.');
        }
    }

    public function setUp()
    {
        $timezone = new DateTimeZone(date_default_timezone_get());

        $coreExtension = $this->getMock('Twig_Extension_Core');
        $coreExtension
            ->expects($this->any())
            ->method('getTimezone')
            ->will($this->returnValue($timezone));

        $this->env = $this->getMock('Twig_Environment');
        $this->env
            ->expects($this->any())
            ->method('getExtension')
            ->with('core')
            ->will($this->returnValue($coreExtension))
        ;
    }

    /**
     * @dataProvider getDiffTestData()
     */
    public function testDiffWithStringsFromGivenNow($expected, $translated, $date, $now)
    {
        $extension = new Twig_Extensions_Extension_Date();
        $this->assertEquals($expected, $extension->diff($this->env, $date, $now));
    }

    public function testDiffWithStringsFromNow()
    {
        $extension = new Twig_Extensions_Extension_Date();
        $this->assertRegExp('/^[0-9]+ (second|minute|hour|day|month|year)s* ago$/', $extension->diff($this->env, '24-07-2014'));
    }

    /**
     * @dataProvider getDiffTestData()
     */
    public function testDiffWithDateTimeFromGivenNow($expected, $translated, $date, $now)
    {
        $extension = new Twig_Extensions_Extension_Date();
        $this->assertEquals($expected, $extension->diff($this->env, new DateTime($date), new DateTime($now)));
    }

    public function testDiffWithDateTimeFromNow()
    {
        $extension = new Twig_Extensions_Extension_Date();
        $this->assertRegExp('/^[0-9]+ (second|minute|hour|day|month|year)s* ago$/', $extension->diff($this->env, new DateTime('24-07-2014')));
    }

    /**
     * @dataProvider getDiffTestData()
     */
    public function testDiffCanReturnTranslatableString($expected, $translated, $date, $now)
    {
        $translator = $this->getMock('Symfony\Component\Translation\TranslatorInterface');
        $translator
            ->expects($this->once())
            ->method('transChoice')
            ->with($translated);

        $extension = new Twig_Extensions_Extension_Date($translator);
        $extension->diff($this->env, $date, $now);
    }

    public function getDiffTestData()
    {
        return array_merge($this->getDiffAgoTestData(), $this->getDiffInTestData());
    }

    public function getDiffAgoTestData()
    {
        return array(
            array('1 second ago', 'diff.ago.second', '24-07-2014 17:28:01', '24-07-2014 17:28:02'),
            array('5 seconds ago', 'diff.ago.second', '24-07-2014 17:28:01', '24-07-2014 17:28:06'),
            array('1 minute ago', 'diff.ago.minute', '24-07-2014 17:28:01', '24-07-2014 17:29:01'),
            array('5 minutes ago', 'diff.ago.minute', '24-07-2014 17:28:01', '24-07-2014 17:33:03'),
            array('1 hour ago', 'diff.ago.hour', '24-07-2014 17:28:01', '24-07-2014 18:29:01'),
            array('9 hours ago', 'diff.ago.hour', '24-07-2014 17:28:01', '25-07-2014 02:33:03'),
            array('1 day ago', 'diff.ago.day', '23-07-2014', '24-07-2014'),
            array('5 days ago', 'diff.ago.day', '19-07-2014', '24-07-2014'),
            array('1 month ago', 'diff.ago.month', '23-07-2014', '24-08-2014'),
            array('6 months ago', 'diff.ago.month', '19-07-2014', '24-01-2015'),
            array('1 year ago', 'diff.ago.year', '19-07-2014', '20-08-2015'),
            array('3 years ago', 'diff.ago.year', '19-07-2014', '20-08-2017'),
        );
    }

    public function getDiffInTestData()
    {
        return array(
            array('in 1 second', 'diff.in.second', '24-07-2014 17:28:02', '24-07-2014 17:28:01'),
            array('in 5 seconds', 'diff.in.second', '24-07-2014 17:28:06', '24-07-2014 17:28:01'),
            array('in 1 minute', 'diff.in.minute', '24-07-2014 17:29:01', '24-07-2014 17:28:01'),
            array('in 5 minutes', 'diff.in.minute', '24-07-2014 17:33:03', '24-07-2014 17:28:01'),
            array('in 1 hour', 'diff.in.hour', '24-07-2014 18:29:01', '24-07-2014 17:28:01'),
            array('in 9 hours', 'diff.in.hour', '25-07-2014 02:33:03', '24-07-2014 17:28:01'),
            array('in 1 day', 'diff.in.day', '24-07-2014', '23-07-2014'),
            array('in 5 days', 'diff.in.day', '24-07-2014', '19-07-2014'),
            array('in 1 month', 'diff.in.month', '24-08-2014', '23-07-2014'),
            array('in 6 months', 'diff.in.month', '24-01-2015', '19-07-2014'),
            array('in 1 year', 'diff.in.year', '20-08-2015', '19-07-2014'),
            array('in 3 years', 'diff.in.year', '20-08-2017', '19-07-2014'),
        );
    }
}
