type migration tests fixed
[idea/community.git] / python / src / com / jetbrains / python / testing / nosetest / PythonNoseTestCommandLineState.java
1 /*
2  * Copyright 2000-2014 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.nosetest;
17
18 import com.intellij.execution.configurations.GeneralCommandLine;
19 import com.intellij.execution.configurations.ParamsGroup;
20 import com.intellij.execution.runners.ExecutionEnvironment;
21 import com.intellij.openapi.util.io.FileUtil;
22 import com.intellij.openapi.util.text.StringUtil;
23 import com.jetbrains.python.PythonHelper;
24 import com.jetbrains.python.testing.PythonTestCommandLineStateBase;
25
26 import java.util.ArrayList;
27 import java.util.List;
28
29 /**
30  * User: catherine
31  */
32 public class PythonNoseTestCommandLineState extends PythonTestCommandLineStateBase {
33   private final PythonNoseTestRunConfiguration myConfig;
34
35   public PythonNoseTestCommandLineState(PythonNoseTestRunConfiguration runConfiguration, ExecutionEnvironment env) {
36     super(runConfiguration, env);
37     myConfig = runConfiguration;
38   }
39
40   @Override
41   protected PythonHelper getRunner() {
42     return PythonHelper.NOSE;
43   }
44
45   protected List<String> getTestSpecs() {
46     List<String> specs = new ArrayList<String>();
47
48     final String scriptName = FileUtil.toSystemDependentName(myConfig.getScriptName());
49     switch (myConfig.getTestType()) {
50       case TEST_SCRIPT:
51         specs.add(scriptName);
52         break;
53       case TEST_CLASS:
54         specs.add(scriptName + "::" + myConfig.getClassName());
55         break;
56       case TEST_METHOD:
57         specs.add(scriptName + "::" + myConfig.getClassName() + "::" + myConfig.getMethodName());
58         break;
59       case TEST_FOLDER:
60         specs.add(FileUtil.toSystemDependentName(myConfig.getFolderName() + "/"));
61         break;
62       case TEST_FUNCTION:
63         specs.add(scriptName + "::::" + myConfig.getMethodName());
64         break;
65       default:
66         throw new IllegalArgumentException("Unknown test type: " + myConfig.getTestType());
67     }
68     return specs;
69   }
70
71   @Override
72   protected void addAfterParameters(GeneralCommandLine cmd)  {
73     ParamsGroup script_params = cmd.getParametersList().getParamsGroup(GROUP_SCRIPT);
74     assert script_params != null;
75     if (myConfig.useParam() && !StringUtil.isEmptyOrSpaces(myConfig.getParams()))
76       script_params.addParametersString(myConfig.getParams());
77
78   }
79 }