blessing-skin-server/tests/HttpTest/ControllersTest/UpdateControllerTest.php

<?php

namespace Tests;

use App\Services\PackageManager;
use GuzzleHttp\Exception\RequestException;
use GuzzleHttp\Psr7\Request;
use GuzzleHttp\Psr7\Response;
use Illuminate\Contracts\Console\Kernel as Artisan;
use Illuminate\Filesystem\Filesystem;
use Illuminate\Foundation\Testing\DatabaseTransactions;
use Symfony\Component\Finder\SplFileInfo;
use Tests\Concerns\MocksGuzzleClient;

class UpdateControllerTest extends TestCase
{
    use DatabaseTransactions;
    use MocksGuzzleClient;

    protected function setUp(): void
    {
        parent::setUp();
        $this->actAs('superAdmin');
    }

    public function testShowUpdatePage()
    {
        $this->setupGuzzleClientMock();

        // Can't connect to update source
        $this->appendToGuzzleQueue([
            new RequestException('Connection Error', new Request('GET', 'whatever')),
        ]);
        $this->get('/admin/update')->assertSee(config('app.version'));

        // Missing `spec` field
        $this->appendToGuzzleQueue([
            new Response(200, [], $this->mockFakeUpdateInfo('8.9.3', ['spec' => 0])),
        ]);
        $this->get('/admin/update')->assertSee(trans('admin.update.errors.spec'));

        // Low PHP version
        $this->appendToGuzzleQueue([
            new Response(200, [], $this->mockFakeUpdateInfo('8.9.3', ['php' => '100.0.0'])),
        ]);
        $this->get('/admin/update')->assertSee(trans('admin.update.errors.php', ['version' => '100.0.0']));

        // New version available
        $this->appendToGuzzleQueue([
            new Response(200, [], $this->mockFakeUpdateInfo('8.9.3')),
        ]);
        $this->get('/admin/update')->assertSee(config('app.version'))->assertSee('8.9.3');
    }

    public function testDownload()
    {
        $this->setupGuzzleClientMock();

        // Already up-to-date
        $this->appendToGuzzleQueue([
            new Response(200, [], $this->mockFakeUpdateInfo('1.2.3')),
        ]);
        $this->getJson('/admin/update/download')
            ->assertJson([
                'code' => 1,
                'message' => trans('admin.update.info.up-to-date'),
            ]);

        // Download
        $this->appendToGuzzleQueue([
            new Response(200, [], $this->mockFakeUpdateInfo('8.9.3')),
            new Response(200, [], $this->mockFakeUpdateInfo('8.9.3')),
        ]);
        $this->mock(PackageManager::class, function ($mock) {
            $mock->shouldReceive('download')->andThrow(new \Exception('ddd'));
        });
        $this->getJson('/admin/update/download')->assertJson(['code' => 1]);
        $this->mock(PackageManager::class, function ($mock) {
            $mock->shouldReceive('download')->andReturnSelf();
            $mock->shouldReceive('extract')->andReturn(true);
        });
        $this->mock(\Illuminate\Filesystem\Filesystem::class, function ($mock) {
            $mock->shouldReceive('delete')->with(storage_path('options.php'))->once();
            $mock->shouldReceive('exists')->with(storage_path('install.lock'))->andReturn(true);
        });
        $this->getJson('/admin/update/download')
            ->assertJson(['code' => 0, 'message' => trans('admin.update.complete')]);
    }

    public function testUpdate()
    {
        $this->mock(Filesystem::class, function ($mock) {
            $mock->shouldReceive('exists')
                ->with(storage_path('install.lock'))
                ->andReturn(true);

            $mock->shouldReceive('put')
                ->with(storage_path('install.lock'), '')
                ->once()
                ->andReturn(true);

            $mock->shouldReceive('files')
                ->with(database_path('update_scripts'))
                ->once()
                ->andReturn([
                    new SplFileInfo('/1.0.0.php', '', ''),
                    new SplFileInfo('/99.0.0.php', '', ''),
                    new SplFileInfo('/100.0.0.php', '', ''),
                ]);

            $mock->shouldNotReceive('getRequire')->with('/1.0.0.php');

            $mock->shouldReceive('getRequire')
                ->with('/99.0.0.php')
                ->once();

            $mock->shouldReceive('getRequire')
                ->with('/100.0.0.php')
                ->once();
        });
        $this->spy(Artisan::class, function ($spy) {
            $spy->shouldReceive('call')
                ->with('migrate', ['--force' => true])
                ->once();
            $spy->shouldReceive('call')->with('view:clear')->once();
        });
        config(['app.version' => '100.0.0']);

        $this->actAs('superAdmin')
            ->get('/setup/exec-update')
            ->assertViewIs('setup.updates.success');
        $this->assertEquals('100.0.0', option('version'));
    }

    protected function mockFakeUpdateInfo(string $version, $extra = [])
    {
        return json_encode(array_merge([
            'spec' => 2,
            'php' => '7.2.0',
            'latest' => $version,
            'url' => "https://whatever.test/$version/update.zip",
        ], $extra));
    }
}