pkg/analyzer/test/src/dart/resolution/conditional_expression_test.dart

// Copyright (c) 2019, 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:test_reflective_loader/test_reflective_loader.dart';

import 'context_collection_resolution.dart';
import 'node_text_expectations.dart';

main() {
  defineReflectiveSuite(() {
    defineReflectiveTests(ConditionalExpressionResolutionTest);
    defineReflectiveTests(InferenceUpdate3Test);
    defineReflectiveTests(UpdateNodeTextExpectations);
  });
}

@reflectiveTest
class ConditionalExpressionResolutionTest extends PubPackageResolutionTest {
  test_condition_super() async {
    var result = await resolveTestCodeWithDiagnostics('''
class A {
  void f() {
    super ? 0 : 1;
//  ^^^^^
// [diag.missingAssignableSelector] Missing selector such as '.identifier' or '[0]'.
  }
}
''');

    var node = result.findNode.singleConditionalExpression;
    assertResolvedNodeText(node, r'''
ConditionalExpression
  condition2: InvalidSuperExpression
    superReference: SuperReference
      superKeyword: super
    staticType: InvalidType
  condition(v1): SuperExpression
    superKeyword: super
    staticType: A
  question: ?
  thenExpression2: IntegerLiteral
    literal: 0
    staticType: int
  colon: :
  elseExpression2: IntegerLiteral
    literal: 1
    staticType: int
  staticType: int
''');
  }

  test_downward_condition() async {
    var result = await resolveTestCodeWithDiagnostics('''
void f(int b, int c) {
  a() ? b : c;
}

T a<T>() => throw '';
''');

    var node = result.findNode.singleUnqualifiedFunctionInvocation;
    assertResolvedNodeText(node, r'''
UnqualifiedFunctionInvocation
  name: a
  argumentList: ArgumentList
    leftParenthesis: (
    rightParenthesis: )
  resolution: ExecutableInvocationResolution
    element: <testLibrary>::@function::a
    invokeType: bool Function()
    type: bool
  staticType: bool
  typeArgumentTypes
    bool
V1: MethodInvocation
  methodName: SimpleIdentifier
    token: a
    element: <testLibrary>::@function::a
    staticType: T Function<T>()
  argumentList: ArgumentList
    leftParenthesis: (
    rightParenthesis: )
  staticInvokeType: bool Function()
  staticType: bool
  typeArgumentTypes
    bool
''');
  }

  test_else_super() async {
    var result = await resolveTestCodeWithDiagnostics('''
class A {
  void f(bool c) {
    c ? 0 : super;
//          ^^^^^
// [diag.missingAssignableSelector] Missing selector such as '.identifier' or '[0]'.
  }
}
''');

    var node = result.findNode.singleConditionalExpression;
    assertResolvedNodeText(node, r'''
ConditionalExpression
  condition2: UnqualifiedNameExpression
    name: c
    resolution: VariableReadResolution
      element: <testLibrary>::@class::A::@method::f::@formalParameter::c
      type: bool
    staticType: bool
  condition(v1): SimpleIdentifier
    token: c
    element: <testLibrary>::@class::A::@method::f::@formalParameter::c
    staticType: bool
  question: ?
  thenExpression2: IntegerLiteral
    literal: 0
    staticType: int
  colon: :
  elseExpression2: InvalidSuperExpression
    superReference: SuperReference
      superKeyword: super
    staticType: InvalidType
  elseExpression(v1): SuperExpression
    superKeyword: super
    staticType: A
  staticType: InvalidType
''');
  }

