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.annotation.Annotation;
23 import java.lang.reflect.Method;
24 
25 
26 public class HasMethodAnnotationTest extends TestCase {
27 
testMethodWithSpecifiedAttribute()28     public void testMethodWithSpecifiedAttribute() throws Exception {
29         assertTrue(methodHasAnnotation(AnnotatedMethodExample.class,
30                 "testThatIsAnnotated", Smoke.class));
31     }
32 
testMethodWithoutSpecifiedAttribute()33     public void testMethodWithoutSpecifiedAttribute() throws Exception {
34         assertFalse(methodHasAnnotation(AnnotatedMethodExample.class,
35                 "testThatIsNotAnnotated", Smoke.class));
36     }
37 
methodHasAnnotation(Class<? extends TestCase> aClass, String methodName, Class<? extends Annotation> expectedClassification )38     private boolean methodHasAnnotation(Class<? extends TestCase> aClass,
39             String methodName,
40             Class<? extends Annotation> expectedClassification
41     ) throws NoSuchMethodException {
42         Method method = aClass.getMethod(methodName);
43         TestMethod testMethod = new TestMethod(method, aClass);
44         return new HasMethodAnnotation(expectedClassification).apply(testMethod);
45     }
46 
47     static class AnnotatedMethodExample extends TestCase {
48 
49         @Smoke
testThatIsAnnotated()50         public void testThatIsAnnotated() {
51         }
52 
testThatIsNotAnnotated()53         public void testThatIsNotAnnotated() {
54         }
55     }
56 }
57