flutter/dev/devicelab/lib/tasks/keyboard_hot_restart_test.dart

// Copyright 2014 The Flutter Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

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

import 'package:path/path.dart' as path;

import '../framework/devices.dart';
import '../framework/framework.dart';
import '../framework/task_result.dart';
import '../framework/utils.dart';

// This test verifies that hot restart hides the keyboard if it is visible.
//
// Steps:
//
// 1. Launch an app that focuses a text field at startup.
//    This makes the keyboard visible.
// 2. Wait until the keyboard is visible.
// 3. Update the app's source code to no longer focus a text field at startup.
// 4. Hot restart the app
// 5. Wait until the keyboard is no longer visible.
//
// App under test: //dev/integration_tests/keyboard_hot_restart/lib/main.dart
//
// Since this test must hot restart the app under test, this test cannot use
// testing frameworks like XCUITest or Flutter's integration_test as they don't
// support hot restart. Instead, this test uses the Flutter tool to run the app,
// hot restart it, and verify its log output.
TaskFunction createKeyboardHotRestartTest({
  String? deviceIdOverride,
  bool checkAppRunningOnLocalDevice = false,
  List<String>? additionalOptions,
}) {
  final Directory appDir = dir(
    path.join(flutterDirectory.path, 'dev/integration_tests/keyboard_hot_restart'),
  );

  // This file is modified during the test and needs to be restored at the end.
  final File mainFile = file(path.join(appDir.path, 'lib/main.dart'));
  final String oldContents = mainFile.readAsStringSync();

  // When the test starts, the app forces the keyboard to be visible.
  // The test turns off this behavior by mutating the app's source code from
  // `forceKeyboardOn` to `forceKeyboardOff`.
  // See: //dev/integration_tests/keyboard_hot_restart/lib/main.dart
  const String forceKeyboardOn = 'const bool forceKeyboard = true;';
  const String forceKeyboardOff = 'const bool forceKeyboard = false;';

  return () async {
    if (deviceIdOverride == null) {
      final Device device = await devices.workingDevice;
      await device.unlock();
      deviceIdOverride = device.deviceId;
    }

    return inDirectory<TaskResult>(appDir, () async {
      try {
        section('Create app');
        await createAppProject();

        // Ensure the app forces the keyboard to be visible.
        final String newContents = oldContents.replaceFirst(forceKeyboardOff, forceKeyboardOn);
        mainFile.writeAsStringSync(newContents);

        section('Launch app');

        bool success = false;
        TestState state = TestState.waitUntilDartVmAvailable;

        final int exitCode = await runApp(
          options: <String>['-d', deviceIdOverride!],
          onLine: (String line, Process process) {
            if (state == TestState.waitUntilDartVmAvailable) {
              if (!line.contains('A Dart VM Service') || !line.contains('is available at')) {
                return;
              }

              section('Wait until keyboard is visible');
              state = TestState.waitUntilKeyboardOpen;
            } else if (state == TestState.waitUntilKeyboardOpen) {
              if (!line.contains('flutter: Keyboard is open')) {
                return;
              }

              section('Update the app to no longer force the keyboard to be visible');
              final String newContents = oldContents.replaceFirst(
                forceKeyboardOn,
                forceKeyboardOff,
              );
              mainFile.writeAsStringSync(newContents);
              process.stdin.writeln('R');

              section('Wait until the app hot restarts');
              state = TestState.waitUntilHotRestart;
            } else if (state == TestState.waitUntilHotRestart) {
              if (!line.contains('Restarted application in ')) {
                return;
              }

              section('Wait until the keyboard is no longer visible');
              state = TestState.waitUntilKeyboardClosed;
            } else if (state == TestState.waitUntilKeyboardClosed) {
              if (!line.contains('flutter: Keyboard is closed')) {
                return;
              }

              // Quit the app. This makes the 'flutter run' process exit.
              process.stdin.writeln('q');
              success = true;
            }
          },
        );

        if (exitCode != 0) {
          return TaskResult.failure('flutter run exited with non-zero exit code: $exitCode');
        }

        if (!success) {
          return TaskResult.failure('Test did not complete successfully');
        }
      } finally {
        mainFile.writeAsStringSync(oldContents);
      }

      return TaskResult.success(null);
    });
  };
}

enum TestState {
  waitUntilDartVmAvailable,
  waitUntilKeyboardOpen,
  waitUntilHotRestart,
  waitUntilKeyboardClosed,
}

Future<void> createAppProject() async {
  await exec(path.join(flutterDirectory.path, 'bin', 'flutter'), <String>[
    'create',
    '--platforms=android,ios',
    '.',
  ]);
}

Future<int> runApp({
  required List<String> options,
  required void Function(String, Process) onLine,
}) async {
  final Process process = await startFlutter('run', options: options);

  final Completer<void> stdoutDone = Completer<void>();
  final Completer<void> stderrDone = Completer<void>();

  void onStdout(String line) {
    print('stdout: $line');
    onLine(line, process);
  }

  process.stdout
      .transform<String>(utf8.decoder)
      .transform<String>(const LineSplitter())
      .listen(onStdout, onDone: stdoutDone.complete);

  process.stderr
      .transform<String>(utf8.decoder)
      .transform<String>(const LineSplitter())
      .listen((String line) => print('stderr: $line'), onDone: stderrDone.complete);

  await Future.wait<void>(<Future<void>>[stdoutDone.future, stderrDone.future]);
  return process.exitCode;
}