bidir1.setChildVal("yam1");
bidir2.setChildVal("yam2");
Book b1 = newBook();
Book b2 = newBook();
b2.setTitle("another title");
HasKeyPkJPA hasKeyPk1 = new HasKeyPkJPA(nextNamedKey());
HasKeyPkJPA hasKeyPk2 = new HasKeyPkJPA(nextNamedKey());
hasKeyPk2.setStr("yar 2");
pojo.getBooks().add(b1);
pojo.getBooks().add(b2);
pojo.getHasKeyPks().add(hasKeyPk1);
pojo.getHasKeyPks().add(hasKeyPk2);
pojo.getBidirChildren().add(bidir1);
pojo.getBidirChildren().add(bidir2);
startEnd.start();
em.persist(pojo);
startEnd.end();
assertCountsInDatastore(pojo.getClass(), bidir1.getClass(), 1, 2);
String bidir1Id = bidir1.getId();
String bookId = b1.getId();
Key hasKeyPk1Key = hasKeyPk1.getId();
pojo.getBidirChildren().remove(bidir1);
pojo.getBooks().remove(b1);
pojo.getHasKeyPks().remove(hasKeyPk1);
startEnd.start();
em.merge(pojo);
startEnd.end();
startEnd.start();
pojo = em.find(pojo.getClass(), pojo.getId());
assertNotNull(pojo.getId());
assertEquals(1, pojo.getBooks().size());
assertEquals(1, pojo.getHasKeyPks().size());
assertEquals(1, pojo.getBidirChildren().size());
assertNotNull(bidir2.getId());
assertNotNull(bidir2.getParent());
assertNotNull(b2.getId());
assertNotNull(hasKeyPk2.getId());
assertEquals(bidir2.getId(), pojo.getBidirChildren().iterator().next().getId());
assertEquals(hasKeyPk2.getId(), pojo.getHasKeyPks().iterator().next().getId());
assertEquals(b2.getId(), pojo.getBooks().iterator().next().getId());
Entity pojoEntity = ds.get(KeyFactory.stringToKey(pojo.getId()));
assertNotNull(pojoEntity);
assertEquals(4, pojoEntity.getProperties().size());
assertEquals(Collections.singletonList(KeyFactory.stringToKey(bidir2.getId())), pojoEntity.getProperty("bidirChildren"));
assertEquals(Collections.singletonList(KeyFactory.stringToKey(b2.getId())), pojoEntity.getProperty("books"));
assertEquals(Collections.singletonList(hasKeyPk2.getId()), pojoEntity.getProperty("hasKeyPks"));
startEnd.end();
try {
ds.get(KeyFactory.stringToKey(bidir1Id));
fail("expected EntityNotFoundException");
} catch (EntityNotFoundException enfe) {
// good
}
try {
ds.get(KeyFactory.stringToKey(bookId));
fail("expected EntityNotFoundException");
} catch (EntityNotFoundException enfe) {
// good
}
try {
ds.get(hasKeyPk1Key);
fail("expected EntityNotFoundException");
} catch (EntityNotFoundException enfe) {
// good
}
Entity bidirChildEntity = ds.get(KeyFactory.stringToKey(bidir2.getId()));
assertNotNull(bidirChildEntity);
assertEquals("yam2", bidirChildEntity.getProperty("childVal"));
assertEquals(KeyFactory.stringToKey(bidir2.getId()), bidirChildEntity.getKey());
assertKeyParentEquals(pojo.getId(), bidirChildEntity, bidir2.getId());
Entity bookEntity = ds.get(KeyFactory.stringToKey(b2.getId()));
assertNotNull(bookEntity);
assertEquals("max", bookEntity.getProperty("author"));
assertEquals("22333", bookEntity.getProperty("isbn"));
assertEquals("another title", bookEntity.getProperty("title"));
assertEquals(KeyFactory.stringToKey(b2.getId()), bookEntity.getKey());
assertKeyParentEquals(pojo.getId(), bookEntity, b2.getId());
Entity hasKeyPkEntity = ds.get(hasKeyPk2.getId());
assertNotNull(hasKeyPkEntity);
assertEquals("yar 2", hasKeyPkEntity.getProperty("str"));
assertEquals(hasKeyPk2.getId(), hasKeyPkEntity.getKey());
assertKeyParentEquals(pojo.getId(), hasKeyPkEntity, hasKeyPk2.getId());
Entity parentEntity = ds.get(KeyFactory.stringToKey(pojo.getId()));
assertNotNull(parentEntity);
assertEquals(4, parentEntity.getProperties().size());
assertEquals("yar", parentEntity.getProperty("val"));