  test_ifNull_lubUsedEvenIfItDoesNotSatisfyContext_beforeInferenceUpdate3() async {
    var result = await resolveTestCodeWithDiagnostics('''
// %before-language-feature: inference-update-3
class A {}
class B1 extends A {}
class B2 extends A {}
class C1 implements B1, B2 {}
class C2 implements B1, B2 {}
f(bool b, C1 c1, C2 c2, Object? o) {
  if (o is B1) {
    o = b ? c1 : c2;
  }
}
''');

    var node = result.findNode.conditionalExpression('b ? c1 : c2');
    assertResolvedNodeText(node, r'''
ConditionalExpression
  condition2: UnqualifiedNameExpression
    name: b
    resolution: VariableReadResolution
      element: <testLibrary>::@function::f::@formalParameter::b
      type: bool
    staticType: bool
  condition(v1): SimpleIdentifier
    token: b
    element: <testLibrary>::@function::f::@formalParameter::b
    staticType: bool
  question: ?
  thenExpression2: UnqualifiedNameExpression
    name: c1
    resolution: VariableReadResolution
      element: <testLibrary>::@function::f::@formalParameter::c1
      type: C1
    staticType: C1
  thenExpression(v1): SimpleIdentifier
    token: c1
    element: <testLibrary>::@function::f::@formalParameter::c1
    staticType: C1
  colon: :
  elseExpression2: UnqualifiedNameExpression
    name: c2
    resolution: VariableReadResolution
      element: <testLibrary>::@function::f::@formalParameter::c2
      type: C2
    staticType: C2
  elseExpression(v1): SimpleIdentifier
    token: c2
    element: <testLibrary>::@function::f::@formalParameter::c2
    staticType: C2
  correspondingParameter: <null>
  staticType: A
''');
  }

  test_issue49692() async {
    var result = await resolveTestCodeWithDiagnostics('''
T f<T>(T t, bool b) {
  if (t is int) {
    final u = b ? t : null;
    return u;
//         ^
// [diag.returnOfInvalidTypeFromFunction] A value of type 'int?' can't be returned from the function 'f' because it has a return type of 'T'.
  } else {
    return t;
  }
}
''');

    var node = result.findNode.conditionalExpression('b ?');
    assertResolvedNodeText(node, r'''
ConditionalExpression
  condition2: UnqualifiedNameExpression
    name: b
    resolution: VariableReadResolution
      element: <testLibrary>::@function::f::@formalParameter::b
      type: bool
    staticType: bool
  condition(v1): SimpleIdentifier
    token: b
    element: <testLibrary>::@function::f::@formalParameter::b
    staticType: bool
  question: ?
  thenExpression2: UnqualifiedNameExpression
    name: t
    resolution: VariableReadResolution
      element: <testLibrary>::@function::f::@formalParameter::t
      type: T & int
    staticType: T & int
  thenExpression(v1): SimpleIdentifier
    token: t
    element: <testLibrary>::@function::f::@formalParameter::t
    staticType: T & int
  colon: :
  elseExpression2: NullLiteral
    literal: null
    staticType: Null
  staticType: int?
''');
  }

  test_recordType_differentShape() async {
    var result = await resolveTestCodeWithDiagnostics('''
void f(bool b, (int, String) r1, ({int a}) r2) {
  b ? r1 : r2;
}
''');

    var node = result.findNode.conditionalExpression('b ?');
    assertResolvedNodeText(node, r'''
ConditionalExpression
  condition2: UnqualifiedNameExpression
    name: b
    resolution: VariableReadResolution
      element: <testLibrary>::@function::f::@formalParameter::b
      type: bool
    staticType: bool
  condition(v1): SimpleIdentifier
    token: b
    element: <testLibrary>::@function::f::@formalParameter::b
    staticType: bool
  question: ?
  thenExpression2: UnqualifiedNameExpression
    name: r1
    resolution: VariableReadResolution
      element: <testLibrary>::@function::f::@formalParameter::r1
      type: (int, String)
    staticType: (int, String)
  thenExpression(v1): SimpleIdentifier
    token: r1
    element: <testLibrary>::@function::f::@formalParameter::r1
    staticType: (int, String)
  colon: :
  elseExpression2: UnqualifiedNameExpression
    name: r2
    resolution: VariableReadResolution
      element: <testLibrary>::@function::f::@formalParameter::r2
      type: ({int a})
    staticType: ({int a})
  elseExpression(v1): SimpleIdentifier
    token: r2
    element: <testLibrary>::@function::f::@formalParameter::r2
    staticType: ({int a})
  staticType: Record
''');
  }

