@Test public void testWithGenericsExtends() { String expectedClass = "package com.optible.processor.model;\n\n" + "import org.junit.Test;\n\n" + "public class ClassModelTest extends BaseTest<View.view, Presenter.presenter> {\n" + "}\n"; TopLevelModel model = new TopLevelModel("com.optible.processor.model", "ClassModelTest"); model.addExtends("BaseTest<View.view, Presenter.presenter>"); model.addModifiers(Modifier.PUBLIC); model.addImports("org.junit.Test"); String resultClass = model.toString(); assertEquals(expectedClass, resultClass); }
@Test public void testWithExtendsWithPath() { String expectedClass = "package com.optible.processor.model;\n\n" + "import test.BaseTest;\nimport org.junit.Test;\n\n" + "public class ClassModelTest extends BaseTest {\n" + "}\n"; TopLevelModel model = new TopLevelModel("com.optible.processor.model", "ClassModelTest"); model.addExtends(new TypeModel("test", "BaseTest")); model.addModifiers(Modifier.PUBLIC); model.addImports("org.junit.Test"); String resultClass = model.toString(); assertEquals(expectedClass, resultClass); }