CaiWuXiTong/tests/Unit/Deployment/FinanceBatchIsolationTest.php

<?php

namespace Tests\Unit\Deployment;

use PHPUnit\Framework\TestCase;
use Symfony\Component\Process\Process;

class FinanceBatchIsolationTest extends TestCase
{
    public function test_busy_sync_lock_prevents_starting_a_second_heavy_job(): void
    {
        $directory = sys_get_temp_dir() . '/caiwu-batch-' . bin2hex(random_bytes(6));
        mkdir($directory);
        $lock = fopen($directory . '/sync.lock', 'c');
        flock($lock, LOCK_EX);
        try {
            $process = $this->runBatch($directory, ['jushuitan:sync-shipping', '--days=3']);
            $this->assertSame(75, $process->getExitCode());
            $this->assertFileDoesNotExist($directory . '/called');
            $import = $this->runBatch($directory, ['order-data:import-after-sales-cost', '/input.xlsx', '--dry-run']);
            $this->assertSame(75, $import->getExitCode());
            $this->assertFileDoesNotExist($directory . '/called');
            flock($lock, LOCK_UN);
            $process = $this->runBatch($directory, ['jushuitan:sync-shipping', '--days=3']);
            $this->assertTrue($process->isSuccessful(), $process->getErrorOutput());
            $arguments = file($directory . '/called', FILE_IGNORE_NEW_LINES);
            $this->assertContains('batch', $arguments);
            $this->assertSame(['jushuitan:sync-shipping', '--days=3'], array_slice($arguments, -2));
            $this->assertContains('--no-deps', $arguments);
            $this->assertSame('image=123456789abc', $arguments[0]);
        } finally {
            fclose($lock);
            foreach (glob($directory . '/*') as $path) { unlink($path); }
            rmdir($directory);
        }
    }

    public function test_shipping_sync_retries_after_upstream_failure_and_keeps_a_log(): void
    {
        $directory = sys_get_temp_dir() . '/caiwu-jst-retry-' . bin2hex(random_bytes(6));
        mkdir($directory);
        // 第一次同步输出"同步失败"(副本库超时)，第二次成功
        file_put_contents($directory . '/docker', '#!/bin/sh' . "\n"
            . 'if [ "$1" = inspect ]; then echo caiwu_akaunting:123456789abc; exit 0; fi' . "\n"
            . 'n=$(cat "$CAIWU_REMOTE_DIR/count" 2>/dev/null || echo 0); n=$((n + 1)); echo $n > "$CAIWU_REMOTE_DIR/count"' . "\n"
            . 'if [ $n -eq 1 ]; then echo "同步失败：Adaptive Server connection timed out"; exit 1; fi' . "\n"
            . 'echo "同步完成"; exit 0' . "\n");
        chmod($directory . '/docker', 0700);
        try {
            $process = new Process(
                ['sh', dirname(__DIR__, 3) . '/deploy/run-finance-batch.sh', 'jushuitan:sync-shipping', '--days=3'],
                $directory,
                [
                    'PATH' => $directory . ':' . getenv('PATH'), 'CAIWU_REMOTE_DIR' => $directory,
                    'CAIWU_UPSTREAM_LOCK' => $directory . '/sync.lock', 'CAIWU_BATCH_LOCK_WAIT' => '0.1',
                    'CAIWU_JST_SYNC_RETRY_WAIT' => '0',
                ],
            );
            $process->run();

            $this->assertTrue($process->isSuccessful(), $process->getErrorOutput());
            $this->assertSame('2', trim((string) file_get_contents($directory . '/count')));
            $log = (string) file_get_contents($directory . '/jst-sync-shipping.log');
            $this->assertStringContainsString('attempt 1/3', $log);
            $this->assertStringContainsString('同步失败', $log);
            $this->assertStringContainsString('attempt 2/3', $log);
            $this->assertStringContainsString('同步完成', $log);
        } finally {
            foreach (glob($directory . '/*') as $path) { unlink($path); }
            rmdir($directory);
        }
    }

