fix: cleanup .test_optimizer.dart on SIGINT

Author: luisredondo

lib/src/cli/flutter_cli.dart

@@ -1,7 +1,45 @@
+// ignore_for_file: public_member_api_docs
+
 part of 'cli.dart';
 
 const _testOptimizerFileName = '.test_optimizer.dart';
 
+abstract class ProcessSignalOverrides {
+  static final _token = Object();
+  StreamController<ProcessSignal>? _sigintStreamController;
+
+  static ProcessSignalOverrides? get current {
+    return Zone.current[_token] as ProcessSignalOverrides?;
+  }
+
+  static R runZoned<R>(
+    R Function() body, {
+    Stream<ProcessSignal>? sigintStream,
+  }) {
+    final overrides = _ProcessSignalOverridesScope(sigintStream);
+    return _asyncRunZoned(body, zoneValues: {_token: overrides});
+  }
+
+  Stream<ProcessSignal>? get sigintWatch;
+
+  void addSIGINT() {
+    _sigintStreamController?.add(ProcessSignal.sigint);
+  }
+}
+
+class _ProcessSignalOverridesScope extends ProcessSignalOverrides {
+  _ProcessSignalOverridesScope(Stream<ProcessSignal>? mockSigintStream) {
+    if (mockSigintStream != null) {
+      _sigintStreamController = StreamController<ProcessSignal>();
+    }
+  }
+
+  @override
+  Stream<ProcessSignal>? get sigintWatch {
+    return _sigintStreamController?.stream;
+  }
+}
+
 /// Thrown when `flutter pub get` is executed without a `pubspec.yaml`.
 class PubspecNotFound implements Exception {}
 
@@ -211,9 +249,7 @@ class Flutter {
             stderr: stderr ?? noop,
           ).whenComplete(() async {
             if (optimizePerformance) {
-              File(p.join(cwd, 'test', _testOptimizerFileName))
-                  .delete()
-                  .ignore();
+              await _cleanupOptimizerFile(cwd);
             }
 
             if (collectCoverage) {
@@ -325,6 +361,8 @@ Future<int> _flutterTest({
   final groups = <int, TestGroup>{};
   final tests = <int, Test>{};
   final failedTestErrorMessages = <String, List<String>>{};
+  final sigintWatch = ProcessSignalOverrides.current?.sigintWatch ??
+      ProcessSignal.sigint.watch();
 
   var successCount = 0;
   var skipCount = 0;
@@ -351,6 +389,15 @@ Future<int> _flutterTest({
   );
 
   late final StreamSubscription<TestEvent> subscription;
+  late final StreamSubscription<ProcessSignal> sigintWatchSubscription;
+
+  sigintWatchSubscription = sigintWatch.listen((_) async {
+    await _cleanupOptimizerFile(cwd);
+    await subscription.cancel();
+    await sigintWatchSubscription.cancel();
+    return completer.complete(ExitCode.success.code);
+  });
+
   subscription = testRunner(
     workingDirectory: cwd,
     arguments: [
@@ -483,6 +530,8 @@ Future<int> _flutterTest({
       if (event is ExitTestEvent) {
         if (completer.isCompleted) return;
         subscription.cancel();
+        sigintWatchSubscription.cancel();
+
         completer.complete(
           event.exitCode == ExitCode.success.code
               ? ExitCode.success.code
@@ -506,6 +555,10 @@ String? _topGroupName(Test test, Map<int, TestGroup> groups) => test.groupIDs
     .map((groupID) => groups[groupID]?.name)
     .firstWhereOrNull((groupName) => groupName?.isNotEmpty ?? false);
 
+Future<void> _cleanupOptimizerFile(String cwd) async => File(
+      p.join(cwd, 'test', _testOptimizerFileName),
+    ).delete().ignore();
+
 final int _lineLength = () {
   try {
     return stdout.terminalColumns;

test/src/cli/flutter_cli_test.dart

@@ -317,6 +317,53 @@ void main() {
         stderrLogs = [];
       });
 
+      test('cleanup the .test_optimizer file when SIGINT is emitted', () async {
+        final streamController = StreamController<ProcessSignal>();
+        await ProcessSignalOverrides.runZoned(
+          () async {
+            final tempDirectory = Directory.systemTemp.createTempSync();
+            addTearDown(() => tempDirectory.deleteSync(recursive: true));
+
+            final updatedVars = <String, dynamic>{
+              'package-root': tempDirectory.path,
+              'foo': 'bar',
+            };
+
+            File(p.join(tempDirectory.path, 'pubspec.yaml')).createSync();
+            Directory(p.join(tempDirectory.path, 'test')).createSync();
+            when(
+              () => hooks.preGen(
+                vars: any(named: 'vars'),
+                onVarsChanged: any(named: 'onVarsChanged'),
+                workingDirectory: any(named: 'workingDirectory'),
+              ),
+            ).thenAnswer((invocation) async {
+              (invocation.namedArguments[#onVarsChanged] as void Function(
+                Map<String, dynamic> vars,
+              ))
+                  .call(updatedVars);
+            });
+            ProcessSignalOverrides.current?.addSIGINT();
+            await Flutter.test(
+              cwd: tempDirectory.path,
+              optimizePerformance: true,
+              stdout: stdoutLogs.add,
+              stderr: stderrLogs.add,
+              logger: logger,
+              buildGenerator: generatorBuilder(),
+            );
+            final filePath = p.join(
+              tempDirectory.path,
+              'test',
+              '.test_optimizer.dart',
+            );
+            final testOptimizerFile = File(filePath);
+            expect(testOptimizerFile.existsSync(), isFalse);
+          },
+          sigintStream: streamController.stream,
+        );
+      });
+
       test('throws when pubspec not found', () async {
         await expectLater(
           () => Flutter.test(cwd: Directory.systemTemp.path, logger: logger),