Examples with DbAttribute org.apache.cayenne.map.DbAttribute used on opensource projects

Example 31 with DbAttribute

use of org.apache.cayenne.map.DbAttribute in project cayenne by apache.

the class CreateTableToModelIT method testAddTable.

@Test
public void testAddTable() throws Exception {
    dropTableIfPresent("NEW_TABLE");
    assertTokensAndExecute(0, 0);
    DbEntity dbEntity = new DbEntity("NEW_TABLE");
    DbAttribute column1 = new DbAttribute("ID", Types.INTEGER, dbEntity);
    column1.setMandatory(true);
    column1.setPrimaryKey(true);
    dbEntity.addAttribute(column1);
    DbAttribute column2 = new DbAttribute("NAME", Types.VARCHAR, dbEntity);
    column2.setMaxLength(10);
    column2.setMandatory(false);
    dbEntity.addAttribute(column2);
    // for the new entity to the db
    execute(mergerFactory().createCreateTableToDb(dbEntity));
    List<MergerToken> tokens = createMergeTokens();
    assertEquals(1, tokens.size());
    MergerToken token = tokens.get(0);
    if (token.getDirection().isToDb()) {
        token = token.createReverse(mergerFactory());
    }
    assertTrue(token.getClass().getName(), token instanceof CreateTableToModel);
    execute(token);
    ObjEntity objEntity = null;
    for (ObjEntity candidate : map.getObjEntities()) {
        if (dbEntity.getName().equalsIgnoreCase(candidate.getDbEntityName())) {
            objEntity = candidate;
            break;
        }
    }
    assertNotNull(objEntity);
    assertEquals(objEntity.getClassName(), map.getDefaultPackage() + "." + objEntity.getName());
    assertEquals(objEntity.getSuperClassName(), map.getDefaultSuperclass());
    assertEquals(objEntity.getClientClassName(), map.getDefaultClientPackage() + "." + objEntity.getName());
    assertEquals(objEntity.getClientSuperClassName(), map.getDefaultClientSuperclass());
    assertEquals(1, objEntity.getAttributes().size());
    assertEquals("java.lang.String", objEntity.getAttributes().iterator().next().getType());
    // clear up
    // fix psql case issue
    map.removeDbEntity(objEntity.getDbEntity().getName(), true);
    map.removeObjEntity(objEntity.getName(), true);
    map.removeDbEntity(dbEntity.getName(), true);
    resolver.refreshMappingCache();
    assertNull(map.getObjEntity(objEntity.getName()));
    assertNull(map.getDbEntity(dbEntity.getName()));
    assertFalse(map.getDbEntities().contains(dbEntity));
    assertTokensAndExecute(1, 0);
    assertTokensAndExecute(0, 0);
}

Example 32 with DbAttribute

use of org.apache.cayenne.map.DbAttribute in project cayenne by apache.

the class DropColumnToModelIT method testRemoveFKColumnWithoutRelationshipInDb.

