Rd: update to 2022.2.0
[idea/community.git] / python / src / com / jetbrains / python / testing / tox / PyToxCommandLineState.java
1 /*
2  * Copyright 2000-2016 JetBrains s.r.o.
3  *
4  * Licensed under the Apache License, Version 2.0 (the "License");
5  * you may not use this file except in compliance with the License.
6  * You may obtain a copy of the License at
7  *
8  * http://www.apache.org/licenses/LICENSE-2.0
9  *
10  * Unless required by applicable law or agreed to in writing, software
11  * distributed under the License is distributed on an "AS IS" BASIS,
12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13  * See the License for the specific language governing permissions and
14  * limitations under the License.
15  */
16 package com.jetbrains.python.testing.tox;
17
18 import com.intellij.execution.runners.ExecutionEnvironment;
19 import com.intellij.execution.target.TargetEnvironment;
20 import com.intellij.execution.target.TargetEnvironmentRequest;
21 import com.intellij.execution.target.value.TargetEnvironmentFunctions;
22 import com.intellij.execution.testframework.sm.runner.SMTestLocator;
23 import com.intellij.util.containers.ContainerUtil;
24 import com.jetbrains.python.HelperPackage;
25 import com.jetbrains.python.PythonHelper;
26 import com.jetbrains.python.run.PythonScriptExecution;
27 import com.jetbrains.python.testing.PythonTestCommandLineStateBase;
28 import org.jetbrains.annotations.NotNull;
29 import org.jetbrains.annotations.Nullable;
30
31 import java.util.Arrays;
32 import java.util.List;
33 import java.util.function.Function;
34
35 import static com.intellij.execution.target.value.TargetEnvironmentFunctions.constant;
36 import static com.intellij.execution.target.value.TargetEnvironmentFunctions.joinToStringFunction;
37
38 /**
39  * @author Ilya.Kazakevich
40  */
41 class PyToxCommandLineState extends PythonTestCommandLineStateBase<PyToxConfiguration> {
42   PyToxCommandLineState(@NotNull final PyToxConfiguration configuration,
43                         @NotNull final ExecutionEnvironment environment) {
44     super(configuration, environment);
45   }
46
47   @Nullable
48   @Override
49   protected SMTestLocator getTestLocator() {
50     return PyToxTestLocator.INSTANCE;
51   }
52
53   @Override
54   protected HelperPackage getRunner() {
55     return PythonHelper.TOX;
56   }
57
58   @Override
59   protected void addTestSpecsAsParameters(@NotNull PythonScriptExecution testScriptExecution,
60                                           @NotNull List<Function<TargetEnvironment, String>> testSpecs) {
61     if (!testSpecs.isEmpty()) {
62       Function<TargetEnvironment, String> envOptionValue = joinToStringFunction(testSpecs, ",");
63       testScriptExecution.addParameter(joinToStringFunction(List.of(constant("-e"), envOptionValue), " "));
64     }
65     for (String argument : myConfiguration.getArguments()) {
66       testScriptExecution.addParameter(argument);
67     }
68   }
69
70   /**
71    * <i>To be deprecated. The part of the legacy implementation based on
72    * {@link com.intellij.execution.configurations.GeneralCommandLine}.</i>
73    */
74   @NotNull
75   @Override
76   protected List<String> getTestSpecs() {
77     return Arrays.asList(myConfiguration.getRunOnlyEnvs());
78   }
79
80   @Override
81   protected @NotNull List<Function<TargetEnvironment, String>> getTestSpecs(@NotNull TargetEnvironmentRequest request) {
82     return ContainerUtil.map(myConfiguration.getRunOnlyEnvs(), TargetEnvironmentFunctions::constant);
83   }
84 }