observer/flutter_app/test/model_catalog_section_test.dart

import 'dart:convert';
import 'dart:io';

import 'package:crypto/crypto.dart' show sha256;
import 'package:flutter/material.dart';
import 'package:flutter_test/flutter_test.dart';
import 'package:http/http.dart' as http;
import 'package:http/testing.dart';
import 'package:observer/camera/model_catalog_section.dart';
import 'package:observer/models/model_manager.dart';

const _modelBytes = [1, 2, 3, 4, 5, 6, 7, 8];
String _shaHex(List<int> bytes) => sha256.convert(bytes).toString();

/// 分两段到达的下载流（中间 500ms 停顿：足够「取消」按钮出现并点击，
/// 且第二次分块到达前取消标志必然已置位——每轮 pump 只推进 400ms fake 时钟）
Stream<List<int>> _delayedChunks() async* {
  yield [1, 2, 3, 4];
  await Future<void>.delayed(const Duration(milliseconds: 500));
  yield [5, 6, 7, 8];
}

Map<String, dynamic> _item() => {
      'datasetId': 7,
      'datasetName': '环颈雉鸡数据集',
      'version': 'v1.0.0',
      'labels': ['pheasant', 'suspect'],
      'sizeBytes': _modelBytes.length,
      'sha256': _shaHex(_modelBytes),
      'downloadUrl': '/download/models/7/latest.tflite',
      'coverUrl': '/api/v1/app/cover?namePrefix=RNPHE',
    };

/// 真实流式下载客户端：send 立即返回分块流（MockClient 的 Response.fromStream
/// 会先消费完整流再返回，导致进度永远停在 0）。
class _StreamingClient extends http.BaseClient {
  @override
  Future<http.StreamedResponse> send(http.BaseRequest request) async {
    if (request.url.path == '/api/v1/app/update') {
      return http.StreamedResponse(
          http.ByteStream.fromBytes(utf8.encode(jsonEncode({
            'code': 0,
            'message': 'ok',
            'data': {'models': [_item()]}
          }))),
          200);
    }
    return http.StreamedResponse(_delayedChunks(), 200,
        contentLength: _modelBytes.length);
  }
}

void main() {
  late Directory root;

  setUp(() async {
    root = await Directory.systemTemp.createTemp('catalog-section-test');
  });

  tearDown(() => root.delete(recursive: true));

  ModelManager manager() => ModelManager(
        baseUrl: 'http://test.local',
        client: MockClient((req) async {
          if (req.url.path == '/api/v1/app/update') {
            return http.Response.bytes(
                utf8.encode(jsonEncode({
                  'code': 0,
                  'message': 'ok',
                  'data': {'models': [_item()]}
                })),
                200);
          }
          if (req.url.path.startsWith('/download/models/')) {
            return http.Response.bytes(_modelBytes, 200);
          }
          return http.Response('not found', 404);
        }),
        rootDir: () async => root,
      );

  /// 推进真实异步链：pump 冲刷 fake 微任务队列（并推进 fake 时钟触发 fake 定时器），
  /// 再用 runAsync 让真实事件循环转动，送达文件 IO 完成事件。
  /// testWidgets 的 fake-async 区中发起的真实 IO 必须这样交替驱动，否则异步链卡死。
  Future<void> pumpRealIo(WidgetTester tester,
      {Duration fake = Duration.zero}) async {
    await tester.pump(fake);
    await tester.runAsync(
        () => Future<void>.delayed(const Duration(milliseconds: 5)));
    await tester.pump();
  }

  /// 交替 pump/真实等待，直至 finder 命中（UI 呈现目标状态）。
  Future<void> pumpUntilFound(WidgetTester tester, Finder finder,
      {Duration fake = Duration.zero}) async {
    for (var i = 0; i < 100 && finder.evaluate().isEmpty; i++) {
      await pumpRealIo(tester, fake: fake);
    }
  }

  Future<void> pumpSection(WidgetTester tester, ModelManager m) async {
    await tester.pumpWidget(MaterialApp(
        home: Scaffold(body: ModelCatalogSection(manager: m))));
    await tester.pumpAndSettle(const Duration(milliseconds: 50),
        EnginePhase.sendSemanticsUpdate, const Duration(seconds: 5));
  }

  testWidgets('未下载：显示使用按钮，点击后下载完成自动变为已使用', (tester) async {
    final m = manager();
    await tester.runAsync(() => m.refresh());
    await pumpSection(tester, m);

    expect(find.text('环颈雉鸡数据集'), findsOneWidget);
    expect(find.text('使用'), findsOneWidget);

    await tester.tap(find.text('使用'));
    await tester.pump(); // 下载启动，进度条出现
    // 驱动下载 + 自动激活的 IO 链走完，直至 UI 呈现「已使用」
    await pumpUntilFound(tester, find.text('已使用'));
    await tester.pumpAndSettle();

    expect(m.isDownloaded(7), isTrue);
    expect(m.isActive(7), isTrue);
    expect(find.text('已使用'), findsOneWidget);
  });

  testWidgets('已激活：再次点击取消使用', (tester) async {
    final m = manager();
    await tester.runAsync(() => m.refresh());
    await tester.runAsync(() => m.downloadModel(m.catalog.first));
    await pumpSection(tester, m);

    expect(find.text('已使用'), findsOneWidget);
    await tester.tap(find.text('已使用'));
    // 排空 setActive 的激活集重载与落盘 IO，直至 UI 呈现「使用」
    await pumpUntilFound(tester, find.text('使用'));
    await tester.pumpAndSettle();

    expect(m.isActive(7), isFalse);
    expect(find.text('使用'), findsOneWidget);
  });

  testWidgets('目录为空：显示暂无已发布模型', (tester) async {
    final m = ModelManager(
      baseUrl: 'http://test.local',
      client: MockClient((_) async => http.Response.bytes(
          utf8.encode(jsonEncode(
              {'code': 0, 'message': 'ok', 'data': {'models': []}})),
          200)),
      rootDir: () async => root,
    );
    await tester.runAsync(() => m.refresh());
    await pumpSection(tester, m);

    expect(find.text('暂无已发布模型'), findsOneWidget);
  });

  testWidgets('下载中：显示进度和取消按钮，取消后恢复使用', (tester) async {
    final m = ModelManager(
      baseUrl: 'http://test.local',
      client: _StreamingClient(),
      rootDir: () async => root,
    );
    await tester.runAsync(() => m.refresh());
    await pumpSection(tester, m);

    await tester.tap(find.text('使用'));
    // 首个分块到达，进度条与取消按钮出现（IO 链需交替驱动）
    await pumpUntilFound(tester, find.text('取消'),
        fake: const Duration(milliseconds: 400));
    expect(find.text('取消'), findsOneWidget);

    await tester.tap(find.text('取消'));
    // 推进 fake 时钟触发延迟分块 → 取消分支清场（.part 删除等真实 IO），直至恢复「使用」
    await pumpUntilFound(tester, find.text('使用'),
        fake: const Duration(milliseconds: 400));
    await tester.pumpAndSettle();

    expect(m.isDownloaded(7), isFalse);
    expect(m.isActive(7), isFalse);
    expect(find.text('使用'), findsOneWidget);
  });
}