@Test
public void testRemoveFKColumnWithoutRelationshipInDb() throws Exception {
    dropTableIfPresent("NEW_TABLE");
    dropTableIfPresent("NEW_TABLE2");
    assertTokensAndExecute(0, 0);
    DbEntity dbEntity1 = new DbEntity("NEW_TABLE");
    DbAttribute e1col1 = new DbAttribute("ID", Types.INTEGER, dbEntity1);
    e1col1.setMandatory(true);
    e1col1.setPrimaryKey(true);
    dbEntity1.addAttribute(e1col1);
    DbAttribute e1col2 = new DbAttribute("NAME", Types.VARCHAR, dbEntity1);
    e1col2.setMaxLength(10);
    e1col2.setMandatory(false);
    dbEntity1.addAttribute(e1col2);
    map.addDbEntity(dbEntity1);
    DbEntity dbEntity2 = new DbEntity("NEW_TABLE2");
    DbAttribute e2col1 = new DbAttribute("ID", Types.INTEGER, dbEntity2);
    e2col1.setMandatory(true);
    e2col1.setPrimaryKey(true);
    dbEntity2.addAttribute(e2col1);
    DbAttribute e2col2 = new DbAttribute("FK", Types.INTEGER, dbEntity2);
    dbEntity2.addAttribute(e2col2);
    DbAttribute e2col3 = new DbAttribute("NAME", Types.VARCHAR, dbEntity2);
    e2col3.setMaxLength(10);
    dbEntity2.addAttribute(e2col3);
    map.addDbEntity(dbEntity2);
    assertTokensAndExecute(2, 0);
    assertTokensAndExecute(0, 0);
    // force drop fk column in db
    execute(mergerFactory().createDropColumnToDb(dbEntity2, e2col2));
    // create db relationships, but do not sync them to db
    DbRelationship rel1To2 = new DbRelationship("rel1To2");
    rel1To2.setSourceEntity(dbEntity1);
    rel1To2.setTargetEntityName(dbEntity2);
    rel1To2.setToMany(true);
    rel1To2.addJoin(new DbJoin(rel1To2, e1col1.getName(), e2col2.getName()));
    dbEntity1.addRelationship(rel1To2);
    DbRelationship rel2To1 = new DbRelationship("rel2To1");
    rel2To1.setSourceEntity(dbEntity2);
    rel2To1.setTargetEntityName(dbEntity1);
    rel2To1.setToMany(false);
    rel2To1.addJoin(new DbJoin(rel2To1, e2col2.getName(), e1col1.getName()));
    dbEntity2.addRelationship(rel2To1);
    assertSame(rel1To2, rel2To1.getReverseRelationship());
    assertSame(rel2To1, rel1To2.getReverseRelationship());
    // create ObjEntities
    ObjEntity objEntity1 = new ObjEntity("NewTable");
    objEntity1.setDbEntity(dbEntity1);
    ObjAttribute oatr1 = new ObjAttribute("name");
    oatr1.setDbAttributePath(e1col2.getName());
    oatr1.setType("java.lang.String");
    objEntity1.addAttribute(oatr1);
    map.addObjEntity(objEntity1);
    ObjEntity objEntity2 = new ObjEntity("NewTable2");
    objEntity2.setDbEntity(dbEntity2);
    ObjAttribute o2a1 = new ObjAttribute("name");
    o2a1.setDbAttributePath(e2col3.getName());
    o2a1.setType("java.lang.String");
    objEntity2.addAttribute(o2a1);
    map.addObjEntity(objEntity2);
    // create ObjRelationships
    assertEquals(0, objEntity1.getRelationships().size());
    assertEquals(0, objEntity2.getRelationships().size());
    ObjRelationship objRel1To2 = new ObjRelationship("objRel1To2");
    objRel1To2.addDbRelationship(rel1To2);
    objRel1To2.setSourceEntity(objEntity1);
    objRel1To2.setTargetEntityName(objEntity2);
    objEntity1.addRelationship(objRel1To2);
    ObjRelationship objRel2To1 = new ObjRelationship("objRel2To1");
    objRel2To1.addDbRelationship(rel2To1);
    objRel2To1.setSourceEntity(objEntity2);
    objRel2To1.setTargetEntityName(objEntity1);
    objEntity2.addRelationship(objRel2To1);
    assertEquals(1, objEntity1.getRelationships().size());
    assertEquals(1, objEntity2.getRelationships().size());
    assertSame(objRel1To2, objRel2To1.getReverseRelationship());
    assertSame(objRel2To1, objRel1To2.getReverseRelationship());
    // try do use the merger to remove the column and relationship in the
    // model
    List<MergerToken> tokens = createMergeTokens();
    assertTokens(tokens, 2, 0);
    // TODO: reversing the following two tokens should also reverse the
    // order
    MergerToken token0 = tokens.get(0).createReverse(mergerFactory());
    MergerToken token1 = tokens.get(1).createReverse(mergerFactory());
    if (!(token0 instanceof DropRelationshipToModel && token1 instanceof DropColumnToModel || token1 instanceof DropRelationshipToModel && token0 instanceof DropColumnToModel)) {
        fail();
    }
    // do not execute DropRelationshipToModel, only DropColumnToModel.
    if (token1 instanceof DropColumnToModel) {
        execute(token1);
    } else {
        execute(token0);
    }
    // check after merging
    assertNull(dbEntity2.getAttribute(e2col2.getName()));
    assertEquals(0, dbEntity1.getRelationships().size());
    assertEquals(0, dbEntity2.getRelationships().size());
    assertEquals(0, objEntity1.getRelationships().size());
    assertEquals(0, objEntity2.getRelationships().size());
    // clear up
    dbEntity1.removeRelationship(rel1To2.getName());
    dbEntity2.removeRelationship(rel2To1.getName());
    map.removeObjEntity(objEntity1.getName(), true);
    map.removeDbEntity(dbEntity1.getName(), true);
    map.removeObjEntity(objEntity2.getName(), true);
    map.removeDbEntity(dbEntity2.getName(), true);
    resolver.refreshMappingCache();
    assertNull(map.getObjEntity(objEntity1.getName()));
    assertNull(map.getDbEntity(dbEntity1.getName()));
    assertNull(map.getObjEntity(objEntity2.getName()));
    assertNull(map.getDbEntity(dbEntity2.getName()));
    assertFalse(map.getDbEntities().contains(dbEntity1));
    assertFalse(map.getDbEntities().contains(dbEntity2));
    assertTokensAndExecute(2, 0);
    assertTokensAndExecute(0, 0);
}

Example 33 with DbAttribute

use of org.apache.cayenne.map.DbAttribute in project cayenne by apache.

the class DropColumnToModelIT method testSimpleColumn.

