@Test
public void shouldPrintSuperclassAndInterfacesForClass() throws IOException {
// Given:
unit(
"my",
classDecl(
flags(PUBLIC, FINAL),
annotations(annotation(Generated.class)),
"MyException",
type(Exception.class),
types(type(Serializable.class), type(Cloneable.class)),
statements()));
// When:
printer.print(unit);
final String contents = mosp.getContentMap().get("my/MyException.java");
// Then:
assertEquals(
"package my;\n\n\n"
+ "@javax.annotation.Generated\n"
+ "public final class MyException extends Exception implements java.io.Serializable, Cloneable {\n"
+ "}",
contents);
}
@Test
public void shouldPrintClassDecl() throws IOException {
// Given:
unit("my.pkg", classDecl(flags(), annotations(), "FooClass"));
// When:
printer.print(unit);
// Then:
assertSameGeneratedContent("package my.pkg;\n\n\nclass FooClass {\n}", "my/pkg/FooClass.java");
}
@Test
public void shouldPrintConditionals() throws IOException {
// Given:
unit(
"my",
classDecl(
flags(PUBLIC, ABSTRACT),
annotations(),
"User",
null,
types(),
statements(
// >> public int foo(int a) { if (hashCode() > a) { return 1; } else { return 2; } }
method(
flags(PUBLIC),
annotations(),
"foo",
int.class,
args(var("a", int.class)),
block(
ifs(
op(
Operator.GT,
call(ident("this", "hashCode"), expressions()),
ident("a")),
block(returns(literal(1))),
block(returns(literal(2)))))))));
// When:
printer.print(unit);
final String contents = mosp.getContentMap().get("my/User.java");
// Then:
assertEquals(
"package my;\n\n\n"
+ "public abstract class User {\n\n\n"
+ " public int foo(int a) {\n"
+ " if (this.hashCode() > a) {\n"
+ " return 1;\n"
+ " } else {\n"
+ " return 2;\n"
+ " }\n"
+ " }\n"
+ "}",
contents);
}
@Test
public void shouldPrintFullNestedClassName() throws IOException {
// Given:
unit(
"my.pkg",
classDecl(
flags(), annotations(), "FooClass", null, types(type(Sample.class)), statements()));
// When:
printer.print(unit);
// Then:
assertSameGeneratedContent(
"package my.pkg;\n\n\nclass FooClass implements "
+ JstPrinterTest.class.getName()
+ ".Sample"
+ " {\n"
+ "}",
"my/pkg/FooClass.java");
}
@Test
public void shouldPrintAnnotationDecl() throws IOException {
// Given:
unit(
"my.pkg",
classDecl(
flags(PUBLIC, ANNOTATION),
annotations(annotation(Retention.class, literal(RetentionPolicy.RUNTIME))),
"Foo"));
// When:
printer.print(unit);
// Then:
assertSameGeneratedContent(
"package my.pkg;\n\n\n"
+ "@java.lang.annotation.Retention(java.lang.annotation.RetentionPolicy.RUNTIME)\n"
+ "public @interface Foo {\n}",
"my/pkg/Foo.java");
}
@Test
public void shouldPrintExtendInterfacesForInterface() throws IOException {
// Given:
unit(
"my",
classDecl(
flags(PUBLIC, INTERFACE),
annotations(),
"IDomainObject",
null,
types(type(Serializable.class), type(Cloneable.class)),
statements()));
// When:
printer.print(unit);
final String contents = mosp.getContentMap().get("my/IDomainObject.java");
// Then:
assertEquals(
"package my;\n\n\n"
+ "public interface IDomainObject extends java.io.Serializable, Cloneable {\n"
+ "}",
contents);
}
@Test
public void shouldPrintClassWithProperties() throws IOException {
// Given:
unit(
"my.pkg",
classDecl(
flags(STRICTFP),
annotations(),
"FooClass",
null,
types(),
statements(
var(flags(PRIVATE, FINAL), "a", String.class),
var(flags(PRIVATE, FINAL), "b", int.class))));
// When:
printer.print(unit);
final String contents = mosp.getContentMap().get("my/pkg/FooClass.java");
// Then:
assertTrue(contents.contains("strictfp class FooClass {\n"));
assertTrue(contents.contains("private final String a;\n"));
assertTrue(contents.contains("private final int b;\n"));
}
@Test
public void shouldPrintClassWithOverriddenMethods() throws IOException {
// Given:
unit(
"domain",
classDecl(
flags(),
annotations(),
"User",
null,
types(),
statements(
// >> public static final String DEFAULT_NAME = "default-name";
var(
flags(PUBLIC, STATIC, FINAL),
"DEFAULT_NAME",
String.class,
literal("default-name")),
// >> private String name = DEFAULT_NAME;
var(flags(PRIVATE), "name", String.class, ident("DEFAULT_NAME")),
// >> public final String getName() { return this.name; }
method(
flags(PUBLIC, FINAL),
"getName",
String.class,
args(),
block(returns(ident("this", "name")))),
// >> @Override public String toString() { return this.name; }
method(
flags(PUBLIC),
annotations(override()),
"toString",
String.class,
args(),
block(
// String result
var(flags(FINAL), "result", String.class, ident("this", "name")),
returns(ident("result")))),
// >> @Override public int hashCode() { return 1; }
method(
flags(PUBLIC),
annotations(override()),
"hashCode",
int.class,
args(),
block(returns(literal(1)))),
// >> @Override public boolean equals(Object other) { return false; }
method(
flags(PUBLIC),
annotations(override()),
"equals",
boolean.class,
args(var("other", Object.class)),
block(returns(literal(false)))))));
// When:
printer.print(unit);
final String contents = mosp.getContentMap().get("domain/User.java");
// Then:
assertTrue(contents.contains("package domain;\n"));
assertTrue(contents.contains("\nclass User {\n"));
assertTrue(contents.contains("private String name = DEFAULT_NAME;\n"));
assertTrue(
contents.contains(
"@Override\n public String toString() {\n"
+ " final String result = this.name;\n"
+ " return result;\n"
+ " }\n"));
}
