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