1 files changed, 161 insertions, 0 deletions
diff --git a/src/main/java/com/google/devtools/build/lib/runtime/commands/TestCommand.java b/src/main/java/com/google/devtools/build/lib/runtime/commands/TestCommand.java
new file mode 100644
index 0000000000..561c54a5b2
--- /dev/null
+++ b/src/main/java/com/google/devtools/build/lib/runtime/commands/TestCommand.java
@@ -0,0 +1,161 @@
+// Copyright 2014 Google Inc. All rights reserved.
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//    http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+package com.google.devtools.build.lib.runtime.commands;
+
+import com.google.common.collect.ImmutableList;
+import com.google.devtools.build.lib.analysis.ConfiguredTarget;
+import com.google.devtools.build.lib.buildtool.BuildRequest;
+import com.google.devtools.build.lib.buildtool.BuildResult;
+import com.google.devtools.build.lib.events.Event;
+import com.google.devtools.build.lib.exec.ExecutionOptions;
+import com.google.devtools.build.lib.rules.test.TestStrategy;
+import com.google.devtools.build.lib.rules.test.TestStrategy.TestOutputFormat;
+import com.google.devtools.build.lib.runtime.AggregatingTestListener;
+import com.google.devtools.build.lib.runtime.BlazeCommand;
+import com.google.devtools.build.lib.runtime.BlazeCommandEventHandler;
+import com.google.devtools.build.lib.runtime.BlazeRuntime;
+import com.google.devtools.build.lib.runtime.Command;
+import com.google.devtools.build.lib.runtime.TerminalTestResultNotifier;
+import com.google.devtools.build.lib.runtime.TerminalTestResultNotifier.TestSummaryOptions;
+import com.google.devtools.build.lib.runtime.TestResultAnalyzer;
+import com.google.devtools.build.lib.runtime.TestResultNotifier;
+import com.google.devtools.build.lib.util.AbruptExitException;
+import com.google.devtools.build.lib.util.ExitCode;
+import com.google.devtools.build.lib.util.io.AnsiTerminalPrinter;
+import com.google.devtools.common.options.OptionPriority;
+import com.google.devtools.common.options.OptionsParser;
+import com.google.devtools.common.options.OptionsParsingException;
+import com.google.devtools.common.options.OptionsProvider;
+
+import java.util.Collection;
+import java.util.List;
+
+/**
+ * Handles the 'test' command on the Blaze command line.
+ */
+@Command(name = "test",
+         builds = true,
+         inherits = { BuildCommand.class },
+         options = { TestSummaryOptions.class },
+         shortDescription = "Builds and runs the specified test targets.",
+         help = "resource:test.txt",
+         allowResidue = true)
+public class TestCommand implements BlazeCommand {
+  private AnsiTerminalPrinter printer;
+
+  @Override
+  public void editOptions(BlazeRuntime runtime, OptionsParser optionsParser)
+      throws AbruptExitException {
+    ProjectFileSupport.handleProjectFiles(runtime, optionsParser, "test");
+
+    TestOutputFormat testOutput = optionsParser.getOptions(ExecutionOptions.class).testOutput;
+
+    if (testOutput == TestStrategy.TestOutputFormat.STREAMED) {
+      runtime.getReporter().handle(Event.warn(
+          "Streamed test output requested so all tests will be run locally, without sharding, " +
+           "one at a time"));
+      try {
+        optionsParser.parse(OptionPriority.SOFTWARE_REQUIREMENT,
+            "streamed output requires locally run tests, without sharding",
+            ImmutableList.of("--test_sharding_strategy=disabled", "--test_strategy=exclusive"));
+      } catch (OptionsParsingException e) {
+        throw new IllegalStateException("Known options failed to parse", e);
+      }
+    }
+  }
+
+  @Override
+  public ExitCode exec(BlazeRuntime runtime, OptionsProvider options) {
+    TestResultAnalyzer resultAnalyzer = new TestResultAnalyzer(
+        runtime.getExecRoot(),
+        options.getOptions(TestSummaryOptions.class),
+        options.getOptions(ExecutionOptions.class),
+        runtime.getEventBus());
+
+    printer = new AnsiTerminalPrinter(runtime.getReporter().getOutErr().getOutputStream(),
+        options.getOptions(BlazeCommandEventHandler.Options.class).useColor());
+
+    // Initialize test handler.
+    AggregatingTestListener testListener = new AggregatingTestListener(
+        resultAnalyzer, runtime.getEventBus(), runtime.getReporter());
+
+    runtime.getEventBus().register(testListener);
+    return doTest(runtime, options, testListener);
+  }
+
+  private ExitCode doTest(BlazeRuntime runtime,
+      OptionsProvider options,
+      AggregatingTestListener testListener) {
+    // Run simultaneous build and test.
+    List<String> targets = ProjectFileSupport.getTargets(runtime, options);
+    BuildRequest request = BuildRequest.create(
+        getClass().getAnnotation(Command.class).name(), options,
+        runtime.getStartupOptionsProvider(), targets,
+        runtime.getReporter().getOutErr(), runtime.getCommandId(), runtime.getCommandStartTime());
+    if (request.getBuildOptions().compileOnly) {
+      String message =  "The '" + getClass().getAnnotation(Command.class).name() +
+                        "' command is incompatible with the --compile_only option";
+      runtime.getReporter().handle(Event.error(message));
+      return ExitCode.COMMAND_LINE_ERROR;
+    }
+    request.setRunTests();
+
+    BuildResult buildResult = runtime.getBuildTool().processRequest(request, null);
+
+    Collection<ConfiguredTarget> testTargets = buildResult.getTestTargets();
+    // TODO(bazel-team): don't handle isEmpty here or fix up a bunch of tests
+    if (buildResult.getSuccessfulTargets() == null) {
+      // This can happen if there were errors in the target parsing or loading phase
+      // (original exitcode=BUILD_FAILURE) or if there weren't but --noanalyze was given
+      // (original exitcode=SUCCESS).
+      runtime.getReporter().handle(Event.error("Couldn't start the build. Unable to run tests"));
+      return buildResult.getSuccess() ? ExitCode.PARSING_FAILURE : buildResult.getExitCondition();
+    }
+    // TODO(bazel-team): the check above shadows NO_TESTS_FOUND, but switching the conditions breaks
+    // more tests
+    if (testTargets.isEmpty()) {
+      runtime.getReporter().handle(Event.error(
+          null, "No test targets were found, yet testing was requested"));
+      return buildResult.getSuccess() ? ExitCode.NO_TESTS_FOUND : buildResult.getExitCondition();
+    }
+
+    boolean buildSuccess = buildResult.getSuccess();
+    boolean testSuccess = analyzeTestResults(testTargets, testListener, options);
+
+    if (testSuccess && !buildSuccess) {
+      // If all tests run successfully, test summary should include warning if
+      // there were build errors not associated with the test targets.
+      printer.printLn(AnsiTerminalPrinter.Mode.ERROR
+          + "One or more non-test targets failed to build.\n"
+          + AnsiTerminalPrinter.Mode.DEFAULT);
+    }
+
+    return buildSuccess ?
+           (testSuccess ? ExitCode.SUCCESS : ExitCode.TESTS_FAILED)
+           : buildResult.getExitCondition();
+  }
+
+  /**
+   * Analyzes test results and prints summary information.
+   * Returns true if and only if all tests were successful.
+   */
+  private boolean analyzeTestResults(Collection<ConfiguredTarget> testTargets,
+                                     AggregatingTestListener listener,
+                                     OptionsProvider options) {
+    TestResultNotifier notifier = new TerminalTestResultNotifier(printer, options);
+    return listener.getAnalyzer().differentialAnalyzeAndReport(
+        testTargets, listener, notifier);
+  }
+}