65 lines
1.8 KiB
PHP
65 lines
1.8 KiB
PHP
<?php
|
|
|
|
namespace Pterodactyl\Tests\Unit\Http\Middleware;
|
|
|
|
use Mockery as m;
|
|
use Mockery\MockInterface;
|
|
use Pterodactyl\Models\Node;
|
|
use Illuminate\Http\Response;
|
|
use Pterodactyl\Models\Server;
|
|
use Illuminate\Contracts\Routing\ResponseFactory;
|
|
use Pterodactyl\Http\Middleware\MaintenanceMiddleware;
|
|
|
|
class MaintenanceMiddlewareTest extends MiddlewareTestCase
|
|
{
|
|
private MockInterface $response;
|
|
|
|
/**
|
|
* Setup tests.
|
|
*/
|
|
public function setUp(): void
|
|
{
|
|
parent::setUp();
|
|
|
|
$this->response = m::mock(ResponseFactory::class);
|
|
}
|
|
|
|
/**
|
|
* Test that a node not in maintenance mode continues through the request cycle.
|
|
*/
|
|
public function testHandle()
|
|
{
|
|
$server = Server::factory()->make();
|
|
$node = Node::factory()->make(['maintenance' => 0]);
|
|
|
|
$server->setRelation('node', $node);
|
|
$this->setRequestAttribute('server', $server);
|
|
|
|
$this->getMiddleware()->handle($this->request, $this->getClosureAssertions());
|
|
}
|
|
|
|
/**
|
|
* Test that a node in maintenance mode returns an error view.
|
|
*/
|
|
public function testHandleInMaintenanceMode()
|
|
{
|
|
$server = Server::factory()->make();
|
|
$node = Node::factory()->make(['maintenance_mode' => 1]);
|
|
|
|
$server->setRelation('node', $node);
|
|
$this->setRequestAttribute('server', $server);
|
|
|
|
$this->response->shouldReceive('view')
|
|
->once()
|
|
->with('errors.maintenance')
|
|
->andReturn(new Response());
|
|
|
|
$response = $this->getMiddleware()->handle($this->request, $this->getClosureAssertions());
|
|
$this->assertInstanceOf(Response::class, $response);
|
|
}
|
|
|
|
private function getMiddleware(): MaintenanceMiddleware
|
|
{
|
|
return new MaintenanceMiddleware($this->response);
|
|
}
|
|
}
|