updateTestFiles method

Implementation

void updateTestFiles(String projectDir) {
  print('🧪 Updating test files...');
  final testDir = path.join(projectDir, 'test');

  // Update widget_test.dart if it exists
  final widgetTestFile = File(path.join(testDir, 'widget_test.dart'));
  if (widgetTestFile.existsSync()) {
    var content = widgetTestFile.readAsStringSync();

    // Add import for app.dart if not already present
    if (!content.contains("import '../lib/app.dart'")) {
      // Find where to insert the import
      final importIndex = content.lastIndexOf("import ");
      final endOfImportLine = content.indexOf(';', importIndex) + 1;

      final newImport = "\nimport '../lib/app.dart';";
      content = content.substring(0, endOfImportLine) +
          newImport +
          content.substring(endOfImportLine);

      // Replace any references to MyApp with the correct import
      content = content.replaceAll("import '../lib/main.dart';",
          "import '../lib/main.dart';\nimport '../lib/app.dart';");

      widgetTestFile.writeAsStringSync(content);
      print('Updated widget_test.dart with app.dart import');
    }
  } else {
    print('No widget_test.dart found, skipping test file updates');
  }
}