dev/integration_tests/flutter_driver_screenshot_test/test_driver/main_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 'package:flutter_driver/flutter_driver.dart';
import 'package:test/test.dart' hide TypeMatcher, isInstanceOf;
import 'package:flutter_test/src/buffer_matcher.dart';

Future<void> main() async {
  FlutterDriver driver;
  String deviceModel;

  setUpAll(() async {
    driver = await FlutterDriver.connect();
    deviceModel = await driver.requestData('device_model');
  });

  tearDownAll(() => driver.close());

  test('A page with an image screenshot', () async {
    final SerializableFinder imagePageListTile =
        find.byValueKey('image_page');
    await driver.waitFor(imagePageListTile);
    await driver.tap(imagePageListTile);
    await driver.waitFor(find.byValueKey('red_square_image'));
    await driver.waitUntilNoTransientCallbacks();

    // TODO(egarciad): This is currently a no-op on LUCI.
    // https://github.com/flutter/flutter/issues/49837
    await expectLater(
      driver.screenshot(),
      bufferMatchesGoldenFile('red_square_driver_screenshot__$deviceModel.png'),
    );

    await driver.tap(find.byTooltip('Back'));
  });
}