forked from Alex/Pterodactyl-Panel
Fix unit tests
This commit is contained in:
parent
8697185900
commit
ebe3f6a32a
@ -1,182 +0,0 @@
|
||||
<?php
|
||||
|
||||
namespace Tests\Unit\Services\Databases;
|
||||
|
||||
use Mockery as m;
|
||||
use Tests\TestCase;
|
||||
use Pterodactyl\Models\Server;
|
||||
use Pterodactyl\Models\Database;
|
||||
use Pterodactyl\Services\Databases\DatabaseManagementService;
|
||||
use Pterodactyl\Services\Databases\DeployServerDatabaseService;
|
||||
use Pterodactyl\Contracts\Repository\DatabaseRepositoryInterface;
|
||||
use Pterodactyl\Contracts\Repository\DatabaseHostRepositoryInterface;
|
||||
use Pterodactyl\Exceptions\Service\Database\NoSuitableDatabaseHostException;
|
||||
|
||||
class DeployServerDatabaseServiceTest extends TestCase
|
||||
{
|
||||
/**
|
||||
* @var \Pterodactyl\Contracts\Repository\DatabaseHostRepositoryInterface|\Mockery\Mock
|
||||
*/
|
||||
private $databaseHostRepository;
|
||||
|
||||
/**
|
||||
* @var \Pterodactyl\Services\Databases\DatabaseManagementService|\Mockery\Mock
|
||||
*/
|
||||
private $managementService;
|
||||
|
||||
/**
|
||||
* @var \Pterodactyl\Contracts\Repository\DatabaseRepositoryInterface|\Mockery\Mock
|
||||
*/
|
||||
private $repository;
|
||||
|
||||
/**
|
||||
* Setup tests.
|
||||
*/
|
||||
public function setUp(): void
|
||||
{
|
||||
parent::setUp();
|
||||
|
||||
$this->databaseHostRepository = m::mock(DatabaseHostRepositoryInterface::class);
|
||||
$this->managementService = m::mock(DatabaseManagementService::class);
|
||||
$this->repository = m::mock(DatabaseRepositoryInterface::class);
|
||||
|
||||
// Set configs for testing instances.
|
||||
config()->set('pterodactyl.client_features.databases.enabled', true);
|
||||
config()->set('pterodactyl.client_features.databases.allow_random', true);
|
||||
}
|
||||
|
||||
/**
|
||||
* Test handling of non-random hosts when a host is found.
|
||||
*
|
||||
* @dataProvider databaseLimitDataProvider
|
||||
*/
|
||||
public function testNonRandomFoundHost($limit, $count)
|
||||
{
|
||||
$server = factory(Server::class)->make(['database_limit' => $limit]);
|
||||
$model = factory(Database::class)->make();
|
||||
|
||||
$this->databaseHostRepository->shouldReceive('setColumns->findWhere')
|
||||
->once()
|
||||
->with([['node_id', '=', $server->node_id]])
|
||||
->andReturn(collect([$model]));
|
||||
|
||||
$this->managementService->shouldReceive('create')
|
||||
->once()
|
||||
->with($server, [
|
||||
'database_host_id' => $model->id,
|
||||
'database' => 'testdb',
|
||||
'remote' => null,
|
||||
])
|
||||
->andReturn($model);
|
||||
|
||||
$response = $this->getService()->handle($server, ['database' => 'testdb']);
|
||||
|
||||
$this->assertInstanceOf(Database::class, $response);
|
||||
$this->assertSame($model, $response);
|
||||
}
|
||||
|
||||
/**
|
||||
* Test that an exception is thrown if in non-random mode and no host is found.
|
||||
*/
|
||||
public function testNonRandomNoHost()
|
||||
{
|
||||
$this->expectException(NoSuitableDatabaseHostException::class);
|
||||
|
||||
$server = factory(Server::class)->make(['database_limit' => 1]);
|
||||
|
||||
$this->databaseHostRepository->shouldReceive('setColumns->findWhere')
|
||||
->once()
|
||||
->with([['node_id', '=', $server->node_id]])
|
||||
->andReturn(collect());
|
||||
|
||||
$this->databaseHostRepository->expects('setColumns->all')->withNoArgs()->andReturn(collect());
|
||||
|
||||
$this->getService()->handle($server, []);
|
||||
}
|
||||
|
||||
/**
|
||||
* Test handling of random host selection.
|
||||
*/
|
||||
public function testRandomFoundHost()
|
||||
{
|
||||
$server = factory(Server::class)->make(['database_limit' => 1]);
|
||||
$model = factory(Database::class)->make();
|
||||
|
||||
$this->databaseHostRepository->shouldReceive('setColumns->findWhere')
|
||||
->once()
|
||||
->with([['node_id', '=', $server->node_id]])
|
||||
->andReturn(collect());
|
||||
|
||||
$this->databaseHostRepository->shouldReceive('setColumns->all')
|
||||
->once()
|
||||
->andReturn(collect([$model]));
|
||||
|
||||
$this->managementService->shouldReceive('create')
|
||||
->once()
|
||||
->with($server, [
|
||||
'database_host_id' => $model->id,
|
||||
'database' => 'testdb',
|
||||
'remote' => null,
|
||||
])
|
||||
->andReturn($model);
|
||||
|
||||
$response = $this->getService()->handle($server, ['database' => 'testdb']);
|
||||
|
||||
$this->assertInstanceOf(Database::class, $response);
|
||||
$this->assertSame($model, $response);
|
||||
}
|
||||
|
||||
/**
|
||||
* Test that an exception is thrown when no host is found and random is allowed.
|
||||
*/
|
||||
public function testRandomNoHost()
|
||||
{
|
||||
$this->expectException(NoSuitableDatabaseHostException::class);
|
||||
|
||||
$server = factory(Server::class)->make(['database_limit' => 1]);
|
||||
|
||||
$this->databaseHostRepository->expects('setColumns->findWhere')
|
||||
->with([['node_id', '=', $server->node_id]])
|
||||
->andReturn(collect());
|
||||
|
||||
$this->databaseHostRepository->expects('setColumns->all')->withNoArgs()->andReturn(collect());
|
||||
|
||||
$this->getService()->handle($server, []);
|
||||
}
|
||||
|
||||
/**
|
||||
* Provide limits and current database counts for testing.
|
||||
*
|
||||
* @return array
|
||||
*/
|
||||
public function databaseLimitDataProvider(): array
|
||||
{
|
||||
return [
|
||||
[null, 10],
|
||||
[1, 0],
|
||||
];
|
||||
}
|
||||
|
||||
/**
|
||||
* Provide data for servers over their database limit.
|
||||
*
|
||||
* @return array
|
||||
*/
|
||||
public function databaseExceedingLimitDataProvider(): array
|
||||
{
|
||||
return [
|
||||
[2, 2],
|
||||
[2, 3],
|
||||
];
|
||||
}
|
||||
|
||||
/**
|
||||
* Return an instance of the service with mocked dependencies for testing.
|
||||
*
|
||||
* @return \Pterodactyl\Services\Databases\DeployServerDatabaseService
|
||||
*/
|
||||
private function getService(): DeployServerDatabaseService
|
||||
{
|
||||
return new DeployServerDatabaseService($this->repository, $this->databaseHostRepository, $this->managementService);
|
||||
}
|
||||
}
|
Loading…
Reference in New Issue
Block a user