| /* |
| * Copyright 2016 The Bazel Authors. 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.idea.blaze.android.run.test; |
| |
| import com.android.tools.idea.run.ValidationError; |
| import com.google.common.collect.ImmutableList; |
| import com.google.common.collect.Lists; |
| import com.google.idea.blaze.android.run.BlazeAndroidRunConfigurationCommonState; |
| import com.google.idea.blaze.android.run.BlazeAndroidRunConfigurationHandler; |
| import com.google.idea.blaze.android.run.BlazeAndroidRunConfigurationValidationUtil; |
| import com.google.idea.blaze.android.run.runner.BlazeAndroidRunConfigurationRunner; |
| import com.google.idea.blaze.android.run.runner.BlazeAndroidRunContext; |
| import com.google.idea.blaze.android.sync.projectstructure.BlazeAndroidProjectStructureSyncer; |
| import com.google.idea.blaze.base.model.primitives.Kind; |
| import com.google.idea.blaze.base.model.primitives.Label; |
| import com.google.idea.blaze.base.model.primitives.TargetExpression; |
| import com.google.idea.blaze.base.projectview.ProjectViewManager; |
| import com.google.idea.blaze.base.projectview.ProjectViewSet; |
| import com.google.idea.blaze.base.run.BlazeCommandRunConfiguration; |
| import com.google.idea.blaze.base.run.BlazeConfigurationNameBuilder; |
| import com.google.idea.blaze.base.run.confighandler.BlazeCommandRunConfigurationRunner; |
| import com.google.idea.blaze.base.settings.Blaze; |
| import com.intellij.execution.ExecutionException; |
| import com.intellij.execution.Executor; |
| import com.intellij.execution.JavaExecutionUtil; |
| import com.intellij.execution.configurations.RunConfiguration; |
| import com.intellij.execution.configurations.RuntimeConfigurationException; |
| import com.intellij.execution.runners.ExecutionEnvironment; |
| import com.intellij.openapi.module.Module; |
| import com.intellij.openapi.project.Project; |
| import java.util.List; |
| import javax.swing.Icon; |
| import org.jetbrains.android.facet.AndroidFacet; |
| import org.jetbrains.annotations.Nullable; |
| |
| /** |
| * {@link com.google.idea.blaze.base.run.confighandler.BlazeCommandRunConfigurationHandler} for |
| * android_test targets. |
| */ |
| public class BlazeAndroidTestRunConfigurationHandler |
| implements BlazeAndroidRunConfigurationHandler { |
| |
| private final BlazeCommandRunConfiguration configuration; |
| private final BlazeAndroidTestRunConfigurationState configState; |
| |
| BlazeAndroidTestRunConfigurationHandler(BlazeCommandRunConfiguration configuration) { |
| this.configuration = configuration; |
| configState = |
| new BlazeAndroidTestRunConfigurationState( |
| Blaze.buildSystemName(configuration.getProject())); |
| } |
| |
| @Override |
| public BlazeAndroidTestRunConfigurationState getState() { |
| return configState; |
| } |
| |
| @Override |
| public BlazeAndroidRunConfigurationCommonState getCommonState() { |
| return configState.getCommonState(); |
| } |
| |
| @Override |
| @Nullable |
| public Label getLabel() { |
| TargetExpression target = configuration.getTarget(); |
| if (target instanceof Label) { |
| return (Label) target; |
| } |
| return null; |
| } |
| |
| @Nullable |
| private Module getModule() { |
| Label target = getLabel(); |
| return target != null |
| ? BlazeAndroidProjectStructureSyncer.ensureRunConfigurationModule( |
| configuration.getProject(), target) |
| : null; |
| } |
| |
| @Override |
| public BlazeCommandRunConfigurationRunner createRunner( |
| Executor executor, ExecutionEnvironment environment) throws ExecutionException { |
| Project project = environment.getProject(); |
| |
| Module module = getModule(); |
| AndroidFacet facet = module != null ? AndroidFacet.getInstance(module) : null; |
| ProjectViewSet projectViewSet = ProjectViewManager.getInstance(project).getProjectViewSet(); |
| BlazeAndroidRunConfigurationValidationUtil.validateExecution(module, facet, projectViewSet); |
| |
| ImmutableList<String> buildFlags = configState.getBuildFlags(project, projectViewSet); |
| BlazeAndroidRunContext runContext = createRunContext(project, facet, environment, buildFlags); |
| |
| return new BlazeAndroidRunConfigurationRunner( |
| module, |
| runContext, |
| getCommonState().getDeployTargetManager(), |
| getCommonState().getDebuggerManager(), |
| configuration.getUniqueID()); |
| } |
| |
| private BlazeAndroidRunContext createRunContext( |
| Project project, |
| AndroidFacet facet, |
| ExecutionEnvironment env, |
| ImmutableList<String> buildFlags) { |
| return new BlazeAndroidTestRunContext( |
| project, facet, configuration, env, configState, getLabel(), buildFlags); |
| } |
| |
| @Override |
| public final void checkConfiguration() throws RuntimeConfigurationException { |
| BlazeAndroidRunConfigurationValidationUtil.throwTopConfigurationError(validate()); |
| } |
| |
| /** |
| * We collect errors rather than throwing to avoid missing fatal errors by exiting early for a |
| * warning. We use a separate method for the collection so the compiler prevents us from |
| * accidentally throwing. |
| */ |
| private List<ValidationError> validate() { |
| List<ValidationError> errors = Lists.newArrayList(); |
| Module module = getModule(); |
| errors.addAll(BlazeAndroidRunConfigurationValidationUtil.validateModule(module)); |
| AndroidFacet facet = null; |
| if (module != null) { |
| facet = AndroidFacet.getInstance(module); |
| errors.addAll(BlazeAndroidRunConfigurationValidationUtil.validateFacet(facet, module)); |
| } |
| errors.addAll(configState.validate(facet)); |
| errors.addAll( |
| BlazeAndroidRunConfigurationValidationUtil.validateLabel( |
| getLabel(), configuration.getProject(), Kind.ANDROID_TEST)); |
| return errors; |
| } |
| |
| @Override |
| @Nullable |
| public String suggestedName(BlazeCommandRunConfiguration configuration) { |
| Label target = getLabel(); |
| if (target == null) { |
| return null; |
| } |
| BlazeConfigurationNameBuilder nameBuilder = |
| new BlazeConfigurationNameBuilder(this.configuration); |
| |
| boolean isClassTest = |
| configState.getTestingType() == BlazeAndroidTestRunConfigurationState.TEST_CLASS; |
| boolean isMethodTest = |
| configState.getTestingType() == BlazeAndroidTestRunConfigurationState.TEST_METHOD; |
| if ((isClassTest || isMethodTest) && configState.getClassName() != null) { |
| // Get the class name without the package. |
| String className = JavaExecutionUtil.getPresentableClassName(configState.getClassName()); |
| if (className != null) { |
| String targetString = className; |
| if (isMethodTest) { |
| targetString += "#" + configState.getMethodName(); |
| } |
| return nameBuilder.setTargetString(targetString).build(); |
| } |
| } |
| return nameBuilder.build(); |
| } |
| |
| @Override |
| @Nullable |
| public String getCommandName() { |
| return "test"; |
| } |
| |
| @Override |
| public String getHandlerName() { |
| return "Android Test Handler"; |
| } |
| |
| @Override |
| @Nullable |
| public Icon getExecutorIcon(RunConfiguration configuration, Executor executor) { |
| return null; |
| } |
| } |