/*-
* See the file LICENSE for redistribution information.
*
* Copyright (c) 2000-2005
* Sleepycat Software. All rights reserved.
*
* $Id: TupleSerialFactoryTest.java,v 1.36 2004/12/22 14:11:42 linda Exp $
*/
package com.sleepycat.collections.test.serial;
import java.util.Map;
import junit.framework.Test;
import junit.framework.TestCase;
import junit.framework.TestSuite;
import com.sleepycat.bind.serial.StoredClassCatalog;
import com.sleepycat.bind.serial.test.MarshalledObject;
import com.sleepycat.collections.TransactionRunner;
import com.sleepycat.collections.TransactionWorker;
import com.sleepycat.collections.TupleSerialFactory;
import com.sleepycat.collections.test.DbTestUtil;
import com.sleepycat.collections.test.TestEnv;
import com.sleepycat.compat.DbCompat;
import com.sleepycat.je.Database;
import com.sleepycat.je.DatabaseConfig;
import com.sleepycat.je.Environment;
import com.sleepycat.je.ForeignKeyDeleteAction;
import com.sleepycat.je.SecondaryConfig;
import com.sleepycat.je.SecondaryDatabase;
/**
* @author Mark Hayes
*/
public class TupleSerialFactoryTest extends TestCase
implements TransactionWorker {
public static void main(String[] args)
throws Exception {
junit.framework.TestResult tr =
junit.textui.TestRunner.run(suite());
if (tr.errorCount() > 0 ||
tr.failureCount() > 0) {
System.exit(1);
} else {
System.exit(0);
}
}
public static Test suite()
throws Exception {
TestSuite suite = new TestSuite();
for (int i = 0; i < TestEnv.ALL.length; i += 1) {
for (int sorted = 0; sorted < 2; sorted += 1) {
suite.addTest(new TupleSerialFactoryTest(TestEnv.ALL[i],
sorted != 0));
}
}
return suite;
}
private TestEnv testEnv;
private Environment env;
private StoredClassCatalog catalog;
private TransactionRunner runner;
private TupleSerialFactory factory;
private Database store1;
private Database store2;
private SecondaryDatabase index1;
private SecondaryDatabase index2;
private boolean isSorted;
private Map storeMap1;
private Map storeMap2;
private Map indexMap1;
private Map indexMap2;
public TupleSerialFactoryTest(TestEnv testEnv, boolean isSorted) {
super(null);
this.testEnv = testEnv;
this.isSorted = isSorted;
String name = "TupleSerialFactoryTest-" + testEnv.getName();
name += isSorted ? "-sorted" : "-unsorted";
setName(name);
}
public void setUp()
throws Exception {
DbTestUtil.printTestName(getName());
env = testEnv.open(getName());
runner = new TransactionRunner(env);
createDatabase();
}
public void tearDown() {
try {
if (index1 != null) {
index1.close();
}
if (index2 != null) {
index2.close();
}
if (store1 != null) {
store1.close();
}
if (store2 != null) {
store2.close();
}
if (catalog != null) {
catalog.close();
}
if (env != null) {
env.close();
}
} catch (Exception e) {
System.out.println("Ignored exception during tearDown: " + e);
} finally {
/* Ensure that GC can cleanup. */
index1 = null;
index2 = null;
store1 = null;
store2 = null;
catalog = null;
env = null;
testEnv = null;
runner = null;
factory = null;
storeMap1 = null;
storeMap2 = null;
indexMap1 = null;
indexMap2 = null;
}
}
public void runTest()
throws Exception {
runner.run(this);
}
public void doWork()
throws Exception {
createViews();
writeAndRead();
}
private void createDatabase()
throws Exception {
catalog = new StoredClassCatalog(openDb("catalog.db"));
factory = new TupleSerialFactory(catalog);
assertSame(catalog, factory.getCatalog());
store1 = openDb("store1.db");
store2 = openDb("store2.db");
index1 = openSecondaryDb(factory, "1", store1, "index1.db", null);
index2 = openSecondaryDb(factory, "2", store2, "index2.db", store1);
}
private Database openDb(String file)
throws Exception {
DatabaseConfig config = new DatabaseConfig();
config.setTransactional(testEnv.isTxnMode());
config.setAllowCreate(true);
return DbCompat.openDatabase(env, null, file, null, config);
}
private SecondaryDatabase openSecondaryDb(TupleSerialFactory factory,
String keyName,
Database primary,
String file,
Database foreignStore)
throws Exception {
SecondaryConfig secConfig = new SecondaryConfig();
secConfig.setTransactional(testEnv.isTxnMode());
secConfig.setAllowCreate(true);
secConfig.setKeyCreator(factory.getKeyCreator(MarshalledObject.class,
keyName));
if (foreignStore != null) {
secConfig.setForeignKeyDatabase(foreignStore);
secConfig.setForeignKeyDeleteAction(
ForeignKeyDeleteAction.CASCADE);
}
return DbCompat.openSecondaryDatabase(env, null,
file, null,
primary, secConfig);
}
private void createViews()
throws Exception {
if (isSorted) {
storeMap1 = factory.newSortedMap(store1, String.class,
MarshalledObject.class, true);
storeMap2 = factory.newSortedMap(store2, String.class,
MarshalledObject.class, true);
indexMap1 = factory.newSortedMap(index1, String.class,
MarshalledObject.class, true);
indexMap2 = factory.newSortedMap(index2, String.class,
MarshalledObject.class, true);
} else {
storeMap1 = factory.newMap(store1, String.class,
MarshalledObject.class, true);
storeMap2 = factory.newMap(store2, String.class,
MarshalledObject.class, true);
indexMap1 = factory.newMap(index1, String.class,
MarshalledObject.class, true);
indexMap2 = factory.newMap(index2, String.class,
MarshalledObject.class, true);
}
}
private void writeAndRead()
throws Exception {
MarshalledObject o1 = new MarshalledObject("data1", "pk1", "ik1", "");
assertNull(storeMap1.put(null, o1));
assertEquals(o1, storeMap1.get("pk1"));
assertEquals(o1, indexMap1.get("ik1"));
MarshalledObject o2 = new MarshalledObject("data2", "pk2", "", "pk1");
assertNull(storeMap2.put(null, o2));
assertEquals(o2, storeMap2.get("pk2"));
assertEquals(o2, indexMap2.get("pk1"));
/*
* store1 contains o1 with primary key "pk1" and index key "ik1"
* store2 contains o2 with primary key "pk2" and foreign key "pk1"
* which is the primary key of store1
*/
storeMap1.remove("pk1");
assertNull(storeMap1.get("pk1"));
assertNull(indexMap1.get("ik1"));
assertNull(storeMap2.get("pk2"));
assertNull(indexMap2.get("pk1"));
}
}