tools/layoutlib/create/tests/com/android/tools/layoutlib/create/ClassHasNativeVisitorTest.java

   1 /*
   2  * Copyright (C) 2010 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 com.android.tools.layoutlib.create;
  18
  19 import static org.junit.Assert.*;
  20
  21 import org.junit.Test;
  22 import org.objectweb.asm.ClassReader;
  23
  24 import java.io.IOException;
  25 import java.util.ArrayList;
  26
  27
  28 /**
  29  * Tests {@link ClassHasNativeVisitor}.
  30  */
  31 public class ClassHasNativeVisitorTest {
  32
  33     @Test
  34     public void testHasNative() throws IOException {
  35         MockClassHasNativeVisitor cv = new MockClassHasNativeVisitor();
  36         String className =
  37                 this.getClass().getCanonicalName() + "$" + ClassWithNative.class.getSimpleName();
  38         ClassReader cr = new ClassReader(className);
  39
  40         cr.accept(cv, 0 /* flags */);
  41         assertArrayEquals(new String[] { "native_method" }, cv.getMethodsFound());
  42         assertTrue(cv.hasNativeMethods());
  43     }
  44
  45     @Test
  46     public void testHasNoNative() throws IOException {
  47         MockClassHasNativeVisitor cv = new MockClassHasNativeVisitor();
  48         String className =
  49             this.getClass().getCanonicalName() + "$" + ClassWithoutNative.class.getSimpleName();
  50         ClassReader cr = new ClassReader(className);
  51
  52         cr.accept(cv, 0 /* flags */);
  53         assertArrayEquals(new String[0], cv.getMethodsFound());
  54         assertFalse(cv.hasNativeMethods());
  55     }
  56
  57     //-------
  58
  59     /**
  60      * Overrides {@link ClassHasNativeVisitor} to collec the name of the native methods found.
  61      */
  62     private static class MockClassHasNativeVisitor extends ClassHasNativeVisitor {
  63         private ArrayList<String> mMethodsFound = new ArrayList<String>();
  64
  65         public String[] getMethodsFound() {
  66             return mMethodsFound.toArray(new String[mMethodsFound.size()]);
  67         }
  68
  69         @Override
  70         protected void setHasNativeMethods(boolean hasNativeMethods, String methodName) {
  71             if (hasNativeMethods) {
  72                 mMethodsFound.add(methodName);
  73             }
  74             super.setHasNativeMethods(hasNativeMethods, methodName);
  75         }
  76     }
  77
  78     /**
  79      * Dummy test class with a native method.
  80      */
  81     public static class ClassWithNative {
  82         public ClassWithNative() {
  83         }
  84
  85         public void callTheNativeMethod() {
  86             native_method();
  87         }
  88
  89         private native void native_method();
  90     }
  91
  92     /**
  93      * Dummy test class with no native method.
  94      */
  95     public static class ClassWithoutNative {
  96         public ClassWithoutNative() {
  97         }
  98
  99         public void someMethod() {
 100         }
 101     }
 102 }