@Test
public void testSimpleColumn() throws Exception {
    dropTableIfPresent("NEW_TABLE");
    assertTokensAndExecute(0, 0);
    DbEntity dbEntity = new DbEntity("NEW_TABLE");
    DbAttribute column1 = new DbAttribute("ID", Types.INTEGER, dbEntity);
    column1.setMandatory(true);
    column1.setPrimaryKey(true);
    dbEntity.addAttribute(column1);
    DbAttribute column2 = new DbAttribute("NAME", Types.VARCHAR, dbEntity);
    column2.setMaxLength(10);
    column2.setMandatory(false);
    dbEntity.addAttribute(column2);
    map.addDbEntity(dbEntity);
    assertTokensAndExecute(1, 0);
    assertTokensAndExecute(0, 0);
    ObjEntity objEntity = new ObjEntity("NewTable");
    objEntity.setDbEntity(dbEntity);
    ObjAttribute oatr1 = new ObjAttribute("name");
    oatr1.setDbAttributePath(column2.getName());
    oatr1.setType("java.lang.String");
    objEntity.addAttribute(oatr1);
    map.addObjEntity(objEntity);
    // force drop name column in db
    MergerToken token = mergerFactory().createDropColumnToDb(dbEntity, column2);
    execute(token);
    List<MergerToken> tokens = createMergeTokens();
    assertEquals(1, tokens.size());
    token = tokens.get(0);
    if (token.getDirection().isToDb()) {
        token = token.createReverse(mergerFactory());
    }
    assertTrue(token instanceof DropColumnToModel);
    execute(token);
    assertNull(dbEntity.getAttribute(column2.getName()));
    assertNull(objEntity.getAttribute(oatr1.getName()));
    // clear up
    map.removeObjEntity(objEntity.getName(), true);
    map.removeDbEntity(dbEntity.getName(), true);
    resolver.refreshMappingCache();
    assertNull(map.getObjEntity(objEntity.getName()));
    assertNull(map.getDbEntity(dbEntity.getName()));
    assertFalse(map.getDbEntities().contains(dbEntity));
    assertTokensAndExecute(1, 0);
    assertTokensAndExecute(0, 0);
}

Example 34 with DbAttribute

use of org.apache.cayenne.map.DbAttribute in project cayenne by apache.

the class AttributeLoaderIT method testAttributeLoad.

@Test
public void testAttributeLoad() throws Exception {
    createDbEntities();
    AttributeLoader loader = new AttributeLoader(adapter, EMPTY_CONFIG, new DefaultDbLoaderDelegate());
    loader.load(connection.getMetaData(), store);
    DbEntity artist = getDbEntity("ARTIST");
    DbAttribute a = getDbAttribute(artist, "ARTIST_ID");
    assertNotNull(a);
    if (accessStackAdapter.onlyGenericNumberType()) {
        assertEquals(Types.INTEGER, a.getType());
    } else {
        assertEquals(Types.BIGINT, a.getType());
    }
    assertTrue(a.isMandatory());
    assertFalse(a.isGenerated());
    a = getDbAttribute(artist, "ARTIST_NAME");
    assertNotNull(a);
    assertEquals(Types.CHAR, a.getType());
    assertEquals(254, a.getMaxLength());
    assertTrue(a.isMandatory());
    a = getDbAttribute(artist, "DATE_OF_BIRTH");
    assertNotNull(a);
    if (accessStackAdapter.onlyGenericDateType()) {
        assertTrue(Types.DATE == a.getType() || Types.TIMESTAMP == a.getType());
    } else {
        assertEquals(Types.DATE, a.getType());
    }
    assertFalse(a.isMandatory());
    if (accessStackAdapter.supportsLobs()) {
        assertLobDbEntities();
    }
    if (adapter.supportsGeneratedKeys()) {
        assertGenerated();
    }
}

Example 35 with DbAttribute

use of org.apache.cayenne.map.DbAttribute in project cayenne by apache.

the class AttributeLoaderIT method assertLobDbEntities.

private void assertLobDbEntities() {
    DbEntity blobEnt = getDbEntity("BLOB_TEST");
    assertNotNull(blobEnt);
    DbAttribute blobAttr = getDbAttribute(blobEnt, "BLOB_COL");
    assertNotNull(blobAttr);
    assertTrue(msgForTypeMismatch(Types.BLOB, blobAttr), Types.BLOB == blobAttr.getType() || Types.VARBINARY == blobAttr.getType() || Types.LONGVARBINARY == blobAttr.getType());
    DbEntity clobEnt = getDbEntity("CLOB_TEST");
    assertNotNull(clobEnt);
    DbAttribute clobAttr = getDbAttribute(clobEnt, "CLOB_COL");
    assertNotNull(clobAttr);
    assertTrue(msgForTypeMismatch(Types.CLOB, clobAttr), Types.CLOB == clobAttr.getType() || Types.VARCHAR == clobAttr.getType() || Types.LONGVARCHAR == clobAttr.getType());
}