Add support for more server functionality

2017-07-23 14:51:18 -05:00 · 2017-07-23 14:51:18 -05:00 · 5144e0126b
commit 5144e0126b
parent acbc52506c
16 changed files with 1049 additions and 119 deletions
--- a/tests/Unit/Services/Servers/ContainerRebuildServiceTest.php
+++ b/tests/Unit/Services/Servers/ContainerRebuildServiceTest.php
@ -0,0 +1,139 @@
+<?php
+/**
+ * Pterodactyl - Panel
+ * Copyright (c) 2015 - 2017 Dane Everitt <dane@daneeveritt.com>.
+ *
+ * Permission is hereby granted, free of charge, to any person obtaining a copy
+ * of this software and associated documentation files (the "Software"), to deal
+ * in the Software without restriction, including without limitation the rights
+ * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+ * copies of the Software, and to permit persons to whom the Software is
+ * furnished to do so, subject to the following conditions:
+ *
+ * The above copyright notice and this permission notice shall be included in all
+ * copies or substantial portions of the Software.
+ *
+ * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+ * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+ * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+ * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+ * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+ * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+ * SOFTWARE.
+ */
+
+namespace Tests\Unit\Services\Servers;
+
+use Exception;
+use GuzzleHttp\Exception\RequestException;
+use Mockery as m;
+use Pterodactyl\Exceptions\DisplayException;
+use Pterodactyl\Models\Server;
+use Pterodactyl\Services\Servers\ContainerRebuildService;
+use Tests\TestCase;
+use Pterodactyl\Contracts\Repository\ServerRepositoryInterface;
+use Pterodactyl\Contracts\Repository\Daemon\ServerRepositoryInterface as DaemonServerRepositoryInterface;
+
+class ContainerRebuildServiceTest extends TestCase
+{
+    /**
+     * @var \Pterodactyl\Contracts\Repository\Daemon\ServerRepositoryInterface
+     */
+    protected $daemonServerRepository;
+
+    /**
+     * @var \GuzzleHttp\Exception\RequestException
+     */
+    protected $exception;
+
+    /**
+     * @var \Pterodactyl\Contracts\Repository\ServerRepositoryInterface
+     */
+    protected $repository;
+
+    /**
+     * @var \Pterodactyl\Models\Server
+     */
+    protected $server;
+
+    /**
+     * @var \Pterodactyl\Services\Servers\ContainerRebuildService
+     */
+    protected $service;
+
+    /**
+     * Setup tests.
+     */
+    public function setUp()
+    {
+        parent::setUp();
+
+        $this->daemonServerRepository = m::mock(DaemonServerRepositoryInterface::class);
+        $this->exception = m::mock(RequestException::class)->makePartial();
+        $this->repository = m::mock(ServerRepositoryInterface::class);
+        $this->server = factory(Server::class)->make(['node_id' => 1]);
+
+        $this->service = new ContainerRebuildService($this->daemonServerRepository, $this->repository);
+    }
+
+    /**
+     * Test that a server is marked for rebuild when it's model is passed to the function.
+     */
+    public function testServerShouldBeMarkedForARebuildWhenModelIsPassed()
+    {
+        $this->repository->shouldNotReceive('find');
+        $this->daemonServerRepository->shouldReceive('setNode')->with($this->server->node_id)->once()->andReturnSelf()
+            ->shouldReceive('setAccessServer')->with($this->server->uuid)->once()->andReturnSelf()
+            ->shouldReceive('rebuild')->withNoArgs()->once()->andReturnNull();
+
+        $this->service->rebuild($this->server);
+    }
+
+    /**
+     * Test that a server is marked for rebuild when the ID of the server is passed to the function.
+     */
+    public function testServerShouldBeMarkedForARebuildWhenServerIdIsPassed()
+    {
+        $this->repository->shouldReceive('find')->with($this->server->id)->once()->andReturn($this->server);
+
+        $this->daemonServerRepository->shouldReceive('setNode')->with($this->server->node_id)->once()->andReturnSelf()
+            ->shouldReceive('setAccessServer')->with($this->server->uuid)->once()->andReturnSelf()
+            ->shouldReceive('rebuild')->withNoArgs()->once()->andReturnNull();
+
+        $this->service->rebuild($this->server->id);
+    }
+
+    /**
+     * Test that an exception thrown by guzzle is rendered as a displayable exception.
+     */
+    public function testExceptionThrownByGuzzleShouldBeReRenderedAsDisplayable()
+    {
+        $this->daemonServerRepository->shouldReceive('setNode')->with($this->server->node_id)
+            ->once()->andThrow($this->exception);
+
+        $this->exception->shouldReceive('getResponse')->withNoArgs()->once()->andReturnSelf()
+            ->shouldReceive('getStatusCode')->withNoArgs()->once()->andReturn(400);
+
+        try {
+            $this->service->rebuild($this->server);
+        } catch (Exception $exception) {
+            $this->assertInstanceOf(DisplayException::class, $exception);
+            $this->assertEquals(
+                trans('admin/server.exceptions.daemon_exception', ['code' => 400,]), $exception->getMessage()
+            );
+        }
+    }
+
+    /**
+     * Test that an exception thrown by something other than guzzle is not transformed to a displayable.
+     *
+     * @expectedException \Exception
+     */
+    public function testExceptionNotThrownByGuzzleShouldNotBeTransformedToDisplayable()
+    {
+        $this->daemonServerRepository->shouldReceive('setNode')->with($this->server->node_id)
+            ->once()->andThrow(new Exception());
+
+        $this->service->rebuild($this->server);
+    }
+}