Add exception tests

Author: lancepioch

tests/Unit/Http/Controllers/Base/IndexControllerTest.php

@@ -9,17 +9,22 @@
 
 namespace Tests\Unit\Http\Controllers\Base;
 
+use GuzzleHttp\Exception\ConnectException;
+use GuzzleHttp\Exception\RequestException;
+use GuzzleHttp\Psr7\ServerRequest;
 use Mockery as m;
 use Pterodactyl\Models\User;
 use GuzzleHttp\Psr7\Response;
 use Pterodactyl\Models\Server;
+use Symfony\Component\HttpKernel\Exception\HttpException;
 use Tests\Assertions\ControllerAssertionsTrait;
 use Tests\Unit\Http\Controllers\ControllerTestCase;
 use Pterodactyl\Http\Controllers\Base\IndexController;
 use Illuminate\Contracts\Pagination\LengthAwarePaginator;
 use Pterodactyl\Services\DaemonKeys\DaemonKeyProviderService;
 use Pterodactyl\Contracts\Repository\ServerRepositoryInterface;
 use Pterodactyl\Contracts\Repository\Daemon\ServerRepositoryInterface as DaemonServerRepositoryInterface;
+use Psr\Http\Message\RequestInterface;
 
 class IndexControllerTest extends ControllerTestCase
 {
@@ -134,4 +139,44 @@ public function testStatusControllerWhenServerIsSuspended()
         $this->assertResponseCodeEquals(200, $response);
         $this->assertResponseJsonEquals(['status' => 30], $response);
     }
+
+    /**
+     * Test the status controller when we can't connect to a server.
+     */
+    public function testStatusControllerWithServerConnectionException()
+    {
+        $user = factory(User::class)->make();
+        $server = factory(Server::class)->make(['suspended' => 0, 'installed' => 1]);
+
+        $this->request->shouldReceive('user')->withNoArgs()->once()->andReturn($user);
+        $this->repository->shouldReceive('findFirstWhere')->with([['uuidShort', '=', $server->uuidShort]])->once()->andReturn($server);
+        $this->keyProviderService->shouldReceive('handle')->with($server, $user)->once()->andReturn('test123');
+
+        $this->daemonRepository->shouldReceive('setServer')->with($server)->once()->andReturnSelf()
+            ->shouldReceive('setToken')->with('test123')->once()->andReturnSelf()
+            ->shouldReceive('details')->withNoArgs()->once()->andThrow(new ConnectException('bad connection', new ServerRequest('', '')));
+
+        $this->expectExceptionObject(new HttpException(500, 'bad connection'));
+        $this->controller->status($this->request, $server->uuidShort);
+    }
+
+    /**
+     * Test the status controller when we can't connect to a server.
+     */
+    public function testStatusControllerWithRequestException()
+    {
+        $user = factory(User::class)->make();
+        $server = factory(Server::class)->make(['suspended' => 0, 'installed' => 1]);
+
+        $this->request->shouldReceive('user')->withNoArgs()->once()->andReturn($user);
+        $this->repository->shouldReceive('findFirstWhere')->with([['uuidShort', '=', $server->uuidShort]])->once()->andReturn($server);
+        $this->keyProviderService->shouldReceive('handle')->with($server, $user)->once()->andReturn('test123');
+
+        $this->daemonRepository->shouldReceive('setServer')->with($server)->once()->andReturnSelf()
+            ->shouldReceive('setToken')->with('test123')->once()->andReturnSelf()
+            ->shouldReceive('details')->withNoArgs()->once()->andThrow(new RequestException('bad request', new ServerRequest('', '')));
+
+        $this->expectExceptionObject(new HttpException(500, 'bad request'));
+        $this->controller->status($this->request, $server->uuidShort);
+    }
 }