app/bundles/StatsBundle/Tests/Aggregate/Collection/DAO/StatsDAOTest.php

<?php

namespace Mautic\StatsBundle\Tests\Aggregate\Collection\DAO;

use Mautic\StatsBundle\Aggregate\Collection\DAO\StatsDAO;
use Mautic\StatsBundle\Aggregate\Collection\Stats\DayStat;
use Mautic\StatsBundle\Aggregate\Collection\Stats\MonthStat;
use Mautic\StatsBundle\Aggregate\Collection\Stats\WeekStat;
use Mautic\StatsBundle\Aggregate\Collection\Stats\YearStat;
use PHPUnit\Framework\TestCase;

class StatsDAOTest extends TestCase
{
    public function testGetYearsReturnsYears(): void
    {
        $expected = [
            2018,
            2019,
        ];

        $stats = $this->getStats()->getYears();
        $this->assertEquals($expected, array_keys($stats));

        array_walk($stats, function ($stat): void {
            $this->assertInstanceOf(YearStat::class, $stat);
        });
    }

    public function testGetMonthsReturnsFlattenedMonths(): void
    {
        $expected = [
            '2018-12',
            '2019-11',
            '2019-12',
        ];

        $stats = $this->getStats()->getMonths();
        $this->assertEquals($expected, array_keys($stats));

        array_walk($stats, function ($stat): void {
            $this->assertInstanceOf(MonthStat::class, $stat);
        });
    }

    public function testGetWeekReturnsFlattenedMonths(): void
    {
        $expected = [
            '2018-49',
            '2019-45',
            '2019-49',
        ];

        $stats = $this->getStats()->getWeeks();
        $this->assertEquals($expected, array_keys($stats));

        array_walk($stats, function ($stat): void {
            $this->assertInstanceOf(WeekStat::class, $stat);
        });
    }

    public function testGetDaysReturnsFlattenedDays(): void
    {
        $expected = [
            '2018-12-07',
            '2019-11-07',
            '2019-11-08',
            '2019-12-07',
        ];

        $stats = $this->getStats()->getDays();
        $this->assertEquals($expected, array_keys($stats));

        array_walk($stats, function ($stat): void {
            $this->assertInstanceOf(DayStat::class, $stat);
        });
    }

    public function testGetHoursReturnsFlattenedHours(): void
    {
        $expected = [
            '2018-12-07 12',
            '2018-12-07 13',
            '2018-12-07 14',
            '2019-11-07 12',
            '2019-11-08 12',
            '2019-12-07 12',
        ];

        $stats = $this->getStats()->getHours();
        $this->assertEquals($expected, array_keys($stats));

        array_walk($stats, function ($stat): void {
            $this->assertTrue(is_int($stat->getCount()));
        });
    }

    private function getStats(): StatsDAO
    {
        $stats = new StatsDAO();

        $stats->getYear(2019)
            ->getMonth(11)
            ->getDay(8)
            ->getHour(12)
            ->setCount(100);

        $stats->getYear(2018)
            ->getMonth(12)
            ->getDay(7)
            ->getHour(12)
            ->setCount(100);

        $stats->getYear(2018)
            ->getMonth(12)
            ->getDay(7)
            ->getHour(14)
            ->setCount(300);

        $stats->getYear(2018)
            ->getMonth(12)
            ->getDay(7)
            ->getHour(13)
            ->setCount(200);

        $stats->getYear(2019)
            ->getMonth(12)
            ->getDay(7)
            ->getHour(12)
            ->setCount(100);

        $stats->getYear(2019)
            ->getMonth(11)
            ->getDay(7)
            ->getHour(12)
            ->setCount(100);

        return $stats;
    }
}