PartitionTest.php

<?php

declare(strict_types=1);

namespace Umbrellio\Postgres\Unit\Schema\Blueprint;

use Illuminate\Support\Carbon;
use InvalidArgumentException;
use Umbrellio\Postgres\Tests\TestCase;
use Umbrellio\Postgres\Tests\Unit\Helpers\BlueprintAssertions;

class PartitionTest extends TestCase
{
    use BlueprintAssertions;

    private const TABLE = 'test_table';

    protected function setUp(): void
    {
        parent::setUp();

        $this->initializeMock(static::TABLE);
    }

    /**
     * @test
     */
    public function detachPartition(): void
    {
        $this->blueprint->detachPartition('some_partition');
        $this->assertSameSql('alter table "test_table" detach partition some_partition');
    }

    /**
     * @test
     */
    public function attachPartitionRangeInt(): void
    {
        $this->blueprint->attachPartition('some_partition')
            ->range([
                'from' => 10,
                'to' => 100,
            ]);
        $this->assertSameSql('alter table "test_table" attach partition some_partition for values from (10) to (100)');
    }

    /**
     * @test
     */
    public function attachPartitionFailedWithoutForValuesPart(): void
    {
        $this->blueprint->attachPartition('some_partition');
        $this->expectException(InvalidArgumentException::class);
        $this->runToSql();
    }

    /**
     * @test
     */
    public function attachPartitionRangeDates(): void
    {
        $today = Carbon::today();
        $tomorrow = Carbon::tomorrow();
        $this->blueprint->attachPartition('some_partition')
            ->range([
                'from' => $today,
                'to' => $tomorrow,
            ]);

        $this->assertSameSql(sprintf(
            'alter table "test_table" attach partition some_partition for values from (\'%s\') to (\'%s\')',
            $today->toDateTimeString(),
            $tomorrow->toDateTimeString()
        ));
    }

    /**
     * @test
     */
    public function attachPartitionStringDates(): void
    {
        $today = '2010-01-01';
        $tomorrow = '2010-12-31';
        $this->blueprint->attachPartition('some_partition')
            ->range([
                'from' => $today,
                'to' => $tomorrow,
            ]);

        $this->assertSameSql(sprintf(
            'alter table "test_table" attach partition some_partition for values from (\'%s\') to (\'%s\')',
            $today,
            $tomorrow
        ));
    }

    /**
     * @test
     */
    public function addingTsrangeColumn()
    {
        $this->blueprint->tsrange('foo');
        $this->assertSameSql('alter table "test_table" add column "foo" tsrange not null');
    }

    /**
     * @test
     */
    public function addingDaterangeColumn()
    {
        $this->blueprint->daterange('foo');
        $this->assertSameSql('alter table "test_table" add column "foo" daterange not null');
    }

    /**
     * @test
     */
    public function addingNumericColumnWithVariablePrecicion()
    {
        $this->blueprint->numeric('foo');
        $this->assertSameSql('alter table "test_table" add column "foo" numeric not null');
    }

    /**
     * @test
     */
    public function addingNumericColumnWithDefinedPrecicion()
    {
        $this->blueprint->numeric('foo', 8);
        $this->assertSameSql('alter table "test_table" add column "foo" numeric(8) not null');
    }

    /**
     * @test
     */
    public function addingNumericColumnWithDefinedPrecicionAndScope()
    {
        $this->blueprint->numeric('foo', 8, 2);
        $this->assertSameSql('alter table "test_table" add column "foo" numeric(8, 2) not null');
    }
}