1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
|
/*
* Copyright (C) 2009 The Android Open Source Project
*
* 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.android.hosttest;
import java.util.ArrayList;
import java.util.List;
import junit.framework.Test;
import junit.framework.TestResult;
import junit.textui.TestRunner;
import com.android.ddmlib.AndroidDebugBridge;
import com.android.ddmlib.IDevice;
import com.android.ddmlib.Log;
import com.android.ddmlib.AndroidDebugBridge.IDeviceChangeListener;
/**
* Command line interface for running DeviceTest tests.
*
* Extends junit.textui.TestRunner to handle optional -s (device serial) and -p (test data)
* arguments, and then pass their values to the instantiated DeviceTests.
*
* Provided test class must be a DeviceTest.
*
* @see junit.textui.TestRunner for more information on command line syntax.
*/
public class DeviceTestRunner extends TestRunner {
private static final String LOG_TAG = "DeviceTestRunner";
private String mDeviceSerial = null;
private IDevice mDevice = null;
private String mTestDataPath = null;
private DeviceTestRunner() {
}
/**
* Starts the test run.
* Extracts out DeviceTestCase specific command line arguments, then passes control to parent
* TestRunner.
* @param args command line arguments
* @return {@link TestResult}
*/
@Override
public TestResult start(String[] args) throws Exception {
// holds unprocessed arguments to pass to parent
List<String> parentArgs = new ArrayList<String>();
for (int i=0; i < args.length; i++) {
if (args[i].equals("-s")) {
i++;
mDeviceSerial = extractArg(args, i);
} else if (args[i].equals("-p")) {
i++;
mTestDataPath = extractArg(args, i);
} else {
// unrecognized arg, must be for parent
parentArgs.add(args[i]);
}
}
DeviceConnector connector = new DeviceConnector();
mDevice = connector.connectToDevice(mDeviceSerial);
return super.start(parentArgs.toArray(new String[parentArgs.size()]));
}
private String extractArg(String[] args, int index) {
if (args.length <= index) {
printUsage();
throw new IllegalArgumentException("Error: not enough arguments");
}
return args[index];
}
/**
* Main entry point.
*
* Establishes connection to provided adb device and runs tests
*
* @param args expects:
* test class to run
* optionally, device serial number. If unspecified, will connect to first device found
* optionally, file system path to test data files
*/
public static void main(String[] args) {
DeviceTestRunner aTestRunner = new DeviceTestRunner();
try {
TestResult r = aTestRunner.start(args);
if (!r.wasSuccessful())
System.exit(FAILURE_EXIT);
System.exit(SUCCESS_EXIT);
} catch(Exception e) {
System.err.println(e.getMessage());
System.exit(EXCEPTION_EXIT);
}
}
private static void printUsage() {
System.out.println("Usage: DeviceTestRunner <test_class> [-s device_serial] " +
"[-p test_data_path]");
}
/**
* Override parent to set DeviceTest data
*/
@Override
public TestResult doRun(Test test, boolean wait) {
if (test instanceof DeviceTest) {
DeviceTest deviceTest = (DeviceTest)test;
deviceTest.setDevice(mDevice);
deviceTest.setTestAppPath(mTestDataPath);
} else {
Log.w(LOG_TAG, String.format("%s test class is not a DeviceTest.",
test.getClass().getName()));
}
return super.doRun(test, wait);
}
/**
* Override parent to create DeviceTestSuite wrapper, instead of TestSuite
*/
@SuppressWarnings("unchecked")
@Override
protected TestResult runSingleMethod(String testCase, String method, boolean wait)
throws Exception {
Class testClass = loadSuiteClass(testCase);
Test test = DeviceTestSuite.createTest(testClass, method);
return doRun(test, wait);
}
}
|
