portal/app/tests/Unit/External/ExternalBalanceRefresherTest.php

<?php

declare(strict_types=1);
use App\Mail\ExternalServiceDownMail;
use App\Services\External\BalanceReading;
use App\Services\External\DadataBalanceProvider;
use App\Services\External\ExternalBalanceRefresher;
use App\Services\External\LivenessReading;
use App\Services\External\SupplierBalanceProvider;
use App\Services\External\Yandex360BalanceProvider;
use App\Services\External\YandexCloudBalanceProvider;
use Illuminate\Foundation\Testing\DatabaseTransactions;
use Illuminate\Support\Collection;
use Illuminate\Support\Facades\DB;
use Illuminate\Support\Facades\Http;
use Illuminate\Support\Facades\Mail;
use Tests\Concerns\SharesSupplierPdo;
use Tests\TestCase;

uses(TestCase::class, DatabaseTransactions::class, SharesSupplierPdo::class);

beforeEach(function () {
    Mail::fake();
    foreach ([
        DadataBalanceProvider::class => 'dadata',
        SupplierBalanceProvider::class => 'supplier',
        YandexCloudBalanceProvider::class => 'yandex_cloud',
        Yandex360BalanceProvider::class => 'email',
    ] as $cls => $key) {
        app()->instance($cls, fakeProvider($key, BalanceReading::ok($key, 50000, 'RUB', null)));
    }
    ExternalBalanceRefresher::useLivenessProbes([]);
});

afterEach(fn () => ExternalBalanceRefresher::resetLivenessProbes());

function rows(): Collection
{
    return DB::connection('pgsql_supplier')->table('external_service_balances')->get()->keyBy('service_key');
}

it('обновляет только переданные ключи', function () {
    app(ExternalBalanceRefresher::class)->refresh(['dadata'], force: true, sendAlerts: false);
    expect(rows()->keys()->all())->toBe(['dadata']);
});

it('считает светофор по порогам (минус < red_floor → red)', function () {
    config()->set('services.yandex_cloud.red_floor_rub', 1000);
    app()->instance(YandexCloudBalanceProvider::class, fakeProvider('yandex_cloud', BalanceReading::ok('yandex_cloud', -540.48, 'RUB', 600)));
    app(ExternalBalanceRefresher::class)->refresh(['yandex_cloud'], force: true, sendAlerts: false);
    expect(rows()['yandex_cloud']->light)->toBe('red');
    expect((bool) rows()['yandex_cloud']->ok)->toBeTrue();
});

it('alive-провайдер пишет balance NULL + light green + ok true', function () {
    app()->instance(DadataBalanceProvider::class, fakeProvider('dadata', BalanceReading::alive('dadata')));
    app(ExternalBalanceRefresher::class)->refresh(['dadata'], force: true, sendAlerts: false);
    $r = rows()['dadata'];
    expect($r->balance_amount)->toBeNull();
    expect($r->light)->toBe('green');
    expect((bool) $r->ok)->toBeTrue();
});

it('упавший провайдер сохраняет ошибку, не роняет остальных', function () {
    app()->instance(DadataBalanceProvider::class, fakeProvider('dadata', BalanceReading::fail('dadata', 'HTTP 403')));
    app(ExternalBalanceRefresher::class)->refresh(['dadata', 'supplier'], force: true, sendAlerts: false);
    expect((bool) rows()['dadata']->ok)->toBeFalse();
    expect(rows()['dadata']->error)->toContain('403');
    expect((bool) rows()['supplier']->ok)->toBeTrue();
});

it('force=false пропускает свежую строку (окно 60с)', function () {
    app(ExternalBalanceRefresher::class)->refresh(['dadata'], force: true, sendAlerts: false);
    app()->instance(DadataBalanceProvider::class, fakeProvider('dadata', BalanceReading::ok('dadata', 111, 'RUB', null)));
    app(ExternalBalanceRefresher::class)->refresh(['dadata'], force: false, sendAlerts: false);
    expect((float) rows()['dadata']->balance_amount)->toBe(50000.0);
});

it('пробы живости: balance NULL, цвет из пробы', function () {
    ExternalBalanceRefresher::useLivenessProbes([
        fakeProbe('captcha', LivenessReading::unknown('captcha', 'выключена')),
    ]);
    app(ExternalBalanceRefresher::class)->refresh(['captcha'], force: true, sendAlerts: false);
    expect(rows()['captcha']->light)->toBe('grey');
    expect((bool) rows()['captcha']->ok)->toBeFalse();
});

it('sendAlerts=false не шлёт письмо даже на свежий red (фон/кнопка не спамят почту)', function () {
    ExternalBalanceRefresher::useLivenessProbes([
        fakeProbe('captcha', LivenessReading::down('captcha', 'HTTP 500')),
    ]);
    app(ExternalBalanceRefresher::class)->refresh(['captcha'], force: true, sendAlerts: false);
    Mail::assertNothingSent();
});

it('sendAlerts=true шлёт письмо на переход green→red (свежий red, prior отсутствует)', function () {
    ExternalBalanceRefresher::useLivenessProbes([
        fakeProbe('captcha', LivenessReading::down('captcha', 'HTTP 500')),
    ]);
    app(ExternalBalanceRefresher::class)->refresh(['captcha'], force: true, sendAlerts: true);
    Mail::assertSent(ExternalServiceDownMail::class, 1);
});

it('allKeys содержит 12 сервисов и refresh(allKeys) их пишет', function () {
    config()->set('services.aitunnel.key', 'k');
    config()->set('services.aitunnel.balance_url', '');
    config()->set('services.exa.key', 'k');
    config()->set('services.exa.balance_url', '');
    config()->set('services.xfetch.key', 'k');
    config()->set('services.xfetch.balance_url', '');
    config()->set('services.self_render.health_url', 'https://vm.local/h');
    config()->set('services.sales_finder.base_url', 'https://lk.liderra.ru/finder/');
    config()->set('services.exa.proxy', 'socks5h://127.0.0.1:10808');
    Http::fake(['*' => Http::response('ok', 200)]);

    $refresher = app(ExternalBalanceRefresher::class);
    expect($refresher->allKeys())->toHaveCount(12);
    $refresher->refresh($refresher->allKeys(), force: true, sendAlerts: false);
    $keys = DB::connection('pgsql_supplier')->table('external_service_balances')->pluck('service_key');
    foreach (['aitunnel', 'exa', 'xfetch', 'self_render', 'sales_finder', 'exa_tunnel'] as $k) {
        expect($keys)->toContain($k);
    }
});

it('lightKeys не содержит тяжёлого supplier', function () {
    expect(app(ExternalBalanceRefresher::class)->lightKeys())->not->toContain('supplier');
});