Exemplo n.º 1
0
  @Test
  public void testSimple() throws Exception {
    String testValue = "TEST";
    String methodName = "getName";
    String className = "org.teatrove.trove.test.Simple";

    ClassFile cf = new ClassFile(className);

    MethodInfo ctor = cf.addDefaultConstructor();
    CodeBuilder builder = new CodeBuilder(ctor);
    builder.loadThis();
    builder.invokeSuperConstructor();
    builder.returnVoid();

    Modifiers mods = new Modifiers(Modifier.PUBLIC);
    MethodInfo getName = cf.addMethod(mods, methodName, TypeDesc.STRING);
    builder = new CodeBuilder(getName);
    builder.loadConstant(testValue);
    builder.returnValue(TypeDesc.STRING);

    ClassInjector injector = ClassInjector.getInstance();
    OutputStream os = injector.getStream(className);
    cf.writeTo(os);
    os.close();

    Class<?> clazz = injector.loadClass(className);
    assertEquals("expected class name", className, clazz.getName());

    Object instance = clazz.newInstance();
    Method method = clazz.getMethod(methodName);
    String result = (String) method.invoke(instance);
    assertEquals("expected test value", testValue, result);
  }
Exemplo n.º 2
0
  @Test
  public void testAnnotations() throws Exception {
    String testValue = "TEST";
    String methodName = "getTest";
    String className = "org.teatrove.trove.test.Annotated";

    ClassFile cf = new ClassFile(className);
    Annotation ann = cf.addRuntimeVisibleAnnotation(TypeDesc.forClass(TestAnnotation.class));
    ann.putMemberValue("value", testValue);

    MethodInfo ctor = cf.addDefaultConstructor();
    CodeBuilder builder = new CodeBuilder(ctor);
    builder.loadThis();
    builder.invokeSuperConstructor();
    builder.returnVoid();

    Modifiers mods = new Modifiers(Modifier.PUBLIC);
    MethodInfo getName = cf.addMethod(mods, methodName, TypeDesc.STRING);
    builder = new CodeBuilder(getName);
    builder.loadConstant(null);
    builder.returnValue(TypeDesc.STRING);

    getName.addRuntimeVisibleAnnotation(TypeDesc.forClass(Deprecated.class));

    ClassInjector injector = ClassInjector.getInstance();
    OutputStream os = injector.getStream(className);
    cf.writeTo(os);
    os.close();

    Class<?> clazz = injector.loadClass(className);
    assertEquals("expected class name", className, clazz.getName());

    TestAnnotation annotation = clazz.getAnnotation(TestAnnotation.class);
    assertNotNull("expected annotation", annotation);
    assertEquals("expected test value", testValue, annotation.value());

    Method method = clazz.getMethod(methodName);
    assertNotNull("expected deprecated", method.getAnnotation(Deprecated.class));
  }