package org.apache.cayenne.access;
import static org.apache.cayenne.exp.ExpressionFactory.matchExp;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;
import org.apache.cayenne.PersistenceState;
import org.apache.cayenne.ValueHolder;
import org.apache.cayenne.di.Inject;
import org.apache.cayenne.query.PrefetchTreeNode;
import org.apache.cayenne.query.SelectQuery;
import org.apache.cayenne.query.SortOrder;
import org.apache.cayenne.test.jdbc.DBHelper;
import org.apache.cayenne.test.jdbc.TableHelper;
import org.apache.cayenne.testdo.testmap.Bag;
import org.apache.cayenne.testdo.testmap.Ball;
import org.apache.cayenne.testdo.testmap.Box;
import org.apache.cayenne.testdo.testmap.BoxInfo;
import org.apache.cayenne.testdo.testmap.Thing;
import org.apache.cayenne.unit.di.DataChannelInterceptor;
import org.apache.cayenne.unit.di.UnitTestClosure;
import org.apache.cayenne.unit.di.server.ServerCase;
import org.apache.cayenne.unit.di.server.UseServerRuntime;
@UseServerRuntime(ServerCase.TESTMAP_PROJECT)
public class DataContextDisjointByIdPrefetchTest extends ServerCase {
@Inject
protected DataContext context;
@Inject
protected DBHelper dbHelper;
@Inject
protected DataChannelInterceptor queryInterceptor;
protected TableHelper tBag;
protected TableHelper tBox;
protected TableHelper tBoxInfo;
protected TableHelper tBall;
protected TableHelper tThing;
protected TableHelper tBoxThing;
@Override
protected void setUpAfterInjection() throws Exception {
dbHelper.deleteAll("BALL");
dbHelper.deleteAll("BOX_THING");
dbHelper.deleteAll("THING");
dbHelper.deleteAll("BOX_INFO");
dbHelper.deleteAll("BOX");
dbHelper.deleteAll("BAG");
tBag = new TableHelper(dbHelper, "BAG");
tBag.setColumns("ID", "NAME");
tBox = new TableHelper(dbHelper, "BOX");
tBox.setColumns("ID", "BAG_ID", "NAME");
tBoxInfo = new TableHelper(dbHelper, "BOX_INFO");
tBoxInfo.setColumns("ID", "BOX_ID", "COLOR");
tBall = new TableHelper(dbHelper, "BALL");
tBall.setColumns("ID", "BOX_ID", "THING_WEIGHT", "THING_VOLUME");
tThing = new TableHelper(dbHelper, "THING");
tThing.setColumns("ID", "WEIGHT", "VOLUME");
tBoxThing = new TableHelper(dbHelper, "BOX_THING");
tBoxThing.setColumns("BOX_ID", "THING_WEIGHT", "THING_VOLUME");
}
private void createBagWithTwoBoxesDataSet() throws Exception {
tBag.insert(1, "X");
tBox.insert(1, 1, "Y");
tBox.insert(2, 1, "Z");
}
private void createThreeBagsWithPlentyOfBoxesDataSet() throws Exception {
tBag.insert(1, "bag1");
tBag.insert(2, "bag2");
tBag.insert(3, "bag3");
tBox.insert(1, 1, "box1");
tBox.insert(2, 1, "box2");
tBox.insert(3, 1, "box3");
tBox.insert(4, 1, "box4");
tBox.insert(5, 1, "box5");
tBox.insert(6, 2, "box6");
tBox.insert(7, 2, "box7");
tBox.insert(8, 3, "box8");
tBox.insert(9, 3, "box9");
tBox.insert(10, 3, "box10");
}
private void createBagWithTwoBoxesAndPlentyOfBallsDataSet() throws Exception {
tBag.insert(1, "bag1");
tBox.insert(1, 1, "big");
tBoxInfo.insert(1, 1, "red");
tBox.insert(2, 1, "small");
tBoxInfo.insert(2, 2, "green");
tThing.insert(1, 10, 10);
tBoxThing.insert(1, 10, 10);
tBall.insert(1, 1, 10, 10);
tThing.insert(2, 20, 20);
tBoxThing.insert(1, 20, 20);
tBall.insert(2, 1, 20, 20);
tThing.insert(3, 30, 30);
tBoxThing.insert(2, 30, 30);
tBall.insert(3, 2, 30, 30);
tThing.insert(4, 40, 40);
tBoxThing.insert(2, 40, 40);
tBall.insert(4, 2, 40, 40);
tThing.insert(5, 10, 20);
tBoxThing.insert(2, 10, 20);
tBall.insert(5, 2, 10, 20);
tThing.insert(6, 30, 40);
tBoxThing.insert(2, 30, 40);
tBall.insert(6, 2, 30, 40);
}
public void testOneToMany() throws Exception {
createBagWithTwoBoxesDataSet();
SelectQuery query = new SelectQuery(Bag.class);
query.addPrefetch(Bag.BOXES_PROPERTY).setSemantics(
PrefetchTreeNode.DISJOINT_BY_ID_PREFETCH_SEMANTICS);
final List<Bag> result = context.performQuery(query);
queryInterceptor.runWithQueriesBlocked(new UnitTestClosure() {
public void execute() {
assertFalse(result.isEmpty());
Bag b1 = result.get(0);
List<Box> toMany = (List<Box>) b1
.readPropertyDirectly(Bag.BOXES_PROPERTY);
assertNotNull(toMany);
assertFalse(((ValueHolder) toMany).isFault());
assertEquals(2, toMany.size());
List<String> names = new ArrayList<String>();
for (Box b : toMany) {
assertEquals(PersistenceState.COMMITTED, b.getPersistenceState());
names.add(b.getName());
}
assertTrue(names.contains("Y"));
assertTrue(names.contains("Z"));
}
});
}
public void testManyToOne() throws Exception {
createBagWithTwoBoxesDataSet();
SelectQuery query = new SelectQuery(Box.class);
query.addPrefetch(Box.BAG_PROPERTY).setSemantics(
PrefetchTreeNode.DISJOINT_BY_ID_PREFETCH_SEMANTICS);
final List<Box> result = context.performQuery(query);
queryInterceptor.runWithQueriesBlocked(new UnitTestClosure() {
public void execute() {
assertFalse(result.isEmpty());
Box b1 = result.get(0);
assertNotNull(b1.getBag());
assertEquals(PersistenceState.COMMITTED, b1
.getBag()
.getPersistenceState());
assertEquals("X", b1.getBag().getName());
}
});
}
public void testFetchLimit() throws Exception {
createThreeBagsWithPlentyOfBoxesDataSet();
final SelectQuery query = new SelectQuery(Bag.class);
query.addPrefetch(Bag.BOXES_PROPERTY).setSemantics(
PrefetchTreeNode.DISJOINT_BY_ID_PREFETCH_SEMANTICS);
query.addOrdering("db:" + Bag.ID_PK_COLUMN, SortOrder.ASCENDING);
query.setFetchLimit(2);
// There will be only 2 bags in a result. The first bag has 5 boxes and
// the second has 2. So we are expecting exactly 9 snapshots in the data
// row store after performing the query.
final List<Bag> bags = context.performQuery(query);
queryInterceptor.runWithQueriesBlocked(new UnitTestClosure() {
public void execute() {
assertEquals(2, bags.size());
assertEquals(5, bags.get(0).getBoxes().size());
assertEquals(2, bags.get(1).getBoxes().size());
for (Bag b : bags) {
b.getName();
for (Box bx : b.getBoxes()) {
bx.getName();
}
}
}
});
}
public void testOneToOneRelationship() throws Exception {
createBagWithTwoBoxesAndPlentyOfBallsDataSet();
SelectQuery query = new SelectQuery(Box.class);
query.addPrefetch(Box.BOX_INFO_PROPERTY).setSemantics(
PrefetchTreeNode.DISJOINT_BY_ID_PREFETCH_SEMANTICS);
final List<Box> result = context.performQuery(query);
queryInterceptor.runWithQueriesBlocked(new UnitTestClosure() {
public void execute() {
assertFalse(result.isEmpty());
List<String> boxColors = new ArrayList<String>();
for (Box box : result) {
BoxInfo info = (BoxInfo) box
.readPropertyDirectly(Box.BOX_INFO_PROPERTY);
assertNotNull(info);
boxColors.add(info.getColor());
assertEquals(PersistenceState.COMMITTED, info.getPersistenceState());
}
assertTrue(boxColors.containsAll(Arrays.asList("red", "green")));
}
});
}
public void testFlattenedRelationship() throws Exception {
createBagWithTwoBoxesAndPlentyOfBallsDataSet();
SelectQuery query = new SelectQuery(Bag.class);
query.addPrefetch(Bag.BALLS_PROPERTY).setSemantics(
PrefetchTreeNode.DISJOINT_BY_ID_PREFETCH_SEMANTICS);
final List<Bag> result = context.performQuery(query);
queryInterceptor.runWithQueriesBlocked(new UnitTestClosure() {
public void execute() {
assertFalse(result.isEmpty());
Bag b1 = result.get(0);
List<Ball> balls = (List<Ball>) b1
.readPropertyDirectly(Bag.BALLS_PROPERTY);
assertNotNull(balls);
assertFalse(((ValueHolder) balls).isFault());
assertEquals(6, balls.size());
List<Integer> volumes = new ArrayList<Integer>();
for (Ball b : balls) {
assertEquals(PersistenceState.COMMITTED, b.getPersistenceState());
volumes.add(b.getThingVolume());
}
assertTrue(volumes.containsAll(Arrays.asList(10, 20, 30, 40, 20, 40)));
}
});
}
public void testLongFlattenedRelationship() throws Exception {
createBagWithTwoBoxesAndPlentyOfBallsDataSet();
SelectQuery query = new SelectQuery(Bag.class);
query.addPrefetch(Bag.THINGS_PROPERTY).setSemantics(
PrefetchTreeNode.DISJOINT_BY_ID_PREFETCH_SEMANTICS);
final List<Bag> result = context.performQuery(query);
queryInterceptor.runWithQueriesBlocked(new UnitTestClosure() {
public void execute() {
assertFalse(result.isEmpty());
Bag b1 = result.get(0);
List<Thing> things = (List<Thing>) b1
.readPropertyDirectly(Bag.THINGS_PROPERTY);
assertNotNull(things);
assertFalse(((ValueHolder) things).isFault());
assertEquals(6, things.size());
List<Integer> volumes = new ArrayList<Integer>();
for (Thing t : things) {
assertEquals(PersistenceState.COMMITTED, t.getPersistenceState());
volumes.add(t.getVolume());
}
assertTrue(volumes.containsAll(Arrays.asList(10, 20, 20, 30, 40, 40)));
}
});
}
public void testMultiColumnRelationship() throws Exception {
createBagWithTwoBoxesAndPlentyOfBallsDataSet();
SelectQuery query = new SelectQuery(Ball.class);
query.orQualifier(matchExp(Ball.THING_VOLUME_PROPERTY, 40).andExp(
matchExp(Ball.THING_WEIGHT_PROPERTY, 30)));
query.orQualifier(matchExp(Ball.THING_VOLUME_PROPERTY, 20).andExp(
matchExp(Ball.THING_WEIGHT_PROPERTY, 10)));
query.addPrefetch(Ball.THING_PROPERTY).setSemantics(
PrefetchTreeNode.DISJOINT_BY_ID_PREFETCH_SEMANTICS);
final List<Ball> balls = context.performQuery(query);
queryInterceptor.runWithQueriesBlocked(new UnitTestClosure() {
public void execute() {
assertEquals(2, balls.size());
balls.get(0).getThing().getVolume();
balls.get(1).getThing().getVolume();
}
});
}
public void testFlattenedMultiColumnRelationship() throws Exception {
createBagWithTwoBoxesAndPlentyOfBallsDataSet();
SelectQuery query = new SelectQuery(Box.class);
query.addPrefetch(Box.THINGS_PROPERTY).setSemantics(
PrefetchTreeNode.DISJOINT_BY_ID_PREFETCH_SEMANTICS);
final List<Box> result = context.performQuery(query);
queryInterceptor.runWithQueriesBlocked(new UnitTestClosure() {
public void execute() {
assertFalse(result.isEmpty());
List<Integer> volumes = new ArrayList<Integer>();
for (Box box : result) {
List<Thing> things = (List<Thing>) box
.readPropertyDirectly(Box.THINGS_PROPERTY);
assertNotNull(things);
assertFalse(((ValueHolder) things).isFault());
for (Thing t : things) {
assertEquals(PersistenceState.COMMITTED, t.getPersistenceState());
volumes.add(t.getVolume());
}
}
assertEquals(6, volumes.size());
assertTrue(volumes.containsAll(Arrays.asList(10, 20, 30, 40)));
}
});
}
public void testJointPrefetchInParent() throws Exception {
createBagWithTwoBoxesAndPlentyOfBallsDataSet();
SelectQuery query = new SelectQuery(Box.class);
query.addPrefetch(Box.BALLS_PROPERTY).setSemantics(
PrefetchTreeNode.JOINT_PREFETCH_SEMANTICS);
query.addPrefetch(Box.BALLS_PROPERTY + "." + Ball.THING_PROPERTY).setSemantics(
PrefetchTreeNode.DISJOINT_BY_ID_PREFETCH_SEMANTICS);
final List<Box> result = context.performQuery(query);
queryInterceptor.runWithQueriesBlocked(new UnitTestClosure() {
public void execute() {
assertFalse(result.isEmpty());
List<Integer> volumes = new ArrayList<Integer>();
for (Box box : result) {
List<Ball> balls = (List<Ball>) box
.readPropertyDirectly(Box.BALLS_PROPERTY);
assertNotNull(balls);
assertFalse(((ValueHolder) balls).isFault());
for (Ball ball : balls) {
Thing thing = (Thing) ball
.readPropertyDirectly(Ball.THING_PROPERTY);
assertNotNull(thing);
assertEquals(
PersistenceState.COMMITTED,
thing.getPersistenceState());
volumes.add(thing.getVolume());
}
}
assertEquals(6, volumes.size());
assertTrue(volumes.containsAll(Arrays.asList(10, 20, 30, 40)));
}
});
}
public void testJointPrefetchInChild() throws Exception {
createBagWithTwoBoxesAndPlentyOfBallsDataSet();
SelectQuery query = new SelectQuery(Bag.class);
query.addPrefetch(Bag.BOXES_PROPERTY).setSemantics(
PrefetchTreeNode.DISJOINT_BY_ID_PREFETCH_SEMANTICS);
query.addPrefetch(Bag.BOXES_PROPERTY + "." + Box.BALLS_PROPERTY).setSemantics(
PrefetchTreeNode.JOINT_PREFETCH_SEMANTICS);
final List<Bag> result = context.performQuery(query);
queryInterceptor.runWithQueriesBlocked(new UnitTestClosure() {
public void execute() {
assertFalse(result.isEmpty());
Bag bag = result.get(0);
List<Box> boxes = (List<Box>) bag
.readPropertyDirectly(Bag.BOXES_PROPERTY);
assertNotNull(boxes);
assertFalse(((ValueHolder) boxes).isFault());
assertEquals(2, boxes.size());
List<String> names = new ArrayList<String>();
for (Box box : boxes) {
assertEquals(PersistenceState.COMMITTED, box.getPersistenceState());
names.add(box.getName());
}
assertTrue(names.contains("big"));
assertTrue(names.contains("small"));
Box box = boxes.get(0);
List<Ball> balls = (List<Ball>) box
.readPropertyDirectly(Box.BALLS_PROPERTY);
assertNotNull(balls);
assertFalse(((ValueHolder) balls).isFault());
assertEquals(2, balls.size());
List<Integer> volumes = new ArrayList<Integer>();
for (Ball ball : balls) {
assertEquals(PersistenceState.COMMITTED, ball.getPersistenceState());
volumes.add(ball.getThingVolume());
}
assertTrue(volumes.containsAll(Arrays.asList(10, 20)));
}
});
}
}