Xboard/tests/Unit/ServerGfwCheckServiceTest.php

<?php

namespace Tests\Unit;

use App\Models\Server;
use App\Models\ServerGfwCheck;
use App\Services\ServerGfwCheckService;
use Illuminate\Foundation\Testing\RefreshDatabase;
use Tests\TestCase;

class ServerGfwCheckServiceTest extends TestCase
{
    use RefreshDatabase;

    public function test_start_automatic_checks_only_enqueues_enabled_parent_nodes_without_active_task(): void
    {
        $eligible = $this->makeServer(['name' => 'eligible-parent']);
        $active = $this->makeServer(['name' => 'active-parent']);
        $stale = $this->makeServer(['name' => 'stale-parent']);
        $this->makeServer([
            'name' => 'disabled-parent',
            'gfw_check_enabled' => false,
        ]);
        $this->makeServer([
            'name' => 'child-node',
            'parent_id' => $eligible->id,
        ]);

        ServerGfwCheck::create([
            'server_id' => $active->id,
            'status' => ServerGfwCheck::STATUS_PENDING,
        ]);
        $staleCheck = ServerGfwCheck::create([
            'server_id' => $stale->id,
            'status' => ServerGfwCheck::STATUS_PENDING,
        ]);
        $staleCheck->forceFill([
            'created_at' => now()->subMinutes(10),
            'updated_at' => now()->subMinutes(10),
        ])->save();

        $result = app(ServerGfwCheckService::class)->startAutomaticChecks();

        $this->assertSame(3, $result['total']);
        $this->assertSame(1, $result['active']);
        $this->assertSame(1, $result['expired']);
        $this->assertSame([$eligible->id, $stale->id], array_column($result['started'], 'id'));
        $this->assertCount(1, $result['skipped']);
        $this->assertDatabaseHas('server_gfw_checks', [
            'server_id' => $eligible->id,
            'status' => ServerGfwCheck::STATUS_PENDING,
        ]);
        $this->assertDatabaseHas('server_gfw_checks', [
            'id' => $staleCheck->id,
            'status' => ServerGfwCheck::STATUS_FAILED,
            'error_message' => '墙检测任务超时：节点端未领取或未上报结果',
        ]);
    }

    public function test_report_result_hides_blocked_nodes_and_restores_only_auto_hidden_nodes(): void
    {
        $parent = $this->makeServer([
            'name' => 'parent',
            'show' => true,
        ]);
        $visibleChild = $this->makeServer([
            'name' => 'visible-child',
            'parent_id' => $parent->id,
            'show' => true,
        ]);
        $manualHiddenChild = $this->makeServer([
            'name' => 'manual-hidden-child',
            'parent_id' => $parent->id,
            'show' => false,
        ]);

        $service = app(ServerGfwCheckService::class);
        $blockedCheck = ServerGfwCheck::create([
            'server_id' => $parent->id,
            'status' => ServerGfwCheck::STATUS_PENDING,
        ]);

        $this->assertTrue($service->reportResult($parent, [
            'check_id' => $blockedCheck->id,
            'operator_summary' => $this->blockedOperators(),
        ]));

        $this->assertFalse($parent->fresh()->show);
        $this->assertTrue($parent->fresh()->gfw_auto_hidden);
        $this->assertFalse($visibleChild->fresh()->show);
        $this->assertTrue($visibleChild->fresh()->gfw_auto_hidden);
        $this->assertFalse($manualHiddenChild->fresh()->show);
        $this->assertFalse($manualHiddenChild->fresh()->gfw_auto_hidden);

        $normalCheck = ServerGfwCheck::create([
            'server_id' => $parent->id,
            'status' => ServerGfwCheck::STATUS_PENDING,
        ]);

        $this->assertTrue($service->reportResult($parent->fresh(), [
            'check_id' => $normalCheck->id,
            'operator_summary' => $this->normalOperators(),
        ]));

        $this->assertTrue($parent->fresh()->show);
        $this->assertFalse($parent->fresh()->gfw_auto_hidden);
        $this->assertTrue($visibleChild->fresh()->show);
        $this->assertFalse($visibleChild->fresh()->gfw_auto_hidden);
        $this->assertFalse($manualHiddenChild->fresh()->show);
        $this->assertFalse($manualHiddenChild->fresh()->gfw_auto_hidden);
    }

    private function makeServer(array $attributes = []): Server
    {
        return Server::create(array_merge([
            'name' => 'test-node',
            'type' => Server::TYPE_VMESS,
            'host' => '127.0.0.1',
            'port' => 443,
            'server_port' => 443,
            'rate' => '1',
            'group_ids' => [1],
            'show' => true,
            'auto_online' => false,
            'gfw_check_enabled' => true,
            'gfw_auto_hidden' => false,
            'enabled' => true,
        ], $attributes));
    }

    private function blockedOperators(): array
    {
        return [
            'ct' => ['total' => 2, 'success' => 0],
            'cu' => ['total' => 2, 'success' => 0],
            'cm' => ['total' => 2, 'success' => 0],
        ];
    }

    private function normalOperators(): array
    {
        return [
            'ct' => ['total' => 2, 'success' => 2],
            'cu' => ['total' => 2, 'success' => 2],
            'cm' => ['total' => 2, 'success' => 2],
        ];
    }
}