// 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.

/// @assertion If element is a forElement with element e1 then:
///
/// Inference for the iterated expression and the controlling variable is done as
/// for the corresponding for or await for statement.
///
/// If the inferred set element type of e1 is S then the inferred set element
/// type of element is S.
///
/// @description Checks that if element is a forElement with element e1 then if
/// the inferred set element type of e1 is S then the inferred set element type
/// of element is S.
/// @author sgrekhov@unipro.ru

import "../../Utils/expect.dart";

main() {
  var c1 = [1, 2, 3] as List<int>?;
  var c2 = [1.1, 2.2, 3.3] as List<double>?;
  var set1 = {
    for (var i = 1; i < 5; i++)
      if (i.isEven) ...?c1 else ...?c2
  };
  Expect.isTrue(set1 is Set<num>);
  Expect.isFalse(set1 is Set<int>);
  Expect.isFalse(set1 is Set<double>);
  checkType(checkIs<Set<num>>, true, set1);
  checkType(checkIs<Set<int>>, false, set1);
  checkType(checkIs<Set<double>>, false, set1);

  var set2 = {
    "",
    for (var i = 1; i < 5; i++)
      if (i.isEven) ...?c1 else ...?c2
  };
  Expect.isTrue(set2 is Set<Object>);
  Expect.isFalse(set2 is Set<int>);
  Expect.isFalse(set2 is Set<double>);
  checkType(checkIs<Set<Object>>, true, set2);
  checkType(checkIs<Set<int>>, false, set2);
  checkType(checkIs<Set<double>>, false, set2);

  c1 = null;
  c2 = null;
  var set3 = {
    for (var i = 1; i < 5; i++)
      if (i.isEven) ...?c1 else ...?c2
  };
  Expect.isTrue(set3 is Set<num>);
  Expect.isFalse(set3 is Set<int>);
  Expect.isFalse(set3 is Set<double>);
  checkType(checkIs<Set<num>>, true, set3);
  checkType(checkIs<Set<int>>, false, set3);
  checkType(checkIs<Set<double>>, false, set3);
}