  test_recordType_sameShape_named() async {
    var result = await resolveTestCodeWithDiagnostics('''
void f(bool b, ({int a}) r1, ({double a}) r2) {
  b ? r1 : r2;
}
''');

    var node = result.findNode.conditionalExpression('b ?');
    assertResolvedNodeText(node, r'''
ConditionalExpression
  condition2: UnqualifiedNameExpression
    name: b
    resolution: VariableReadResolution
      element: <testLibrary>::@function::f::@formalParameter::b
      type: bool
    staticType: bool
  condition(v1): SimpleIdentifier
    token: b
    element: <testLibrary>::@function::f::@formalParameter::b
    staticType: bool
  question: ?
  thenExpression2: UnqualifiedNameExpression
    name: r1
    resolution: VariableReadResolution
      element: <testLibrary>::@function::f::@formalParameter::r1
      type: ({int a})
    staticType: ({int a})
  thenExpression(v1): SimpleIdentifier
    token: r1
    element: <testLibrary>::@function::f::@formalParameter::r1
    staticType: ({int a})
  colon: :
  elseExpression2: UnqualifiedNameExpression
    name: r2
    resolution: VariableReadResolution
      element: <testLibrary>::@function::f::@formalParameter::r2
      type: ({double a})
    staticType: ({double a})
  elseExpression(v1): SimpleIdentifier
    token: r2
    element: <testLibrary>::@function::f::@formalParameter::r2
    staticType: ({double a})
  staticType: ({num a})
''');
  }

  test_then_super() async {
    var result = await resolveTestCodeWithDiagnostics('''
class A {
  void f(bool c) {
    c ? super : 0;
//      ^^^^^
// [diag.missingAssignableSelector] Missing selector such as '.identifier' or '[0]'.
  }
}
''');

    var node = result.findNode.singleConditionalExpression;
    assertResolvedNodeText(node, r'''
ConditionalExpression
  condition2: UnqualifiedNameExpression
    name: c
    resolution: VariableReadResolution
      element: <testLibrary>::@class::A::@method::f::@formalParameter::c
      type: bool
    staticType: bool
  condition(v1): SimpleIdentifier
    token: c
    element: <testLibrary>::@class::A::@method::f::@formalParameter::c
    staticType: bool
  question: ?
  thenExpression2: InvalidSuperExpression
    superReference: SuperReference
      superKeyword: super
    staticType: InvalidType
  thenExpression(v1): SuperExpression
    superKeyword: super
    staticType: A
  colon: :
  elseExpression2: IntegerLiteral
    literal: 0
    staticType: int
  staticType: InvalidType
''');
  }

  test_type_int_double() async {
    var result = await resolveTestCodeWithDiagnostics('''
void f(bool b) {
  b ? 0 : 1.2;
}
''');

    var node = result.findNode.singleConditionalExpression;
    assertResolvedNodeText(node, r'''
ConditionalExpression
  condition2: UnqualifiedNameExpression
    name: b
    resolution: VariableReadResolution
      element: <testLibrary>::@function::f::@formalParameter::b
      type: bool
    staticType: bool
  condition(v1): SimpleIdentifier
    token: b
    element: <testLibrary>::@function::f::@formalParameter::b
    staticType: bool
  question: ?
  thenExpression2: IntegerLiteral
    literal: 0
    staticType: int
  colon: :
  elseExpression2: DoubleLiteral
    literal: 1.2
    staticType: double
  staticType: num
''');
  }

