flutter-azure-ci-11.dart

import 'dart:io';

import 'package:flutter_driver/flutter_driver.dart';

extension DriverExt on FlutterDriver {
  Future<void> saveScreenshot() async {
    final path = 'screenshots/'
        '${DateTime.now().toIso8601String().replaceAll(':', '-')}'
        '.png';
    final List<int> pixels = await screenshot();
    final file = File(path);
    await file.writeAsBytes(pixels);
  }

  Future<void> assertElementPresent(
    String valueKey, {
    Duration timeout = const Duration(seconds: 5),
  }) async {
    try {
      return await waitFor(find.byValueKey(valueKey), timeout: timeout);
      // ignore: avoid_catching_errors
    } on DriverError catch (_) {
      await saveScreenshot();
      rethrow;
    }
  }
}