<?php

declare(strict_types=1);

namespace Doctrine\Migrations\Tests;

use Doctrine\Migrations\FileQueryWriter;
use Doctrine\Migrations\Generator\FileBuilder;
use Doctrine\Migrations\Query\Query;
use Doctrine\Migrations\Version\Direction;
use Psr\Log\LoggerInterface;

use function file_get_contents;
use function glob;
use function is_dir;
use function is_file;
use function realpath;
use function sys_get_temp_dir;
use function unlink;

final class FileQueryWriterTest extends MigrationTestCase
{
    /** @return string[] */
    private function getSqlFilesList(string $path): array
    {
        if (is_dir($path)) {
            $list = glob(realpath($path) . '/*.sql');
            if ($list === false) {
                return [];
            }

            return $list;
        }

        if (is_file($path)) {
            return [$path];
        }

        return [];
    }

    public function testWrite(): void
    {
        $path                 = sys_get_temp_dir();
        $migrationFileBuilder = $this->createMock(FileBuilder::class);
        $migrationFileBuilder
            ->expects(self::atLeastOnce())
            ->method('buildMigrationFile')
            ->with(['Foo' => [new Query('A')]], Direction::UP)
            ->willReturn('foo');

        $logger = $this->createMock(LoggerInterface::class);
        $logger->expects(self::atLeastOnce())
            ->method('info')
            ->with(self::isType('string'));

        $writer = new FileQueryWriter(
            $migrationFileBuilder,
            $logger
        );

        self::assertTrue($writer->write($path, Direction::UP, ['Foo' => [new Query('A')]]));

        $files = $this->getSqlFilesList($path);

        self::assertCount(1, $files);

        foreach ($files as $file) {
            $contents = (string) file_get_contents($file);
            unlink($file);

            self::assertNotEmpty($contents);
            self::assertStringContainsString('foo', $contents);
        }
    }
}