Search in sources :

Example 6 with TestUtilsV2.randomString

use of org.apache.atlas.TestUtilsV2.randomString in project atlas by apache.

the class AtlasEntityStoreV1Test method testClassUpdate.

@Test(dependsOnMethods = "testCreate")
public void testClassUpdate() throws Exception {
    init();
    // Create new db instance
    final AtlasEntity databaseInstance = TestUtilsV2.createDBEntity();
    EntityMutationResponse response = entityStore.createOrUpdate(new AtlasEntityStream(databaseInstance), false);
    final AtlasEntityHeader dbCreated = response.getFirstCreatedEntityByTypeName(TestUtilsV2.DATABASE_TYPE);
    init();
    Map<String, AtlasEntity> tableCloneMap = new HashMap<>();
    AtlasEntity tableClone = new AtlasEntity(tblEntity.getEntity());
    tableClone.setAttribute("database", new AtlasObjectId(dbCreated.getGuid(), TestUtilsV2.DATABASE_TYPE));
    tableCloneMap.put(dbCreated.getGuid(), databaseInstance);
    tableCloneMap.put(tableClone.getGuid(), tableClone);
    response = entityStore.createOrUpdate(new InMemoryMapEntityStream(tableCloneMap), false);
    final AtlasEntityHeader tableDefinition = response.getFirstUpdatedEntityByTypeName(TABLE_TYPE);
    AtlasEntity updatedTableDefinition = getEntityFromStore(tableDefinition);
    Assert.assertNotNull(updatedTableDefinition.getAttribute("database"));
    Assert.assertEquals(((AtlasObjectId) updatedTableDefinition.getAttribute("database")).getGuid(), dbCreated.getGuid());
}
Also used : HashMap(java.util.HashMap) AtlasEntity(org.apache.atlas.model.instance.AtlasEntity) EntityMutationResponse(org.apache.atlas.model.instance.EntityMutationResponse) AtlasEntityHeader(org.apache.atlas.model.instance.AtlasEntityHeader) AtlasObjectId(org.apache.atlas.model.instance.AtlasObjectId) TestUtilsV2.randomString(org.apache.atlas.TestUtilsV2.randomString) Test(org.testng.annotations.Test) BeforeTest(org.testng.annotations.BeforeTest)

Example 7 with TestUtilsV2.randomString

use of org.apache.atlas.TestUtilsV2.randomString in project drbookings by DrBookings.

the class Anonymiser method main.

public static void main(final String[] args) throws Exception {
    final File file = new File("/home/alex/bookings-anonym.xml");
    new BackupCreator().makeBackup(file);
    final DataStore ds = new XMLStorage().load(file);
    for (final Iterator<BookingBeanSer> it = ds.getBookingsSer().iterator(); it.hasNext(); ) {
        final BookingBeanSer bs = it.next();
        bs.guestName = new RandomString().ofLength(bs.guestName.length()).toString();
    }
    for (final Iterator<CleaningBeanSer> it = ds.getCleaningsSer().iterator(); it.hasNext(); ) {
        final CleaningBeanSer bs = it.next();
        bs.name = new RandomString().ofLength(bs.name.length()).toString();
    }
    new XMLStorage().save(ds, file);
}
Also used : XMLStorage(com.github.drbookings.ser.XMLStorage) CleaningBeanSer(com.github.drbookings.model.ser.CleaningBeanSer) BackupCreator(com.github.ktools1000.io.BackupCreator) DataStore(com.github.drbookings.ser.DataStore) RandomString(com.github.ktools1000.RandomString) File(java.io.File) BookingBeanSer(com.github.drbookings.model.ser.BookingBeanSer)

Aggregations

TestUtilsV2.randomString (org.apache.atlas.TestUtilsV2.randomString)6 AtlasEntity (org.apache.atlas.model.instance.AtlasEntity)6 BeforeTest (org.testng.annotations.BeforeTest)6 Test (org.testng.annotations.Test)6 EntityMutationResponse (org.apache.atlas.model.instance.EntityMutationResponse)5 HashMap (java.util.HashMap)4 AtlasEntityHeader (org.apache.atlas.model.instance.AtlasEntityHeader)4 AtlasEntityWithExtInfo (org.apache.atlas.model.instance.AtlasEntity.AtlasEntityWithExtInfo)2 AtlasObjectId (org.apache.atlas.model.instance.AtlasObjectId)2 AtlasTypesDef (org.apache.atlas.model.typedef.AtlasTypesDef)2 BookingBeanSer (com.github.drbookings.model.ser.BookingBeanSer)1 CleaningBeanSer (com.github.drbookings.model.ser.CleaningBeanSer)1 DataStore (com.github.drbookings.ser.DataStore)1 XMLStorage (com.github.drbookings.ser.XMLStorage)1 RandomString (com.github.ktools1000.RandomString)1 BackupCreator (com.github.ktools1000.io.BackupCreator)1 File (java.io.File)1 Map (java.util.Map)1 AtlasBaseException (org.apache.atlas.exception.AtlasBaseException)1 AtlasClassification (org.apache.atlas.model.instance.AtlasClassification)1