    /** @param array<string,string|null> $environment */
    private function runBatch(
        string $directory,
        array $arguments,
        array $environment = [],
    ): Process
    {
        file_put_contents($directory . '/docker', '#!/bin/sh' . "\n"
            . 'if [ "$1" = inspect ]; then case "$4" in *Mounts*) echo "$CAIWU_STORAGE_ROOT" ;; *) echo caiwu_akaunting:123456789abc ;; esac; exit 0; fi' . "\n"
            . 'printf "%s\n" "image=$IMAGE_TAG" "$@" > "$CAIWU_CAPTURE"' . "\n");
        chmod($directory . '/docker', 0700);
        $processEnvironment = [
            'PATH' => $directory . ':' . getenv('PATH'), 'CAIWU_REMOTE_DIR' => $directory,
            'CAIWU_UPSTREAM_LOCK' => $directory . '/sync.lock', 'CAIWU_BATCH_LOCK_WAIT' => '0.1',
            'CAIWU_CAPTURE' => $directory . '/called',
            'CAIWU_STORAGE_ROOT' => $directory . '/storage',
            ...$environment,
        ];
        if ($processEnvironment['CAIWU_UPSTREAM_LOCK'] === null) {
            unset($processEnvironment['CAIWU_UPSTREAM_LOCK']);
        }
        $process = new Process(
            array_merge(['sh', dirname(__DIR__, 3) . '/deploy/run-finance-batch.sh'], $arguments),
            $directory,
            $processEnvironment,
        );
        $process->run();

        return $process;
    }

    public function test_exports_skip_idle_containers_and_can_read_while_upstream_writer_holds_lock(): void
    {
        $directory = sys_get_temp_dir() . '/caiwu-export-batch-' . bin2hex(random_bytes(6));
        mkdir($directory . '/storage/app/account-bills-exports/1', 0777, true);
        $lock = fopen($directory . '/sync.lock', 'c');
        flock($lock, LOCK_EX);
        $task = $directory . '/storage/app/account-bills-exports/1/task.json';
        try {
            $this->assertTrue($this->runBatch($directory, ['account-bills:exports:run'])->isSuccessful());
            $this->assertFileDoesNotExist($directory . '/called');
            file_put_contents($task, json_encode(['status' => 'queued']));
            $this->assertTrue($this->runBatch($directory, ['account-bills:exports:run'])->isSuccessful());
            $this->assertFileExists($directory . '/called');
        } finally {
            fclose($lock);
            if (is_file($task)) unlink($task);
            foreach (['/storage/app/account-bills-exports/1', '/storage/app/account-bills-exports', '/storage/app', '/storage'] as $path) rmdir($directory . $path);
            foreach (glob($directory . '/*') as $path) unlink($path);
            rmdir($directory);
        }
    }

    public function test_finance_ledger_recalculation_runs_in_the_isolated_batch_profile(): void
    {
        $directory = sys_get_temp_dir() . '/caiwu-ledger-batch-' . bin2hex(random_bytes(6));
        mkdir($directory);
        try {
            $process = $this->runBatch($directory, ['finance-ledger:recalculate']);
            $this->assertTrue($process->isSuccessful(), $process->getErrorOutput());
            $arguments = file($directory . '/called', FILE_IGNORE_NEW_LINES);
            $this->assertContains('batch', $arguments);
            $this->assertSame('finance-ledger:recalculate', end($arguments));
            $this->assertContains('--no-deps', $arguments);
            $this->assertSame('image=123456789abc', $arguments[0]);
        } finally {
            foreach (glob($directory . '/*') as $path) {
                unlink($path);
            }
            rmdir($directory);
        }
    }

    public function test_planting_work_order_sync_uses_the_isolated_import_entry(): void
    {
        $directory = sys_get_temp_dir() . '/caiwu-planting-import-' . bin2hex(random_bytes(6));
        mkdir($directory);
        file_put_contents($directory . '/run-finance-import.py', '');
        file_put_contents(
            $directory . '/python3',
            '#!/bin/sh' . "\n" . 'printf "%s\n" "$@" > "$CAIWU_CAPTURE"' . "\n",
        );
        chmod($directory . '/python3', 0700);

        try {
            $process = $this->runBatch($directory, [
                'planting-details:sync-work-orders',
                '--month=2026-10',
            ]);

            $this->assertTrue($process->isSuccessful(), $process->getErrorOutput());
            $arguments = file($directory . '/called', FILE_IGNORE_NEW_LINES);
            $this->assertContains($directory . '/run-finance-import.py', $arguments);
            $this->assertContains('planting-details:sync-work-orders', $arguments);
            $this->assertContains('--month=2026-10', $arguments);
        } finally {
            foreach (glob($directory . '/*') as $path) {
                unlink($path);
            }
            rmdir($directory);
        }
    }

