1 /*
2  * Copyright (C) 2008 The Android Open Source Project
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 
17 package android.test.suitebuilder.annotation;
18 
19 import android.test.suitebuilder.TestMethod;
20 import junit.framework.TestCase;
21 
22 import java.lang.reflect.Method;
23 
24 public class HasClassAnnotationTest extends TestCase {
25 
testShouldTellIfParentClassHasSpecifiedClassification()26     public void testShouldTellIfParentClassHasSpecifiedClassification()
27             throws NoSuchMethodException {
28         assertTrue(classHasAnnotation(SmokeTestExample.class, Smoke.class));
29     }
30 
testShouldTellIfParentClassDoesNotHaveSpecifiedClassification()31     public void testShouldTellIfParentClassDoesNotHaveSpecifiedClassification()
32             throws NoSuchMethodException {
33         assertFalse(classHasAnnotation(NonSmokeTestExample.class, Smoke.class));
34     }
35 
classHasAnnotation( Class<? extends TestCase> aClass, Class<Smoke> expectedClassification)36     private boolean classHasAnnotation(
37             Class<? extends TestCase> aClass,
38             Class<Smoke> expectedClassification) throws NoSuchMethodException {
39         Method method = aClass.getMethod("testSomeTest");
40 
41         TestMethod testMethod = new TestMethod(method, aClass);
42         return new HasClassAnnotation(expectedClassification).apply(testMethod);
43     }
44 
45     @Smoke
46     static class SmokeTestExample extends TestCase {
47 
testSomeTest()48         public void testSomeTest() {
49         }
50     }
51 
52     static class NonSmokeTestExample extends TestCase {
53 
testSomeTest()54         public void testSomeTest() {
55         }
56     }
57 }
58