use of org.abs_models.frontend.ast.ClassDecl in project abstools by abstools.
the class TraitTest method addAndModifyExistingMethod.
@Test
public void addAndModifyExistingMethod() {
Model model = assertParse("module M; " + " trait T = {Unit myMethod(){ println(\"\"); }} modifies { Unit myMethod(){ skip; } }" + " class C {uses T; }");
ClassDecl cls = (ClassDecl) findDecl(model, "M", "C");
assertNotNull(cls);
assertTrue(cls.getMethods().getNumChild() == 0);
model.applyTraits();
assertTrue(cls.getMethods().getNumChild() == 1);
assertTrue(cls.getMethod(0).getBlock().getStmt(0) instanceof SkipStmt);
}
use of org.abs_models.frontend.ast.ClassDecl in project abstools by abstools.
the class TraitTest method addAddModifierAtRuntimeBackComp.
@Test
public void addAddModifierAtRuntimeBackComp() {
Model model = assertParse("module M;" + "class C { Unit m(){skip;} }");
ClassDecl cls = (ClassDecl) findDecl(model, "M", "C");
MethodSig sig = AbsASTBuilderUtil.createMethodSig("n", AbsASTBuilderUtil.getUnit());
MethodImpl impl = new MethodImpl(sig, new Block(new List<>(), new List<>(new SkipStmt())));
AddMethodModifier opr = new AddMethodModifier(impl);
assertNotNull(opr.getMethodImpl());
ModifyClassModifier mcn = new ModifyClassModifier();
mcn.setName("M.C");
DeltaAccess acc = new DeltaAccess(cls.getModuleDecl().getName());
DeltaDecl dd = new DeltaDecl();
dd.setName("MyDelta");
dd.setImportedModule(acc);
dd.addModuleModifier(mcn);
mcn.addModifier(opr);
mcn.setParent(dd);
acc.setParent(dd);
opr.setParent(mcn);
sig.setParent(opr);
CompilationUnit cu = model.getCompilationUnitList().getChild(0);
cu.addDeltaDecl(dd);
dd.setParent(cu);
model.applyDelta(dd);
assertEquals(2, cls.getMethods().getNumChild());
}
use of org.abs_models.frontend.ast.ClassDecl in project abstools by abstools.
the class TraitTest method multipleNameResolveSteps.
@Test
public void multipleNameResolveSteps() {
Model model = assertParse("\n" + "module M;" + "export T;" + "trait T = { Unit myMethod(){ skip; } }" + "\n" + "module N;" + "export U;" + "import T from M;" + "trait U = T modifies {}" + "class C { uses U; }" + "\n" + "module O;" + "import U from N;" + "class D { uses U; }");
ClassDecl cls = (ClassDecl) findDecl(model, "N", "C");
ClassDecl cls2 = (ClassDecl) findDecl(model, "O", "D");
assertNotNull(cls);
assertTrue(cls.getMethods().getNumChild() == 0);
assertNotNull(cls2);
assertTrue(cls2.getMethods().getNumChild() == 0);
model.applyTraits();
assertTrue(cls.getMethods().getNumChild() == 1);
assertTrue(cls.getMethod(0).getMethodSig().getName().equals("myMethod"));
assertTrue(cls2.getMethods().getNumChild() == 1);
assertTrue(cls2.getMethod(0).getMethodSig().getName().equals("myMethod"));
}
use of org.abs_models.frontend.ast.ClassDecl in project abstools by abstools.
the class TraitTest method collapseTest2.
@Test
public void collapseTest2() {
Model model = assertParse("module TestMod;" + "interface Inter {}" + "trait T = { " + " Unit printLine_1(){println(\"I'm 1!\");}" + " Unit printLine_2(){println(\"I'm 2!\");}" + "}" + "trait T2 = { " + " Unit printLine_2(){println(\"I'm 2!\");}" + "}" + "trait T3 = { " + " Unit printLine_5(){println(\"I'm 5!\");}" + " Unit printLine_6(){println(\"I'm 6!\");}" + "}" + "trait T4 = { " + " Unit printLine_7(){println(\"I'm 7!\");}" + " Unit printLine_8(){println(\"I'm 8!\");}" + "}" + "class InterImpl(Inter inter) implements Inter { }" + "" + "delta D3;" + "modifies class TestMod.InterImpl{" + " adds T modifies T2 removes Unit printLine_2();" + " adds Int i = 0;" + " modifies T3 adds T4 " + "}");
ClassDecl cls = (ClassDecl) findDecl(model, "TestMod", "InterImpl");
assertNotNull(cls);
assertTrue(cls.getMethods().getNumChild() == 0);
model.applyTraits();
model.collapseTraitModifiers();
DeltaDecl delta = findDelta(model, "D3");
assertNotNull(delta);
assertThat(delta, instanceOf(DeltaDecl.class));
ModifyClassModifier mm = (ModifyClassModifier) delta.getModuleModifier(0);
assertTrue(mm.getModifierList().getNumChild() == 6);
DeltaTraitModifier dml = (DeltaTraitModifier) mm.getModifier(0);
AddMethodModifier mcl = (AddMethodModifier) dml.getMethodModifier();
TraitExpr set = mcl.getTraitExpr();
assertTrue(set.getChild(0).getNumChild() == 2);
assertThat(set, instanceOf(TraitSetExpr.class));
DeltaTraitModifier dml2 = (DeltaTraitModifier) mm.getModifier(1);
ModifyMethodModifier mcl2 = (ModifyMethodModifier) dml2.getMethodModifier();
TraitExpr set2 = mcl2.getTraitExpr();
assertTrue(set2.getChild(0).getNumChild() == 1);
assertThat(set2, instanceOf(TraitSetExpr.class));
}
use of org.abs_models.frontend.ast.ClassDecl in project abstools by abstools.
the class TraitTest method removeExistingMethodInClassSucc.
@Test
public void removeExistingMethodInClassSucc() {
Model model = assertParse("module M;" + "trait T = {Unit myMethod(){skip;}} " + "class C {uses T removes Unit myMethod();; }");
ClassDecl cls = (ClassDecl) findDecl(model, "M", "C");
assertNotNull(cls);
assertTrue(cls.getMethods().getNumChild() == 0);
model.applyTraits();
assertTrue(cls.getMethods().getNumChild() == 0);
}
Aggregations