protected void genRxImports(ClassModel model, PrintWriter writer) { for (ClassTypeInfo importedType : model.getImportedTypes()) { if (importedType.getKind() == ClassKind.API) { } else { if (!importedType.getPackageName().equals("java.lang")) { addImport(importedType, false, writer); } } } }
protected void genRxImports(ClassModel model, PrintWriter writer) { for (ClassTypeInfo importedType : model.getImportedTypes()) { if (importedType.getKind() == ClassKind.API) { } else { if (!importedType.getPackageName().equals("java.lang")) { addImport(importedType, false, writer); } } } }
@Override public Map<String, Object> getVars() { Map<String, Object> vars = Model.super.getVars(); vars.put("importedTypes", getImportedTypes()); vars.put("concrete", isConcrete()); vars.put("type", getType());
@Test public void testValidExceptionParam() throws Exception { ClassModel model = new GeneratorHelper().generateClass(MethodWithValidThrowableParam.class); assertEquals(0, model.getReferencedTypes().size()); assertEquals(1, model.getImportedTypes().size()); assertEquals(ClassKind.THROWABLE, model.getImportedTypes().iterator().next().getKind()); assertEquals(1, model.getMethods().size()); String methodName = "methodWithThrowableParam"; MethodInfo method = model.getMethods().get(0); checkMethod(method, methodName, 1, "void", MethodKind.OTHER); List<ParamInfo> params = method.getParams(); checkParam(params.get(0), "t", Throwable.class); }
@Test public void testValidEnumReturn() throws Exception { ClassModel model = new GeneratorHelper().generateClass(MethodWithEnumReturn.class); assertEquals(MethodWithEnumReturn.class.getName(), model.getIfaceFQCN()); assertEquals(MethodWithEnumReturn.class.getSimpleName(), model.getIfaceSimpleName()); assertTrue(model.getImportedTypes().contains(TypeReflectionFactory.create(TestEnum.class))); assertTrue(model.getReferencedTypes().isEmpty()); assertTrue(model.getSuperTypes().isEmpty()); assertEquals(2, model.getMethods().size()); checkMethod(model.getMethods().get(0), "methodWithEnumReturn", 0, TestEnum.class, MethodKind.OTHER); checkMethod(model.getMethods().get(1), "methodWithGenEnumReturn", 0, TestGenEnum.class, MethodKind.OTHER); }
@Test public void testValidEnumParam() throws Exception { ClassModel model = new GeneratorHelper().generateClass(MethodWithEnumParam.class); assertEquals(MethodWithEnumParam.class.getName(), model.getIfaceFQCN()); assertEquals(MethodWithEnumParam.class.getSimpleName(), model.getIfaceSimpleName()); assertTrue(model.getImportedTypes().contains(TypeReflectionFactory.create(TestEnum.class))); assertTrue(model.getReferencedTypes().isEmpty()); assertTrue(model.getSuperTypes().isEmpty()); assertEquals(2, model.getMethods().size()); MethodInfo method = model.getMethods().get(0); checkMethod(method, "methodWithEnumParam", 1, "void", MethodKind.OTHER); List<ParamInfo> params = method.getParams(); checkParam(params.get(0), "weirdo", TestEnum.class); EnumTypeInfo enumType = (EnumTypeInfo) params.get(0).getType(); assertFalse(enumType.isGen()); assertEquals(Arrays.asList("TIM", "JULIEN", "NICK", "WESTON"), enumType.getValues()); method = model.getMethods().get(1); checkMethod(method, "methodWithGenEnumParam", 1, "void", MethodKind.OTHER); params = method.getParams(); checkParam(params.get(0), "weirdo", TestGenEnum.class); enumType = (EnumTypeInfo) params.get(0).getType(); assertTrue(enumType.isGen()); assertEquals(Arrays.asList("LAURA", "BOB", "MIKE", "LELAND"), enumType.getValues()); }