71 lines
2.4 KiB
PHP
71 lines
2.4 KiB
PHP
<?php
|
|
|
|
namespace Tests\Feature;
|
|
|
|
use App\Models\WgTrafficSample;
|
|
use Illuminate\Foundation\Testing\RefreshDatabase;
|
|
use Tests\TestCase;
|
|
|
|
class WgSampleCommandTest extends TestCase
|
|
{
|
|
use RefreshDatabase;
|
|
|
|
private string $file;
|
|
|
|
protected function setUp(): void
|
|
{
|
|
parent::setUp();
|
|
$dir = storage_path('app/restart-signal');
|
|
@mkdir($dir, 0775, true);
|
|
$this->file = $dir.'/wg-status.json';
|
|
@unlink($this->file);
|
|
}
|
|
|
|
protected function tearDown(): void
|
|
{
|
|
@unlink($this->file);
|
|
parent::tearDown();
|
|
}
|
|
|
|
private function writeStatus(array $peers, bool $configured = true): void
|
|
{
|
|
file_put_contents($this->file, json_encode([
|
|
'at' => time(), 'configured' => $configured,
|
|
'gate' => ['marker' => false, 'chain' => false], 'wg_quick' => 'active',
|
|
'server' => ['subnet' => '10.99.0.0/24', 'server_ip' => '10.99.0.1', 'port' => 51820, 'endpoint' => 'x:1', 'pubkey' => 'k'],
|
|
'peers' => $peers,
|
|
]));
|
|
}
|
|
|
|
public function test_stores_one_row_per_peer_when_configured(): void
|
|
{
|
|
$this->writeStatus([
|
|
['name' => 'laptop', 'pubkey' => 'p1', 'endpoint' => '', 'latest_handshake' => time(), 'rx' => 100, 'tx' => 200],
|
|
['name' => 'phone', 'pubkey' => 'p2', 'endpoint' => '', 'latest_handshake' => time(), 'rx' => 5, 'tx' => 6],
|
|
]);
|
|
|
|
$this->artisan('clusev:wg-sample')->assertExitCode(0);
|
|
|
|
$this->assertSame(2, WgTrafficSample::count());
|
|
$this->assertDatabaseHas('wg_traffic_samples', ['peer_pubkey' => 'p1', 'rx' => 100, 'tx' => 200]);
|
|
}
|
|
|
|
public function test_does_nothing_when_unconfigured_or_stale(): void
|
|
{
|
|
@unlink($this->file);
|
|
$this->artisan('clusev:wg-sample')->assertExitCode(0);
|
|
$this->assertSame(0, WgTrafficSample::count());
|
|
}
|
|
|
|
public function test_prunes_samples_older_than_retention(): void
|
|
{
|
|
WgTrafficSample::create(['peer_pubkey' => 'old', 'peer_name' => 'x', 'rx' => 1, 'tx' => 1, 'sampled_at' => now()->subDays(8)]);
|
|
$this->writeStatus([['name' => 'a', 'pubkey' => 'p1', 'endpoint' => '', 'latest_handshake' => time(), 'rx' => 1, 'tx' => 1]]);
|
|
|
|
$this->artisan('clusev:wg-sample')->assertExitCode(0);
|
|
|
|
$this->assertDatabaseMissing('wg_traffic_samples', ['peer_pubkey' => 'old']);
|
|
$this->assertDatabaseHas('wg_traffic_samples', ['peer_pubkey' => 'p1']);
|
|
}
|
|
}
|