  test_type_int_null() async {
    var result = await resolveTestCodeWithDiagnostics('''
void f(bool b) {
  b ? 42 : null;
}
''');

    var node = result.findNode.singleConditionalExpression;
    assertResolvedNodeText(node, r'''
ConditionalExpression
  condition2: UnqualifiedNameExpression
    name: b
    resolution: VariableReadResolution
      element: <testLibrary>::@function::f::@formalParameter::b
      type: bool
    staticType: bool
  condition(v1): SimpleIdentifier
    token: b
    element: <testLibrary>::@function::f::@formalParameter::b
    staticType: bool
  question: ?
  thenExpression2: IntegerLiteral
    literal: 42
    staticType: int
  colon: :
  elseExpression2: NullLiteral
    literal: null
    staticType: Null
  staticType: int?
''');
  }

  test_upward() async {
    var result = await resolveTestCodeWithDiagnostics('''
void f(bool a, int b, int c) {
  var d = a ? b : c;
  print(d);
}
''');
    assertType(result.findNode.unqualifiedNameExpression('d)'), 'int');
  }
}

@reflectiveTest
class InferenceUpdate3Test extends PubPackageResolutionTest {
  test_contextIsConvertedToATypeUsingGreatestClosure() async {
    var result = await resolveTestCodeWithDiagnostics('''
class A {}
class B1<T> extends A {}
class B2<T> extends A {}
class C1<T> implements B1<T>, B2<T> {}
class C2<T> implements B1<T>, B2<T> {}
void contextB1<T>(B1<T> b1) {}
f(bool b, C1<int> c1, C2<double> c2) {
  contextB1(b ? c1 : c2);
}
''');

    var node = result.findNode.conditionalExpression('b ? c1 : c2');
    assertResolvedNodeText(node, r'''ConditionalExpression
  condition2: UnqualifiedNameExpression
    name: b
    resolution: VariableReadResolution
      element: <testLibrary>::@function::f::@formalParameter::b
      type: bool
    staticType: bool
  condition(v1): SimpleIdentifier
    token: b
    element: <testLibrary>::@function::f::@formalParameter::b
    staticType: bool
  question: ?
  thenExpression2: UnqualifiedNameExpression
    name: c1
    resolution: VariableReadResolution
      element: <testLibrary>::@function::f::@formalParameter::c1
      type: C1<int>
    staticType: C1<int>
  thenExpression(v1): SimpleIdentifier
    token: c1
    element: <testLibrary>::@function::f::@formalParameter::c1
    staticType: C1<int>
  colon: :
  elseExpression2: UnqualifiedNameExpression
    name: c2
    resolution: VariableReadResolution
      element: <testLibrary>::@function::f::@formalParameter::c2
      type: C2<double>
    staticType: C2<double>
  elseExpression(v1): SimpleIdentifier
    token: c2
    element: <testLibrary>::@function::f::@formalParameter::c2
    staticType: C2<double>
  correspondingParameter: SubstitutedFormalParameterElementImpl
    baseElement: <testLibrary>::@function::contextB1::@formalParameter::b1
    substitution: {T: Object?}
  staticType: B1<Object?>
''');
  }

  test_contextNotUsedIfLhsDoesNotSatisfyContext() async {
    var result = await resolveTestCodeWithDiagnostics('''
class A {}
class B1 extends A {}
class B2 extends A {}
class C1 implements B1, B2 {}
class C2 implements B1, B2 {}
f(bool b, B2 b2, C1 c1, Object? o) {
  if (o is B1) {
    o = b ? b2 : c1;
  }
}
''');

    var node = result.findNode.conditionalExpression('b ? b2 : c1');
    assertResolvedNodeText(node, r'''ConditionalExpression
  condition2: UnqualifiedNameExpression
    name: b
    resolution: VariableReadResolution
      element: <testLibrary>::@function::f::@formalParameter::b
      type: bool
    staticType: bool
  condition(v1): SimpleIdentifier
    token: b
    element: <testLibrary>::@function::f::@formalParameter::b
    staticType: bool
  question: ?
  thenExpression2: UnqualifiedNameExpression
    name: b2
    resolution: VariableReadResolution
      element: <testLibrary>::@function::f::@formalParameter::b2
      type: B2
    staticType: B2
  thenExpression(v1): SimpleIdentifier
    token: b2
    element: <testLibrary>::@function::f::@formalParameter::b2
    staticType: B2
  colon: :
  elseExpression2: UnqualifiedNameExpression
    name: c1
    resolution: VariableReadResolution
      element: <testLibrary>::@function::f::@formalParameter::c1
      type: C1
    staticType: C1
  elseExpression(v1): SimpleIdentifier
    token: c1
    element: <testLibrary>::@function::f::@formalParameter::c1
    staticType: C1
  correspondingParameter: <null>
  staticType: B2
''');
  }

