From 7546d54b4e29377da31344302ce948b5b0e2ee19 Mon Sep 17 00:00:00 2001
From: Dane Everitt <dane@daneeveritt.com>
Date: Sat, 27 Jun 2020 12:32:27 -0700
Subject: [PATCH] Add test coverage for resource utilization

---
 .../ResourceUtilitizationControllerTest.php   | 44 +++++++++++++++++++
 1 file changed, 44 insertions(+)
 create mode 100644 tests/Integration/Api/Client/Server/ResourceUtilitizationControllerTest.php

diff --git a/tests/Integration/Api/Client/Server/ResourceUtilitizationControllerTest.php b/tests/Integration/Api/Client/Server/ResourceUtilitizationControllerTest.php
new file mode 100644
index 000000000..7c713dd33
--- /dev/null
+++ b/tests/Integration/Api/Client/Server/ResourceUtilitizationControllerTest.php
@@ -0,0 +1,44 @@
+<?php
+
+namespace Pterodactyl\Tests\Integration\Api\Client\Server;
+
+use Mockery;
+use Pterodactyl\Models\Permission;
+use Pterodactyl\Repositories\Wings\DaemonServerRepository;
+use Pterodactyl\Tests\Integration\Api\Client\ClientApiIntegrationTestCase;
+
+class ResourceUtilitizationControllerTest extends ClientApiIntegrationTestCase
+{
+    /**
+     * Test that the resource utilization for a server is returned in the expected format.
+     */
+    public function testServerResourceUtilizationIsReturned()
+    {
+        $service = Mockery::mock(DaemonServerRepository::class);
+        $this->app->instance(DaemonServerRepository::class, $service);
+
+        [$user, $server] = $this->generateTestAccount([Permission::ACTION_WEBSOCKET_CONNECT]);
+
+        $service->expects('setServer')->with(Mockery::on(function ($value) use ($server) {
+            return $server->uuid === $value->uuid;
+        }))->andReturnSelf()->getMock()->expects('getDetails')->andReturns([]);
+
+        $response = $this->actingAs($user)->getJson("/api/client/servers/{$server->uuid}/resources");
+
+        $response->assertOk();
+        $response->assertJson([
+            'object' => 'stats',
+            'attributes' => [
+                'current_state' => 'stopped',
+                'is_suspended' => false,
+                'resources' => [
+                    'memory_bytes' => 0,
+                    'cpu_absolute' => 0,
+                    'disk_bytes' => 0,
+                    'network_rx_bytes' => 0,
+                    'network_tx_bytes' => 0,
+                ],
+            ],
+        ]);
+    }
+}