mirror of
				https://github.com/pelican-dev/panel.git
				synced 2025-11-04 04:26:51 +01:00 
			
		
		
		
	
		
			
				
	
	
		
			283 lines
		
	
	
		
			11 KiB
		
	
	
	
		
			PHP
		
	
	
	
	
	
			
		
		
	
	
			283 lines
		
	
	
		
			11 KiB
		
	
	
	
		
			PHP
		
	
	
	
	
	
<?php
 | 
						|
 | 
						|
namespace App\Tests\Integration\Services\Servers;
 | 
						|
 | 
						|
use Illuminate\Http\Client\ConnectionException;
 | 
						|
use Mockery\MockInterface;
 | 
						|
use App\Models\Egg;
 | 
						|
use App\Models\Node;
 | 
						|
use App\Models\User;
 | 
						|
use App\Models\Server;
 | 
						|
use App\Models\Allocation;
 | 
						|
use Illuminate\Foundation\Testing\WithFaker;
 | 
						|
use Illuminate\Validation\ValidationException;
 | 
						|
use App\Models\Objects\DeploymentObject;
 | 
						|
use App\Tests\Integration\IntegrationTestCase;
 | 
						|
use App\Services\Servers\ServerCreationService;
 | 
						|
use App\Repositories\Daemon\DaemonServerRepository;
 | 
						|
 | 
						|
