/*****************************************************************
* Licensed to the Apache Software Foundation (ASF) under one
* or more contributor license agreements. See the NOTICE file
* distributed with this work for additional information
* regarding copyright ownership. The ASF licenses this file
* to you under the Apache License, Version 2.0 (the
* "License"); you may not use this file except in compliance
* with the License. You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
* KIND, either express or implied. See the License for the
* specific language governing permissions and limitations
* under the License.
****************************************************************/
package org.apache.cayenne;
import java.util.Iterator;
import java.util.List;
import org.apache.cayenne.access.ClientServerChannel;
import org.apache.cayenne.access.DataContext;
import org.apache.cayenne.di.Inject;
import org.apache.cayenne.map.LifecycleEvent;
import org.apache.cayenne.query.ObjectIdQuery;
import org.apache.cayenne.query.QueryCacheStrategy;
import org.apache.cayenne.query.SelectQuery;
import org.apache.cayenne.query.SortOrder;
import org.apache.cayenne.reflect.LifecycleCallbackRegistry;
import org.apache.cayenne.remote.RemoteIncrementalFaultList;
import org.apache.cayenne.test.jdbc.DBHelper;
import org.apache.cayenne.test.jdbc.TableHelper;
import org.apache.cayenne.testdo.mt.ClientMtMeaningfulPk;
import org.apache.cayenne.testdo.mt.ClientMtReflexive;
import org.apache.cayenne.testdo.mt.ClientMtTable1;
import org.apache.cayenne.testdo.mt.ClientMtTable2;
import org.apache.cayenne.testdo.mt.MtReflexive;
import org.apache.cayenne.testdo.mt.MtTable1;
import org.apache.cayenne.unit.di.DataChannelInterceptor;
import org.apache.cayenne.unit.di.UnitTestClosure;
import org.apache.cayenne.unit.di.client.ClientCase;
import org.apache.cayenne.unit.di.server.UseServerRuntime;
@UseServerRuntime(ClientCase.MULTI_TIER_PROJECT)
public class CayenneContextWithDataContextTest extends ClientCase {
@Inject
private CayenneContext clientContext;
@Inject
private DBHelper dbHelper;
@Inject(ClientCase.ROP_CLIENT_KEY)
private DataChannelInterceptor clientServerInterceptor;
@Inject
private ClientServerChannel clientServerChannel;
private TableHelper tMtTable1;
private TableHelper tMtTable2;
private TableHelper tMtMeaningfulPK;
@Override
protected void setUpAfterInjection() throws Exception {
dbHelper.deleteAll("MT_TABLE2");
dbHelper.deleteAll("MT_TABLE1");
tMtTable1 = new TableHelper(dbHelper, "MT_TABLE1");
tMtTable1.setColumns("TABLE1_ID", "GLOBAL_ATTRIBUTE1", "SERVER_ATTRIBUTE1");
tMtTable2 = new TableHelper(dbHelper, "MT_TABLE2");
tMtTable2.setColumns("TABLE2_ID", "TABLE1_ID", "GLOBAL_ATTRIBUTE");
tMtMeaningfulPK = new TableHelper(dbHelper, "MT_MEANINGFUL_PK");
tMtMeaningfulPK.setColumns("PK");
}
private void deleteAndCreateTwoMeaningfulPKsDataSet() throws Exception {
tMtMeaningfulPK.deleteAll();
tMtMeaningfulPK.insert("A");
tMtMeaningfulPK.insert("B");
}
private void createTwoMtTable1sAnd2sDataSet() throws Exception {
tMtTable1.insert(1, "g1", "s1");
tMtTable1.insert(2, "g2", "s2");
tMtTable2.insert(1, 1, "g1");
tMtTable2.insert(2, 1, "g2");
}
private void createEightMtTable1s() throws Exception {
for (int i = 1; i <= 8; i++) {
tMtTable1.insert(i, "g" + i, "s" + i);
}
}
public void testLocalCacheStaysLocal() {
DataContext serverContext = (DataContext) clientServerChannel.getParentChannel();
SelectQuery query = new SelectQuery(ClientMtTable1.class);
query.setCacheStrategy(QueryCacheStrategy.LOCAL_CACHE);
assertEquals(0, clientContext.getQueryCache().size());
assertEquals(0, serverContext.getQueryCache().size());
List<?> results = clientContext.performQuery(query);
assertEquals(1, clientContext.getQueryCache().size());
assertSame(results, clientContext.getQueryCache().get(
query.getMetaData(clientContext.getEntityResolver())));
assertEquals(0, serverContext.getQueryCache().size());
}
public void testPostAddCallback() throws Exception {
LifecycleCallbackRegistry callbackRegistry = clientServerChannel
.getEntityResolver()
.getCallbackRegistry();
final boolean[] flag = new boolean[1];
try {
callbackRegistry.addListener(MtTable1.class, new LifecycleListener() {
public void postLoad(Object entity) {
}
public void postPersist(Object entity) {
}
public void postRemove(Object entity) {
}
public void postUpdate(Object entity) {
}
public void postAdd(Object entity) {
flag[0] = true;
}
public void preRemove(Object entity) {
}
public void preUpdate(Object entity) {
}
public void prePersist(Object entity) {
}
});
clientContext.newObject(ClientMtTable1.class);
assertFalse(flag[0]);
clientContext.commitChanges();
assertTrue(flag[0]);
}
finally {
callbackRegistry.clear();
}
}
public void testPostAddOnObjectCallback() throws Exception {
DataContext serverContext = (DataContext) clientServerChannel.getParentChannel();
LifecycleCallbackRegistry callbackRegistry = serverContext
.getEntityResolver()
.getCallbackRegistry();
try {
callbackRegistry.addListener(
LifecycleEvent.POST_ADD,
MtTable1.class,
"prePersistMethod");
Persistent clientObject = clientContext.newObject(ClientMtTable1.class);
clientContext.commitChanges();
// find peer
MtTable1 peer = (MtTable1) serverContext.getGraphManager().getNode(
clientObject.getObjectId());
assertNotNull(peer);
assertTrue(peer.isPrePersisted());
}
finally {
callbackRegistry.clear();
}
}
public void testPreRemoveCallback() throws Exception {
// an exception was triggered within POST_LOAD callback
LifecycleCallbackRegistry callbackRegistry = clientServerChannel
.getEntityResolver()
.getCallbackRegistry();
final boolean[] flag = new boolean[1];
try {
callbackRegistry.addListener(MtTable1.class, new LifecycleListener() {
public void postLoad(Object entity) {
}
public void postPersist(Object entity) {
}
public void postRemove(Object entity) {
}
public void postUpdate(Object entity) {
}
public void postAdd(Object entity) {
}
public void preRemove(Object entity) {
flag[0] = true;
}
public void preUpdate(Object entity) {
}
public void prePersist(Object entity) {
}
});
ClientMtTable1 object = clientContext.newObject(ClientMtTable1.class);
assertFalse(flag[0]);
clientContext.commitChanges();
assertFalse(flag[0]);
clientContext.deleteObject(object);
clientContext.commitChanges();
assertTrue(flag[0]);
}
finally {
callbackRegistry.clear();
}
}
public void testCAY830() throws Exception {
// an exception was triggered within POST_LOAD callback
LifecycleCallbackRegistry callbackRegistry = clientServerChannel
.getEntityResolver()
.getCallbackRegistry();
try {
callbackRegistry.addListener(MtReflexive.class, new LifecycleListener() {
public void postLoad(Object entity) {
}
public void postPersist(Object entity) {
}
public void postRemove(Object entity) {
}
public void postUpdate(Object entity) {
}
public void postAdd(Object entity) {
}
public void preRemove(Object entity) {
}
public void preUpdate(Object entity) {
}
public void prePersist(Object entity) {
}
});
ClientMtReflexive o1 = clientContext.newObject(ClientMtReflexive.class);
o1.setName("parent");
ClientMtReflexive o2 = clientContext.newObject(ClientMtReflexive.class);
o2.setName("child");
o2.setToParent(o1);
clientContext.commitChanges();
clientContext.deleteObject(o1);
clientContext.deleteObject(o2);
clientContext.commitChanges();
// per CAY-830 an exception is thrown here
}
finally {
callbackRegistry.clear();
}
}
public void testRollbackChanges() throws Exception {
ClientMtTable1 o = clientContext.newObject(ClientMtTable1.class);
o.setGlobalAttribute1("1");
clientContext.commitChanges();
assertEquals("1", o.getGlobalAttribute1());
o.setGlobalAttribute1("2");
assertEquals("2", o.getGlobalAttribute1());
clientContext.rollbackChanges();
assertEquals("1", o.getGlobalAttribute1());
assertTrue(clientContext.modifiedObjects().isEmpty());
}
public void testCreateFault() throws Exception {
tMtTable1.insert(1, "g1", "s1");
ObjectId id = new ObjectId("MtTable1", MtTable1.TABLE1_ID_PK_COLUMN, 1);
Object fault = clientContext.createFault(id);
assertTrue(fault instanceof ClientMtTable1);
ClientMtTable1 o = (ClientMtTable1) fault;
assertEquals(PersistenceState.HOLLOW, o.getPersistenceState());
assertSame(clientContext, o.getObjectContext());
assertNull(o.getGlobalAttribute1Direct());
// make sure we haven't tripped the fault yet
assertEquals(PersistenceState.HOLLOW, o.getPersistenceState());
// try tripping fault
assertEquals("g1", o.getGlobalAttribute1());
assertEquals(PersistenceState.COMMITTED, o.getPersistenceState());
}
public void testCreateBadFault() throws Exception {
tMtTable1.insert(1, "g1", "s1");
ObjectId id = new ObjectId("MtTable1", MtTable1.TABLE1_ID_PK_COLUMN, 2);
Object fault = clientContext.createFault(id);
assertTrue(fault instanceof ClientMtTable1);
ClientMtTable1 o = (ClientMtTable1) fault;
// try tripping fault
try {
o.getGlobalAttribute1();
fail("resolving bad fault should've thrown");
}
catch (FaultFailureException e) {
// expected
}
}
public void testMeaningfulPK() throws Exception {
deleteAndCreateTwoMeaningfulPKsDataSet();
SelectQuery query = new SelectQuery(ClientMtMeaningfulPk.class);
query.addOrdering(ClientMtMeaningfulPk.PK_PROPERTY, SortOrder.DESCENDING);
List<?> results = clientContext.performQuery(query);
assertEquals(2, results.size());
}
public void testPrefetchingToOne() throws Exception {
createTwoMtTable1sAnd2sDataSet();
final ObjectId prefetchedId = new ObjectId(
"MtTable1",
MtTable1.TABLE1_ID_PK_COLUMN,
1);
SelectQuery q = new SelectQuery(ClientMtTable2.class);
q.addOrdering(ClientMtTable2.GLOBAL_ATTRIBUTE_PROPERTY, SortOrder.ASCENDING);
q.addPrefetch(ClientMtTable2.TABLE1_PROPERTY);
final List<ClientMtTable2> results = clientContext.performQuery(q);
clientServerInterceptor.runWithQueriesBlocked(new UnitTestClosure() {
public void execute() {
assertEquals(2, results.size());
for (ClientMtTable2 o : results) {
assertEquals(PersistenceState.COMMITTED, o.getPersistenceState());
assertSame(clientContext, o.getObjectContext());
ClientMtTable1 o1 = o.getTable1();
assertNotNull(o1);
assertEquals(PersistenceState.COMMITTED, o1.getPersistenceState());
assertSame(clientContext, o1.getObjectContext());
assertEquals(prefetchedId, o1.getObjectId());
}
}
});
}
public void testPrefetchingToOneNull() throws Exception {
tMtTable2.insert(15, null, "g3");
SelectQuery q = new SelectQuery(ClientMtTable2.class);
q.addPrefetch(ClientMtTable2.TABLE1_PROPERTY);
final List<ClientMtTable2> results = clientContext.performQuery(q);
clientServerInterceptor.runWithQueriesBlocked(new UnitTestClosure() {
public void execute() {
assertEquals(1, results.size());
ClientMtTable2 o = results.get(0);
assertEquals(PersistenceState.COMMITTED, o.getPersistenceState());
assertSame(clientContext, o.getObjectContext());
assertNull(o.getTable1());
}
});
}
public void testPrefetchingToMany() throws Exception {
createTwoMtTable1sAnd2sDataSet();
SelectQuery q = new SelectQuery(ClientMtTable1.class);
q.addOrdering(ClientMtTable1.GLOBAL_ATTRIBUTE1_PROPERTY, SortOrder.ASCENDING);
q.addPrefetch(ClientMtTable1.TABLE2ARRAY_PROPERTY);
final List<ClientMtTable1> results = clientContext.performQuery(q);
clientServerInterceptor.runWithQueriesBlocked(new UnitTestClosure() {
public void execute() {
ClientMtTable1 o1 = results.get(0);
assertEquals(PersistenceState.COMMITTED, o1.getPersistenceState());
assertSame(clientContext, o1.getObjectContext());
List<?> children1 = o1.getTable2Array();
assertEquals(2, children1.size());
Iterator<?> it = children1.iterator();
while (it.hasNext()) {
ClientMtTable2 o = (ClientMtTable2) it.next();
assertEquals(PersistenceState.COMMITTED, o.getPersistenceState());
assertSame(clientContext, o.getObjectContext());
// TODO: fixme... reverse relationship is not connected and will
// cause a fetch
// assertEquals(o1, o.getTable1());
}
}
});
}
public void testPerformPaginatedQuery() throws Exception {
createEightMtTable1s();
SelectQuery query = new SelectQuery(ClientMtTable1.class);
query.setPageSize(5);
List<ClientMtTable1> objects = clientContext.performQuery(query);
assertNotNull(objects);
assertTrue(objects instanceof RemoteIncrementalFaultList);
}
public void testPrefetchingToManyEmpty() throws Exception {
createTwoMtTable1sAnd2sDataSet();
SelectQuery q = new SelectQuery(ClientMtTable1.class);
q.addOrdering(ClientMtTable1.GLOBAL_ATTRIBUTE1_PROPERTY, SortOrder.ASCENDING);
q.addPrefetch(ClientMtTable1.TABLE2ARRAY_PROPERTY);
final List<ClientMtTable1> results = clientContext.performQuery(q);
clientServerInterceptor.runWithQueriesBlocked(new UnitTestClosure() {
public void execute() {
ClientMtTable1 o2 = results.get(1);
assertEquals(PersistenceState.COMMITTED, o2.getPersistenceState());
assertSame(clientContext, o2.getObjectContext());
List<ClientMtTable2> children2 = o2.getTable2Array();
assertFalse(((ValueHolder) children2).isFault());
assertEquals(0, children2.size());
}
});
}
public void testOIDQueryInterception() throws Exception {
final ClientMtTable1 o = clientContext.newObject(ClientMtTable1.class);
o.setGlobalAttribute1("aaa");
// fetch new
final ObjectIdQuery q1 = new ObjectIdQuery(
o.getObjectId(),
false,
ObjectIdQuery.CACHE);
clientServerInterceptor.runWithQueriesBlocked(new UnitTestClosure() {
public void execute() {
List<?> objects = clientContext.performQuery(q1);
assertEquals(1, objects.size());
assertSame(o, objects.get(0));
}
});
clientContext.commitChanges();
// fetch committed
final ObjectIdQuery q2 = new ObjectIdQuery(
o.getObjectId(),
false,
ObjectIdQuery.CACHE);
clientServerInterceptor.runWithQueriesBlocked(new UnitTestClosure() {
public void execute() {
List<?> objects = clientContext.performQuery(q2);
assertEquals(1, objects.size());
assertSame(o, objects.get(0));
}
});
}
}