xworkmate-app/test/runtime/desktop_thread_artifact_service_test.dart

import 'dart:io';

import 'package:flutter_test/flutter_test.dart';
import 'package:xworkmate/runtime/assistant_artifacts.dart';
import 'package:xworkmate/runtime/desktop_thread_artifact_service.dart';
import 'package:xworkmate/runtime/runtime_models.dart';

void main() {
  test(
    'loadSnapshot hides historical workspace files when current run has no artifacts',
    () async {
      final workspace = await Directory.systemTemp.createTemp(
        'xworkmate-artifact-snapshot-',
      );
      addTearDown(() async {
        if (await workspace.exists()) {
          await workspace.delete(recursive: true);
        }
      });
      await File('${workspace.path}/historical.md').writeAsString('old');

      final snapshot = await DesktopThreadArtifactService().loadSnapshot(
        workspacePath: workspace.path,
        workspaceKind: WorkspaceRefKind.localPath,
        artifactRelativePaths: const <String>[],
      );

      expect(snapshot.resultEntries, isEmpty);
      expect(snapshot.fileEntries, isEmpty);
      expect(
        snapshot.resultMessage,
        'No task artifacts recorded for this run.',
      );
      expect(snapshot.filesMessage, isEmpty);
    },
  );

  test(
    'loadSnapshot keeps the file list scoped to current task artifacts',
    () async {
      final workspace = await Directory.systemTemp.createTemp(
        'xworkmate-artifact-snapshot-',
      );
      addTearDown(() async {
        if (await workspace.exists()) {
          await workspace.delete(recursive: true);
        }
      });
      await File('${workspace.path}/historical.md').writeAsString('old');
      await File('${workspace.path}/current.md').writeAsString('new');

      final snapshot = await DesktopThreadArtifactService().loadSnapshot(
        workspacePath: workspace.path,
        workspaceKind: WorkspaceRefKind.localPath,
        artifactRelativePaths: const <String>['current.md'],
      );

      expect(
        snapshot.resultEntries.map((entry) => entry.relativePath),
        <String>['current.md'],
      );
      expect(snapshot.fileEntries.map((entry) => entry.relativePath), <String>[
        'current.md',
      ]);
    },
  );

  test(
    'loadSnapshot shows the root PDF deliverable before supporting files',
    () async {
      final workspace = await Directory.systemTemp.createTemp(
        'xworkmate-pdf-artifact-order-',
      );
      addTearDown(() async {
        if (await workspace.exists()) {
          await workspace.delete(recursive: true);
        }
      });
      await Directory(
        '${workspace.path}/assets/diagrams',
      ).create(recursive: true);
      await File(
        '${workspace.path}/assets/diagrams/chapter.png',
      ).writeAsBytes(<int>[1, 2, 3]);
      await File(
        '${workspace.path}/assets/安全架构演进白皮书.pdf',
      ).writeAsBytes(<int>[4, 5, 6]);
      await File(
        '${workspace.path}/安全架构演进白皮书.pdf',
      ).writeAsBytes(<int>[4, 5, 6]);

      final snapshot = await DesktopThreadArtifactService().loadSnapshot(
        workspacePath: workspace.path,
        workspaceKind: WorkspaceRefKind.localPath,
        artifactRelativePaths: const <String>[
          'assets/diagrams/chapter.png',
          'assets/安全架构演进白皮书.pdf',
          '安全架构演进白皮书.pdf',
        ],
      );

      expect(snapshot.fileEntries.map((entry) => entry.relativePath), <String>[
        '安全架构演进白皮书.pdf',
        'assets/安全架构演进白皮书.pdf',
        'assets/diagrams/chapter.png',
      ]);
      expect(snapshot.resultEntries.first.mimeType, 'application/pdf');
    },
  );

  test(
    'loadPreview rejects historical files outside current task artifacts',
    () async {
      final workspace = await Directory.systemTemp.createTemp(
        'xworkmate-artifact-preview-',
      );
      addTearDown(() async {
        if (await workspace.exists()) {
          await workspace.delete(recursive: true);
        }
      });
      await File('${workspace.path}/historical.md').writeAsString('# Old\n');
      const historical = AssistantArtifactEntry(
        id: 'historical.md',
        label: 'historical.md',
        relativePath: 'historical.md',
        kind: AssistantArtifactEntryKind.file,
        mimeType: 'text/markdown',
        previewable: true,
        workspacePath: '',
      );

      final preview = await DesktopThreadArtifactService().loadPreview(
        entry: historical,
        workspacePath: workspace.path,
        workspaceKind: WorkspaceRefKind.localPath,
        artifactRelativePaths: const <String>[],
      );

      expect(preview.kind, AssistantArtifactPreviewKind.empty);
      expect(preview.content, isEmpty);
    },
  );
}