  test_contextNotUsedIfRhsDoesNotSatisfyContext() async {
    var result = await resolveTestCodeWithDiagnostics('''
class A {}
class B1 extends A {}
class B2 extends A {}
class C1 implements B1, B2 {}
class C2 implements B1, B2 {}
f(bool b, C1 c1, B2 b2, Object? o) {
  if (o is B1) {
    o = b ? c1 : b2;
  }
}
''');

    var node = result.findNode.conditionalExpression('b ? c1 : b2');
    assertResolvedNodeText(node, r'''ConditionalExpression
  condition2: UnqualifiedNameExpression
    name: b
    resolution: VariableReadResolution
      element: <testLibrary>::@function::f::@formalParameter::b
      type: bool
    staticType: bool
  condition(v1): SimpleIdentifier
    token: b
    element: <testLibrary>::@function::f::@formalParameter::b
    staticType: bool
  question: ?
  thenExpression2: UnqualifiedNameExpression
    name: c1
    resolution: VariableReadResolution
      element: <testLibrary>::@function::f::@formalParameter::c1
      type: C1
    staticType: C1
  thenExpression(v1): SimpleIdentifier
    token: c1
    element: <testLibrary>::@function::f::@formalParameter::c1
    staticType: C1
  colon: :
  elseExpression2: UnqualifiedNameExpression
    name: b2
    resolution: VariableReadResolution
      element: <testLibrary>::@function::f::@formalParameter::b2
      type: B2
    staticType: B2
  elseExpression(v1): SimpleIdentifier
    token: b2
    element: <testLibrary>::@function::f::@formalParameter::b2
    staticType: B2
  correspondingParameter: <null>
  staticType: B2
''');
  }

  test_contextUsedInsteadOfLubIfLubDoesNotSatisfyContext() async {
    var result = await resolveTestCodeWithDiagnostics('''
class A {}
class B1 extends A {}
class B2 extends A {}
class C1 implements B1, B2 {}
class C2 implements B1, B2 {}
B1 f(bool b, C1 c1, C2 c2) => b ? c1 : c2;
''');

    var node = result.findNode.conditionalExpression('b ? c1 : c2');
    assertResolvedNodeText(node, r'''
ConditionalExpression
  condition2: UnqualifiedNameExpression
    name: b
    resolution: VariableReadResolution
      element: <testLibrary>::@function::f::@formalParameter::b
      type: bool
    staticType: bool
  condition(v1): SimpleIdentifier
    token: b
    element: <testLibrary>::@function::f::@formalParameter::b
    staticType: bool
  question: ?
  thenExpression2: UnqualifiedNameExpression
    name: c1
    resolution: VariableReadResolution
      element: <testLibrary>::@function::f::@formalParameter::c1
      type: C1
    staticType: C1
  thenExpression(v1): SimpleIdentifier
    token: c1
    element: <testLibrary>::@function::f::@formalParameter::c1
    staticType: C1
  colon: :
  elseExpression2: UnqualifiedNameExpression
    name: c2
    resolution: VariableReadResolution
      element: <testLibrary>::@function::f::@formalParameter::c2
      type: C2
    staticType: C2
  elseExpression(v1): SimpleIdentifier
    token: c2
    element: <testLibrary>::@function::f::@formalParameter::c2
    staticType: C2
  staticType: B1
''');
  }
}