RevokeSftpAccessJobTest.php

<?php

namespace Pterodactyl\Tests\Integration\Jobs;

use Pterodactyl\Models\Node;
use Pterodactyl\Models\Server;
use Pterodactyl\Jobs\RevokeSftpAccessJob;
use PHPUnit\Framework\Attributes\TestWith;
use GuzzleHttp\Exception\TransferException;
use Pterodactyl\Tests\Integration\IntegrationTestCase;
use Pterodactyl\Repositories\Wings\DaemonRevocationRepository;
use Pterodactyl\Exceptions\Http\Connection\DaemonConnectionException;

class RevokeSftpAccessJobTest extends IntegrationTestCase
{
    #[TestWith([Server::class, 'server'])]
    #[TestWith([Node::class, 'node'])]
    public function testUniqueIdBasedOnModelType(string $class, string $key): void
    {
        $model = $class::factory()->make(['uuid' => 'uuid-1234']);

        $job = new RevokeSftpAccessJob('user-1', $model);

        $this->assertEquals(
            "revoke-sftp:user-1:{$key}:uuid-1234",
            $job->uniqueId()
        );
    }

    public function testJobReleasesBackToQueueOnFailure(): void
    {
        $node = Node::factory()->make(['uuid' => 'uuid-1234']);

        $mock = $this->mock(DaemonRevocationRepository::class, function ($mock) {
            $mock->expects('setNode->deauthorize')->andThrows(
                new DaemonConnectionException(new TransferException('Connection failed'))
            );
        });

        $job = \Mockery::mock(RevokeSftpAccessJob::class, ['user-1', $node])->makePartial();
        $job->expects('release')->with(10);

        $job->handle($mock);
    }

    public function testJobDispatchesForNode(): void
    {
        $node = Node::factory()->make(['uuid' => 'uuid-1234']);

        $mock = $this->mock(DaemonRevocationRepository::class, function ($mock) {
            $mock->expects('setNode')->andReturnSelf();
            $mock->expects('deauthorize')->with('user-1', [])->andReturnUndefined();
        });

        (new RevokeSftpAccessJob('user-1', $node))->handle($mock);
    }

    public function testJobDispatchesForIndividualServer(): void
    {
        $node = Node::factory()->make(['uuid' => 'node-1234']);
        $server = Server::factory()->make(['uuid' => 'server-1234'])->setRelation('node', $node);

        $mock = $this->mock(DaemonRevocationRepository::class, function ($mock) {
            $mock->expects('setNode')->with(\Mockery::on(fn (Node $node) => $node->uuid === 'node-1234'))->andReturnSelf();
            $mock->expects('deauthorize')->with('user-1', ['server-1234'])->andReturnUndefined();
        });

        (new RevokeSftpAccessJob('user-1', $server))->handle($mock);
    }
}