pkg/analyzer/test/src/diagnostics/await_of_incompatible_type_test.dart

// Copyright (c) 2023, the Dart project authors. Please see the AUTHORS file
// for details. All rights reserved. Use of this source code is governed by a
// BSD-style license that can be found in the LICENSE file.

import 'package:analyzer/src/error/codes.dart';
import 'package:test_reflective_loader/test_reflective_loader.dart';

import '../dart/resolution/context_collection_resolution.dart';

main() {
  defineReflectiveSuite(() {
    defineReflectiveTests(AwaitOfExtensionTypeNotFutureTest);
  });
}

@reflectiveTest
class AwaitOfExtensionTypeNotFutureTest extends PubPackageResolutionTest {
  test_extensionType_implementsFuture() async {
    await assertNoErrorsInCode(r'''
extension type A(Future<int> it) implements Future<int> {}

void f(A a) async {
  await a;
}
''');
  }

  test_extensionType_notImplementsFuture() async {
    await assertErrorsInCode(r'''
extension type A(int it) {}

void f(A a) async {
  await a;
}
''', [
      error(CompileTimeErrorCode.AWAIT_OF_INCOMPATIBLE_TYPE, 51, 5),
    ]);
  }

  test_typeParameter_bound_extensionType_implementsFuture() async {
    await assertNoErrorsInCode(r'''
extension type A(Future<int> it) implements Future<int> {}

void f<T extends A>(T a) async {
  await a;
}
''');
  }

  test_typeParameter_bound_extensionType_notImplementsFuture() async {
    await assertErrorsInCode(r'''
extension type A(Future<int> it) {}

void f<T extends A>(T a) async {
  await a;
}
''', [
      error(CompileTimeErrorCode.AWAIT_OF_INCOMPATIBLE_TYPE, 72, 5),
    ]);
  }

  test_typeParameter_promotedBound_extensionType_implementsFuture() async {
    await assertNoErrorsInCode(r'''
extension type A(Future<int> it) implements Future<int> {}

void f<T>(T a) async {
  if (T is A) {
    await a;
  }
}
''');
  }

  test_typeParameter_promotedBound_extensionType_notImplementsFuture() async {
    await assertErrorsInCode(r'''
extension type A(Future<int> it) {}

void f<T>(T a) async {
  if (a is A) {
    await a;
  }
}
''', [
      error(CompileTimeErrorCode.AWAIT_OF_INCOMPATIBLE_TYPE, 80, 5),
    ]);
  }
}