    public function test_jst_label_daily_sync_uses_the_isolated_import_entry(): void
    {
        $directory = sys_get_temp_dir() . '/caiwu-jst-label-import-' . bin2hex(random_bytes(6));
        mkdir($directory);
        file_put_contents($directory . '/run-finance-import.py', '');
        file_put_contents(
            $directory . '/python3',
            '#!/bin/sh' . "\n" . 'printf "%s\n" "$@" > "$CAIWU_CAPTURE"' . "\n",
        );
        chmod($directory . '/python3', 0700);

        try {
            $process = $this->runBatch($directory, [
                'jst-order-labels:sync-daily',
                '--from=2026-10-01',
            ]);

            $this->assertTrue($process->isSuccessful(), $process->getErrorOutput());
            $arguments = file($directory . '/called', FILE_IGNORE_NEW_LINES);
            $this->assertContains($directory . '/run-finance-import.py', $arguments);
            $this->assertContains('jst-order-labels:sync-daily', $arguments);
        } finally {
            foreach (glob($directory . '/*') as $path) {
                unlink($path);
            }
            rmdir($directory);
        }
    }

    public function test_planting_daily_uses_the_isolated_batch_profile(): void
    {
        $directory = sys_get_temp_dir() . '/caiwu-planting-daily-' . bin2hex(random_bytes(6));
        mkdir($directory);
        try {
            $arguments = [
                'store-receipt:planting-daily',
                '--from=2026-10-01',
                '--company=1',
            ];
            $process = $this->runBatch($directory, $arguments);

            $this->assertTrue($process->isSuccessful(), $process->getErrorOutput());
            $called = file($directory . '/called', FILE_IGNORE_NEW_LINES);
            $this->assertContains('batch', $called);
            $this->assertContains('--no-deps', $called);
            $this->assertSame($arguments, array_slice($called, -count($arguments)));
        } finally {
            foreach (glob($directory . '/*') as $path) {
                unlink($path);
            }
            rmdir($directory);
        }
    }

    public function test_business_mirror_history_backfill_runs_in_the_isolated_batch_profile(): void
    {
        $directory = sys_get_temp_dir() . '/caiwu-mirror-batch-' . bin2hex(random_bytes(6));
        mkdir($directory);
        $arguments = [
            'business-mirror:sync-tables',
            'jiansun.dbo.fys_Ticketing_install_item',
            '--full',
            '--batch=500',
            '--max-batches=20',
        ];
        try {
            $process = $this->runBatch($directory, $arguments);
            $this->assertTrue($process->isSuccessful(), $process->getErrorOutput());
            $called = file($directory . '/called', FILE_IGNORE_NEW_LINES);
            $this->assertContains('batch', $called);
            $this->assertContains('--no-deps', $called);
            $this->assertSame('image=123456789abc', $called[0]);
            $this->assertSame($arguments, array_slice($called, -count($arguments)));
        } finally {
            foreach (glob($directory . '/*') as $path) {
                unlink($path);
            }
            rmdir($directory);
        }
    }

    public function test_oa_sync_uses_its_own_lock_when_general_upstream_job_is_busy(): void
    {
        $directory = sys_get_temp_dir() . '/caiwu-oa-lock-' . bin2hex(random_bytes(6));
        mkdir($directory);
        $generalLockPath = $directory . '/general.lock';
        $oaLockPath = $directory . '/oa.lock';
        $generalLock = fopen($generalLockPath, 'c');
        flock($generalLock, LOCK_EX);
        try {
            $process = $this->runBatch(
                $directory,
                ['business-mirror:sync-ding-process-details'],
                [
                    'CAIWU_UPSTREAM_LOCK' => null,
                    'CAIWU_GENERAL_LOCK' => $generalLockPath,
                    'CAIWU_OA_LOCK' => $oaLockPath,
                ],
            );

            $this->assertTrue($process->isSuccessful(), $process->getErrorOutput());
            $this->assertFileExists($directory . '/called');
        } finally {
            flock($generalLock, LOCK_UN);
            fclose($generalLock);
            foreach (glob($directory . '/*') as $path) {
                unlink($path);
            }
            rmdir($directory);
        }
    }
}