19066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project/* 29066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * Copyright (C) 2008 The Android Open Source Project 39066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * 49066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * Licensed under the Apache License, Version 2.0 (the "License"); 59066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * you may not use this file except in compliance with the License. 69066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * You may obtain a copy of the License at 79066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * 89066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * http://www.apache.org/licenses/LICENSE-2.0 99066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * 109066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * Unless required by applicable law or agreed to in writing, software 119066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * distributed under the License is distributed on an "AS IS" BASIS, 129066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 139066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * See the License for the specific language governing permissions and 149066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * limitations under the License. 159066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project */ 169066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project 179066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Projectpackage android.test.suitebuilder; 189066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project 199066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Projectimport android.test.suitebuilder.examples.instrumentation.InstrumentationTest; 209066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project 219066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Projectimport junit.framework.Assert; 229066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Projectimport junit.framework.AssertionFailedError; 239066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Projectimport junit.framework.Test; 249066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Projectimport junit.framework.TestCase; 259066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Projectimport junit.framework.TestListener; 269066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Projectimport junit.framework.TestResult; 279066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Projectimport junit.framework.TestSuite; 289066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project 299066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Projectimport java.util.HashSet; 309066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Projectimport java.util.Set; 319066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project 329066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Projectpublic class UnitTestSuiteBuilderTest extends TestCase { 339066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project 349066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project private UnitTestSuiteBuilder unitTestSuiteBuilder; 359066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project 369066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project protected void setUp() throws Exception { 379066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project super.setUp(); 389066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project unitTestSuiteBuilder = new UnitTestSuiteBuilder(getClass()); 399066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project } 409066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project 419066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project public void testShouldExcludeIntrumentationTests() throws Exception { 429066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project unitTestSuiteBuilder.includePackages(packageFor(InstrumentationTest.class)); 439066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project 449066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project TestSuite testSuite = unitTestSuiteBuilder.build(); 459066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project Assert.assertEquals(0, ListTestCaseNames.getTestCaseNames(testSuite).size()); 469066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project 479066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project SuiteExecutionRecorder recorder = runSuite(unitTestSuiteBuilder); 489066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project 499066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project assertFalse(recorder.saw("InstrumentationTest.testInstrumentation")); 509066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project assertTrue(recorder.testsSeen.isEmpty()); 519066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project } 529066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project 539066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project private static String packageFor(Class clazz) { 549066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project String className = clazz.getName(); 559066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project return className.substring(0, className.lastIndexOf('.')); 569066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project } 579066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project 589066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project private SuiteExecutionRecorder runSuite(TestSuiteBuilder builder) { 599066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project TestSuite suite = builder.build(); 609066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project SuiteExecutionRecorder recorder = new SuiteExecutionRecorder(); 619066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project TestResult result = new TestResult(); 629066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project result.addListener(recorder); 639066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project suite.run(result); 649066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project return recorder; 659066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project } 669066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project 679066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project private class SuiteExecutionRecorder implements TestListener { 689066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project 699066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project private Set<String> failures = new HashSet<String>(); 709066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project private Set<String> errors = new HashSet<String>(); 719066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project private Set<String> testsSeen = new HashSet<String>(); 729066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project 739066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project public void addError(Test test, Throwable t) { 749066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project errors.add(testName(test)); 759066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project } 769066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project 779066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project public void addFailure(Test test, AssertionFailedError t) { 789066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project failures.add(testName(test)); 799066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project } 809066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project 819066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project public void endTest(Test test) { 829066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project } 839066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project 849066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project public void startTest(Test test) { 859066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project testsSeen.add(testName(test)); 869066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project } 879066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project 889066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project public boolean saw(String testName) { 899066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project return testsSeen.contains(testName); 909066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project } 919066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project 929066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project public boolean failed(String testName) { 939066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project return failures.contains(testName); 949066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project } 959066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project 969066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project public boolean errored(String testName) { 979066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project return errors.contains(testName); 989066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project } 999066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project 1009066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project public boolean passed(String testName) { 1019066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project return saw(testName) && !failed(testName) && !errored(testName); 1029066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project } 1039066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project 1049066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project private String testName(Test test) { 1059066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project TestCase testCase = (TestCase) test; 1069066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project return testCase.getClass().getSimpleName() + "." + testCase.getName(); 1079066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project } 1089066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project } 1099066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project} 110