// Copyright (c) 2016, 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.

@TestOn("vm")

import 'package:test_descriptor/test_descriptor.dart' as d;

import 'package:test/src/util/exit_codes.dart' as exit_codes;
import 'package:test/test.dart';

import '../io.dart';

void main() {
  test("divides all the tests among the available shards", () async {
    await d.file("test.dart", """
      import 'package:test/test.dart';

      void main() {
        test("test 1", () {});
        test("test 2", () {});
        test("test 3", () {});
        test("test 4", () {});
        test("test 5", () {});
        test("test 6", () {});
        test("test 7", () {});
        test("test 8", () {});
        test("test 9", () {});
        test("test 10", () {});
      }
    """).create();

    var test =
        await runTest(["test.dart", "--shard-index=0", "--total-shards=3"]);
    expect(
        test.stdout,
        containsInOrder([
          "+0: test 1",
          "+1: test 2",
          "+2: test 3",
          "+3: All tests passed!"
        ]));
    await test.shouldExit(0);

    test = await runTest(["test.dart", "--shard-index=1", "--total-shards=3"]);
    expect(
        test.stdout,
        containsInOrder([
          "+0: test 4",
          "+1: test 5",
          "+2: test 6",
          "+3: test 7",
          "+4: All tests passed!"
        ]));
    await test.shouldExit(0);

    test = await runTest(["test.dart", "--shard-index=2", "--total-shards=3"]);
    expect(
        test.stdout,
        containsInOrder([
          "+0: test 8",
          "+1: test 9",
          "+2: test 10",
          "+3: All tests passed!"
        ]));
    await test.shouldExit(0);
  });

  test("shards each suite", () async {
    await d.file("1_test.dart", """
      import 'package:test/test.dart';

      void main() {
        test("test 1.1", () {});
        test("test 1.2", () {});
        test("test 1.3", () {});
      }
    """).create();

    await d.file("2_test.dart", """
      import 'package:test/test.dart';

      void main() {
        test("test 2.1", () {});
        test("test 2.2", () {});
        test("test 2.3", () {});
      }
    """).create();

    var test = await runTest([".", "--shard-index=0", "--total-shards=3"]);
    expect(
        test.stdout,
        emitsInOrder([
          emitsAnyOf([
            containsInOrder(
                ["+0: ./1_test.dart: test 1.1", "+1: ./2_test.dart: test 2.1"]),
            containsInOrder(
                ["+0: ./2_test.dart: test 2.1", "+1: ./1_test.dart: test 1.1"])
          ]),
          contains("+2: All tests passed!")
        ]));
    await test.shouldExit(0);

    test = await runTest([".", "--shard-index=1", "--total-shards=3"]);
    expect(
        test.stdout,
        emitsInOrder([
          emitsAnyOf([
            containsInOrder(
                ["+0: ./1_test.dart: test 1.2", "+1: ./2_test.dart: test 2.2"]),
            containsInOrder(
                ["+0: ./2_test.dart: test 2.2", "+1: ./1_test.dart: test 1.2"])
          ]),
          contains("+2: All tests passed!")
        ]));
    await test.shouldExit(0);

    test = await runTest([".", "--shard-index=2", "--total-shards=3"]);
    expect(
        test.stdout,
        emitsInOrder([
          emitsAnyOf([
            containsInOrder(
                ["+0: ./1_test.dart: test 1.3", "+1: ./2_test.dart: test 2.3"]),
            containsInOrder(
                ["+0: ./2_test.dart: test 2.3", "+1: ./1_test.dart: test 1.3"])
          ]),
          contains("+2: All tests passed!")
        ]));
    await test.shouldExit(0);
  });

  test("an empty shard reports success", () async {
    await d.file("test.dart", """
      import 'package:test/test.dart';

      void main() {
        test("test 1", () {});
        test("test 2", () {});
      }
    """).create();

    var test =
        await runTest(["test.dart", "--shard-index=1", "--total-shards=3"]);
    expect(test.stdout, emitsThrough("No tests ran."));
    await test.shouldExit(0);
  });

  group("reports an error if", () {
    test("--shard-index is provided alone", () async {
      var test = await runTest(["--shard-index=1"]);
      expect(
          test.stderr,
          emits(
              "--shard-index and --total-shards may only be passed together."));
      await test.shouldExit(exit_codes.usage);
    });

    test("--total-shards is provided alone", () async {
      var test = await runTest(["--total-shards=5"]);
      expect(
          test.stderr,
          emits(
              "--shard-index and --total-shards may only be passed together."));
      await test.shouldExit(exit_codes.usage);
    });

    test("--shard-index is negative", () async {
      var test = await runTest(["--shard-index=-1", "--total-shards=5"]);
      expect(test.stderr, emits("--shard-index may not be negative."));
      await test.shouldExit(exit_codes.usage);
    });

    test("--shard-index is equal to --total-shards", () async {
      var test = await runTest(["--shard-index=5", "--total-shards=5"]);
      expect(test.stderr,
          emits("--shard-index must be less than --total-shards."));
      await test.shouldExit(exit_codes.usage);
    });
  });
}