coolify/tests/Feature/ServerMetricsPrefetchTest.php

<?php

use App\Models\InstanceSettings;
use App\Models\PrivateKey;
use App\Models\Server;
use App\Models\Team;
use Illuminate\Foundation\Testing\RefreshDatabase;
use Illuminate\Process\PendingProcess;
use Illuminate\Support\Facades\Process;

uses(RefreshDatabase::class);

beforeEach(function () {
    InstanceSettings::create(['id' => 0]);
    config(['constants.ssh.mux_enabled' => false]);

    $team = Team::factory()->create();
    $key = PrivateKey::factory()->create(['team_id' => $team->id]);
    $this->server = Server::factory()->create(['team_id' => $team->id, 'private_key_id' => $key->id]);
    $this->server->settings->forceFill(['is_metrics_enabled' => true])->saveQuietly();
    $this->server->refresh();
});

it('fetches several metric types in one SSH command', function () {
    $commands = [];
    Process::fake(function (PendingProcess $process) use (&$commands) {
        $commands[] = $process->command;

        return Process::result(implode("\x1e", [
            '[{"time":1000,"percent":12.5}]',
            '[{"time":1000,"usedPercent":40}]',
            '[{"time":1000,"mount":"/","usedPercent":70}]',
        ]));
    });

    $this->server->prefetchMetrics(['cpu', 'memory', 'disk'], 5);

    expect($this->server->getCpuMetrics(5))->toBe([[1000, 12.5]])
        ->and($this->server->getMemoryMetrics(5))->toBe([[1000, 40.0]])
        ->and($this->server->getDiskMetrics(5))->toBe([[1000, 70.0]])
        ->and($commands)->toHaveCount(1)
        ->and($commands[0])->toContain('/api/cpu/history', '/api/memory/history', '/api/disk/history', '-H @-')
        ->and($commands[0])->not->toContain('Bearer '.$this->server->settings->sentinel_token.'"');
});

it('keeps the Sentinel error of one prefetched metric', function () {
    Process::fake(fn () => Process::result("[{\"time\":1000,\"percent\":1}]\x1e{\"error\":\"Unauthorized\"}"));

    $this->server->prefetchMetrics(['cpu', 'memory'], 5);

    expect($this->server->getCpuMetrics(5))->toBe([[1000, 1.0]])
        ->and(fn () => $this->server->getMemoryMetrics(5))->toThrow(Exception::class, 'Unauthorized');
});

it('falls back to one fetch per metric when the batch fails', function () {
    $commands = [];
    Process::fake(function (PendingProcess $process) use (&$commands) {
        $commands[] = $process->command;

        return count($commands) === 1
            ? Process::result('', 'ssh: connect to host failed', 1)
            : Process::result('[{"time":2000,"percent":5}]');
    });

    $this->server->prefetchMetrics(['cpu', 'memory'], 5);

    expect($this->server->getCpuMetrics(5))->toBe([[2000, 5.0]])
        ->and($commands)->toHaveCount(2);
});

it('quotes the bucketed history URL in a single fetch for a non-root user', function () {
    $commands = [];
    Process::fake(function (PendingProcess $process) use (&$commands) {
        $commands[] = $process->command;

        return Process::result('[{"time":60000,"percent":5}]');
    });
    $this->server->forceFill(['user' => 'deploy'])->saveQuietly();

    expect($this->server->getCpuMetrics(720))->toBe([[60000, 5.0]])
        ->and($commands)->toHaveCount(1)
        ->and($commands[0])->toContain('sudo bash -c')
        ->and($commands[0])->toMatch('/"http:\/\/localhost:8888\/api\/cpu\/history\?from=[^"]+&bucket_ms=60000"/');
});

it('runs the batched metrics script with sudo for a non-root user', function () {
    $commands = [];
    Process::fake(function (PendingProcess $process) use (&$commands) {
        $commands[] = $process->command;

        return Process::result("[]\x1e[]");
    });
    $this->server->forceFill(['user' => 'deploy'])->saveQuietly();

    $this->server->prefetchMetrics(['cpu', 'memory'], 5);

    expect($commands)->toHaveCount(1)
        ->and($commands[0])->toContain('sudo docker exec -i coolify-sentinel sh -s')
        ->and($commands[0])->toContain("\ncurl -s -H @- ")
        ->and($commands[0])->not->toContain('sudo curl')
        ->and($commands[0])->not->toContain('sudo printf');
});

it('keeps long-range charts evenly dense across mixed-resolution Sentinel history', function () {
    // Sentinel keeps raw samples (here every 10 s) for the last one to two days and
    // one-minute averages before that. A 7-day chart must not crowd the raw days.
    $end = 1_800_000_000_000;
    $rawStart = $end - 2 * 86_400_000;
    $rows = [];
    for ($time = $end - 7 * 86_400_000; $time < $rawStart; $time += 60_000) {
        $rows[] = ['time' => $time, 'percent' => 10];
    }
    for ($time = $rawStart; $time < $end; $time += 10_000) {
        $rows[] = ['time' => $time, 'percent' => ($time / 10_000) % 2 === 0 ? 0 : 20];
    }
    Process::fake(fn () => Process::result(json_encode($rows)));

    $metrics = $this->server->getCpuMetrics(10080);
    $recent = array_filter($metrics, fn (array $point) => $point[0] >= $rawStart);

    expect(count($metrics))->toBeLessThanOrEqual(1000)
        ->and(count($recent) / count($metrics))->toBeLessThan(0.35)
        ->and(array_values(array_unique(array_column($recent, 1))))->toBe([10.0]);
});