class ServerCreationServiceTest extends IntegrationTestCase
 | 
						|
{
 | 
						|
    use WithFaker;
 | 
						|
 | 
						|
    protected MockInterface $daemonServerRepository;
 | 
						|
 | 
						|
    protected Egg $bungeecord;
 | 
						|
 | 
						|
    /**
 | 
						|
     * Stub the calls to daemon so that we don't actually hit those API endpoints.
 | 
						|
     */
 | 
						|
    protected function setUp(): void
 | 
						|
    {
 | 
						|
        parent::setUp();
 | 
						|
 | 
						|
        /* @noinspection PhpFieldAssignmentTypeMismatchInspection */
 | 
						|
        $this->bungeecord = Egg::query()
 | 
						|
            ->where('author', 'panel@example.com')
 | 
						|
            ->where('name', 'Bungeecord')
 | 
						|
            ->firstOrFail();
 | 
						|
 | 
						|
        $this->daemonServerRepository = \Mockery::mock(DaemonServerRepository::class);
 | 
						|
        $this->swap(DaemonServerRepository::class, $this->daemonServerRepository);
 | 
						|
    }
 | 
						|
 | 
						|
    /**
 | 
						|
     * Test that a server can be created when a deployment object is provided to the service.
 | 
						|
     *
 | 
						|
     * This doesn't really do anything super complicated, we'll rely on other more specific
 | 
						|
     * tests to cover that the logic being used does indeed find suitable nodes and ports. For
 | 
						|
     * this test we just care that it is recognized and passed off to those functions.
 | 
						|
     */
 | 
						|
    public function test_server_is_created_with_deployment_object(): void
 | 
						|
    {
 | 
						|
        /** @var \App\Models\User $user */
 | 
						|
        $user = User::factory()->create();
 | 
						|
 | 
						|
        /** @var \App\Models\Node $node */
 | 
						|
        $node = Node::factory()->create();
 | 
						|
 | 
						|
        /** @var \App\Models\Allocation[]|\Illuminate\Database\Eloquent\Collection $allocations */
 | 
						|
        $allocations = Allocation::factory()->times(5)->create([
 | 
						|
            'node_id' => $node->id,
 | 
						|
        ]);
 | 
						|
 | 
						|
        $deployment = (new DeploymentObject())->setDedicated(true)->setPorts([
 | 
						|
            $allocations[0]->port,
 | 
						|
        ]);
 | 
						|
 | 
						|
        $egg = $this->cloneEggAndVariables($this->bungeecord);
 | 
						|
        // We want to make sure that the validator service runs as an admin, and not as a regular
 | 
						|
        // user when saving variables.
 | 
						|
        $egg->variables()->first()->update([
 | 
						|
            'user_editable' => false,
 | 
						|
        ]);
 | 
						|
 | 
						|
        $data = [
 | 
						|
            'name' => $this->faker->name(),
 | 
						|
            'description' => $this->faker->sentence(),
 | 
						|
            'owner_id' => $user->id,
 | 
						|
            'memory' => 256,
 | 
						|
            'swap' => 128,
 | 
						|
            'disk' => 100,
 | 
						|
            'io' => 500,
 | 
						|
            'cpu' => 0,
 | 
						|
            'startup' => 'java server2.jar',
 | 
						|
            'image' => 'java:8',
 | 
						|
            'egg_id' => $egg->id,
 | 
						|
            'allocation_additional' => [
 | 
						|
                $allocations[4]->id,
 | 
						|
            ],
 | 
						|
            'environment' => [
 | 
						|
                'BUNGEE_VERSION' => '123',
 | 
						|
                'SERVER_JARFILE' => 'server2.jar',
 | 
						|
            ],
 | 
						|
            'start_on_completion' => true,
 | 
						|
        ];
 | 
						|
 | 
						|
        $this->daemonServerRepository->expects('setServer->create')->with(true)->andReturnUndefined();
 | 
						|
 | 
						|
        try {
 | 
						|
            $this->getService()->handle(array_merge($data, [
 | 
						|
                'environment' => [
 | 
						|
                    'BUNGEE_VERSION' => '',
 | 
						|
                    'SERVER_JARFILE' => 'server2.jar',
 | 
						|
                ],
 | 
						|
            ]), $deployment);
 | 
						|
 | 
						|
            $this->fail('This execution pathway should not be reached.');
 | 
						|
        } catch (ValidationException $exception) {
 | 
						|
            $this->assertCount(1, $exception->errors());
 | 
						|
            $this->assertArrayHasKey('environment.BUNGEE_VERSION', $exception->errors());
 | 
						|
            $this->assertSame('The Bungeecord Version variable field is required.', $exception->errors()['environment.BUNGEE_VERSION'][0]);
 | 
						|
        }
 | 
						|
 | 
						|
        $response = $this->getService()->handle($data, $deployment);
 | 
						|
 | 
						|
        $this->assertInstanceOf(Server::class, $response);
 | 
						|
        $this->assertNotNull($response->uuid);
 | 
						|
        $this->assertSame($response->uuid_short, substr($response->uuid, 0, 8));
 | 
						|
        $this->assertSame($egg->id, $response->egg_id);
 | 
						|
        $this->assertCount(2, $response->variables);
 | 
						|
        $this->assertSame('123', $response->variables()->firstWhere('env_variable', 'BUNGEE_VERSION')->server_value);
 | 
						|
        $this->assertSame('server2.jar', $response->variables()->firstWhere('env_variable', 'SERVER_JARFILE')->server_value);
 | 
						|
 | 
						|
        foreach ($data as $key => $value) {
 | 
						|
            if (in_array($key, ['allocation_additional', 'environment', 'start_on_completion'])) {
 | 
						|
                continue;
 | 
						|
            }
 | 
						|
 | 
						|
            $this->assertSame($value, $response->{$key}, "Failed asserting equality of '$key' in server response. Got: [{$response->{$key}}] Expected: [$value]");
 | 
						|
        }
 | 
						|
 | 
						|
        $this->assertFalse($response->isSuspended());
 | 
						|
        $this->assertFalse($response->oom_killer);
 | 
						|
        $this->assertSame(0, $response->database_limit);
 | 
						|
        $this->assertSame(0, $response->allocation_limit);
 | 
						|
        $this->assertSame(0, $response->backup_limit);
 | 
						|
 | 
						|
        $this->assertCount(2, $response->allocations);
 | 
						|
        $this->assertSame($response->allocation_id, $response->allocations[0]->id);
 | 
						|
        $this->assertSame($allocations[0]->id, $response->allocations[0]->id);
 | 
						|
        $this->assertSame($allocations[4]->id, $response->allocations[1]->id);
 | 
						|
    }
 | 
						|
 | 
						|
    /**
 | 
						|
     * Test that a server without allocation can be created when a deployment object is
 | 
						|
     * provided to the service.
 | 
						|
     */
 | 
						|
    public function test_server_without_allocation_is_created_with_deployment_object(): void
 | 
						|
    {
 | 
						|
        /** @var \App\Models\User $user */
 | 
						|
        $user = User::factory()->create();
 | 
						|
 | 
						|
        /** @var \App\Models\Node $node */
 | 
						|
        $node = Node::factory()->create();
 | 
						|
 | 
						|
        $deployment = (new DeploymentObject())->setNode($node);
 | 
						|
 | 
						|
        $egg = $this->cloneEggAndVariables($this->bungeecord);
 | 
						|
        // We want to make sure that the validator service runs as an admin, and not as a regular
 | 
						|
        // user when saving variables.
 | 
						|
        $egg->variables()->first()->update([
 | 
						|
            'user_editable' => false,
 | 
						|
        ]);
 | 
						|
 | 
						|
        $data = [
 | 
						|
            'name' => $this->faker->name(),
 | 
						|
            'description' => $this->faker->sentence(),
 | 
						|
            'owner_id' => $user->id,
 | 
						|
            'memory' => 256,
 | 
						|
            'swap' => 128,
 | 
						|
            'disk' => 100,
 | 
						|
            'io' => 500,
 | 
						|
            'cpu' => 0,
 | 
						|
            'startup' => 'java server2.jar',
 | 
						|
            'image' => 'java:8',
 | 
						|
            'egg_id' => $egg->id,
 | 
						|
            'allocation_additional' => [],
 | 
						|
            'environment' => [
 | 
						|
                'BUNGEE_VERSION' => '123',
 | 
						|
                'SERVER_JARFILE' => 'server2.jar',
 | 
						|
            ],
 | 
						|
            'start_on_completion' => true,
 | 
						|
        ];
 | 
						|
 | 
						|
        $this->daemonServerRepository->expects('setServer->create')->with(true)->andReturnUndefined();
 | 
						|
 | 
						|
        try {
 | 
						|
            $this->getService()->handle(array_merge($data, [
 | 
						|
                'environment' => [
 | 
						|
                    'BUNGEE_VERSION' => '',
 | 
						|
                    'SERVER_JARFILE' => 'server2.jar',
 | 
						|
                ],
 | 
						|
            ]), $deployment);
 | 
						|
 | 
						|
            $this->fail('This execution pathway should not be reached.');
 | 
						|
        } catch (ValidationException $exception) {
 | 
						|
            $this->assertCount(1, $exception->errors());
 | 
						|
            $this->assertArrayHasKey('environment.BUNGEE_VERSION', $exception->errors());
 | 
						|
            $this->assertSame('The Bungeecord Version variable field is required.', $exception->errors()['environment.BUNGEE_VERSION'][0]);
 | 
						|
        }
 | 
						|
 | 
						|
        $response = $this->getService()->handle($data, $deployment);
 | 
						|
 | 
						|
        $this->assertInstanceOf(Server::class, $response);
 | 
						|
        $this->assertNotNull($response->uuid);
 | 
						|
        $this->assertSame($response->uuid_short, substr($response->uuid, 0, 8));
 | 
						|
        $this->assertSame($egg->id, $response->egg_id);
 | 
						|
        $this->assertCount(2, $response->variables);
 | 
						|
        $this->assertSame('123', $response->variables()->firstWhere('env_variable', 'BUNGEE_VERSION')->server_value);
 | 
						|
        $this->assertSame('server2.jar', $response->variables()->firstWhere('env_variable', 'SERVER_JARFILE')->server_value);
 | 
						|
 | 
						|
        foreach ($data as $key => $value) {
 | 
						|
            if (in_array($key, ['allocation_additional', 'environment', 'start_on_completion'])) {
 | 
						|
                continue;
 | 
						|
            }
 | 
						|
 | 
						|
            $this->assertSame($value, $response->{$key}, "Failed asserting equality of '$key' in server response. Got: [{$response->{$key}}] Expected: [$value]");
 | 
						|
        }
 | 
						|
 | 
						|
        $this->assertFalse($response->isSuspended());
 | 
						|
        $this->assertFalse($response->oom_killer);
 | 
						|
        $this->assertSame(0, $response->database_limit);
 | 
						|
        $this->assertSame(0, $response->allocation_limit);
 | 
						|
        $this->assertSame(0, $response->backup_limit);
 | 
						|
 | 
						|
        $this->assertEmpty($response->allocations);
 | 
						|
        $this->assertNull($response->allocation_id);
 | 
						|
    }
 | 
						|
 | 
						|
    /**
 | 
						|
     * Test that a server is deleted from the Panel if daemon returns an error during the creation
 | 
						|
     * process.
 | 
						|
     */
 | 
						|
    public function test_error_encountered_by_daemon_causes_server_to_be_deleted(): void
 | 
						|
    {
 | 
						|
        /** @var \App\Models\User $user */
 | 
						|
        $user = User::factory()->create();
 | 
						|
 | 
						|
        /** @var \App\Models\Node $node */
 | 
						|
        $node = Node::factory()->create();
 | 
						|
 | 
						|
        /** @var \App\Models\Allocation $allocation */
 | 
						|
        $allocation = Allocation::factory()->create([
 | 
						|
            'node_id' => $node->id,
 | 
						|
        ]);
 | 
						|
 | 
						|
        $data = [
 | 
						|
            'name' => $this->faker->name(),
 | 
						|
            'description' => $this->faker->sentence(),
 | 
						|
            'owner_id' => $user->id,
 | 
						|
            'allocation_id' => $allocation->id,
 | 
						|
            'node_id' => $allocation->node_id,
 | 
						|
            'memory' => 256,
 | 
						|
            'swap' => 128,
 | 
						|
            'disk' => 100,
 | 
						|
            'io' => 500,
 | 
						|
            'cpu' => 0,
 | 
						|
            'startup' => 'java server2.jar',
 | 
						|
            'image' => 'java:8',
 | 
						|
            'egg_id' => $this->bungeecord->id,
 | 
						|
            'environment' => [
 | 
						|
                'BUNGEE_VERSION' => '123',
 | 
						|
                'SERVER_JARFILE' => 'server2.jar',
 | 
						|
            ],
 | 
						|
        ];
 | 
						|
 | 
						|
        $this->daemonServerRepository->expects('setServer->create')->andThrows(new ConnectionException());
 | 
						|
 | 
						|
        $this->daemonServerRepository->expects('setServer->delete')->andReturnUndefined();
 | 
						|
 | 
						|
        $this->expectException(ConnectionException::class);
 | 
						|
 | 
						|
        $this->getService()->handle($data);
 | 
						|
 | 
						|
        $this->assertDatabaseMissing('servers', ['owner_id' => $user->id]);
 | 
						|
    }
 | 
						|
 | 
						|
    private function getService(): ServerCreationService
 | 
						|
    {
 | 
						|
        return $this->app->make(ServerCreationService::class);
 | 
						|
    }
 | 
						|
}
 |