コード例 #1
0
  @Test
  public void testGenJPAEntity() {
    JPAResponderGen rg = new JPAResponderGen();

    FieldInfo keyInfo = new FieldInfo("flightID", "Long", null);
    List<FieldInfo> properties = new ArrayList<FieldInfo>();
    properties.add(new FieldInfo("number", "String", null));
    properties.add(new FieldInfo("fitHostiesName", "String", null));
    properties.add(new FieldInfo("runway", "String", null));
    properties.add(new FieldInfo("passengers", "Integer", null));

    List<String> annotations = new ArrayList<String>();
    annotations.add("@Temporal(TemporalType.TIMESTAMP)");
    properties.add(new FieldInfo("departureDT", "java.util.Date", annotations));

    properties.add(new FieldInfo("dinnerServed", "java.sql.Timestamp", null));

    EntityInfo jpaEntity = new EntityInfo("Flight", "AirlineModel.model", keyInfo, properties);
    String generatedClass = rg.generateJPAEntityClass(jpaEntity);

    assertTrue(generatedClass.contains("package AirlineModel.model;"));
    assertTrue(generatedClass.contains("import javax.persistence.Entity;"));
    assertTrue(generatedClass.contains("@Entity"));
    assertTrue(generatedClass.contains("public class Flight {"));
    assertTrue(generatedClass.contains("@Id"));
    assertTrue(generatedClass.contains("@Basic(optional = false)"));
    assertTrue(generatedClass.contains("private Long flightID;"));
    assertTrue(generatedClass.contains("private String number;"));
    assertTrue(generatedClass.contains("private String fitHostiesName;"));
    assertTrue(generatedClass.contains("private String runway;"));
    assertTrue(generatedClass.contains("private Integer passengers;"));

    // date handling needs some special support
    assertTrue(generatedClass.contains("@Temporal(TemporalType.TIMESTAMP)"));
    assertTrue(generatedClass.contains("private java.util.Date departureDT;"));

    assertTrue(generatedClass.contains("private java.sql.Timestamp dinnerServed;"));
    assertTrue(generatedClass.contains("public Flight() {"));
  }
コード例 #2
0
  @Test
  public void testGenJPAEntitySimpleJoin() {
    JPAResponderGen rg = new JPAResponderGen();

    FieldInfo keyInfo = new FieldInfo("flightID", "Long", null);

    List<FieldInfo> properties = new ArrayList<FieldInfo>();
    properties.add(new FieldInfo("number", "String", null));
    List<String> annotations = new ArrayList<String>();
    annotations.add("@Temporal(TemporalType.TIMESTAMP)");
    properties.add(new FieldInfo("departureDT", "java.util.Date", annotations));

    List<JoinInfo> joins = new ArrayList<JoinInfo>();
    List<String> joinAnnotations = new ArrayList<String>();
    joinAnnotations.add("@ManyToMany(mappedBy=\"Flight\")");
    joins.add(new JoinInfo("passengers", "Person", joinAnnotations));

    EntityInfo jpaEntity =
        new EntityInfo("Flight", "AirlineModel.model", keyInfo, properties, joins, true);
    String generatedClass = rg.generateJPAEntityClass(jpaEntity);

    assertTrue(generatedClass.contains("package AirlineModel.model;"));
    assertTrue(generatedClass.contains("import javax.persistence.Entity;"));
    assertTrue(generatedClass.contains("@Entity"));
    assertTrue(generatedClass.contains("public class Flight {"));
    assertTrue(generatedClass.contains("@Id"));
    assertTrue(generatedClass.contains("@Basic(optional = false)"));
    assertTrue(generatedClass.contains("private Long flightID;"));
    assertTrue(generatedClass.contains("private String number;"));

    assertTrue(generatedClass.contains("@Temporal(TemporalType.TIMESTAMP)"));
    assertTrue(generatedClass.contains("private java.util.Date departureDT;"));

    assertTrue(generatedClass.contains("@ManyToMany(mappedBy=\"Flight\")"));
    assertTrue(generatedClass.contains("private Collection<Person> passengers;"));

    assertTrue(generatedClass.contains("public Flight() {"));
  }
コード例 #3
0
 @Test(expected = AssertionError.class)
 public void testGenJPAEntityNull() {
   JPAResponderGen rg = new JPAResponderGen();
   assertNull(rg.generateJPAEntityClass(null));
 }