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
17package android.test.suitebuilder.annotation;
18
19import android.test.suitebuilder.TestMethod;
20import junit.framework.TestCase;
21
22import java.lang.annotation.Annotation;
23import java.lang.reflect.Method;
24
25
26public class HasMethodAnnotationTest extends TestCase {
27
28    public void testMethodWithSpecifiedAttribute() throws Exception {
29        assertTrue(methodHasAnnotation(AnnotatedMethodExample.class,
30                "testThatIsAnnotated", Smoke.class));
31    }
32
33    public void testMethodWithoutSpecifiedAttribute() throws Exception {
34        assertFalse(methodHasAnnotation(AnnotatedMethodExample.class,
35                "testThatIsNotAnnotated", Smoke.class));
36    }
37
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
50        public void testThatIsAnnotated() {
51        }
52
53        public void testThatIsNotAnnotated() {
54        }
55    }
56}
57