libdb/test/csharp/HeapDatabaseTest.cs
2011-09-13 13:44:24 -04:00

427 lines
15 KiB
C#

/*-
* See the file LICENSE for redistribution information.
*
* Copyright (c) 2011 Oracle and/or its affiliates. All rights reserved.
*
*/
using System;
using System.Collections.Generic;
using System.IO;
using System.Text;
using System.Xml;
using NUnit.Framework;
using BerkeleyDB;
namespace CsharpAPITest {
[TestFixture]
public class HeapDatabaseTest : DatabaseTest {
[TestFixtureSetUp]
public void SetUpTestFixture() {
testFixtureName = "HeapDatabaseTest";
base.SetUpTestfixture();
}
[Test]
public void TestAppendWithoutTxn() {
testName = "TestAppendWithoutTxn";
SetUpTest(true);
string heapDBFileName = testHome + "/" + testName + ".db";
HeapDatabaseConfig heapConfig = new HeapDatabaseConfig();
heapConfig.Creation = CreatePolicy.ALWAYS;
HeapDatabase heapDB = HeapDatabase.Open(
heapDBFileName, heapConfig);
byte[] byteArr = new byte[4];
byteArr = BitConverter.GetBytes((int)1);
DatabaseEntry data = new DatabaseEntry(byteArr);
HeapRecordId rid = heapDB.Append(data);
// Confirm that the record lives on a page larger than 0.
Assert.AreNotEqual(0, rid.PageNumber);
// Confirm that the record exists in the database.
DatabaseEntry key = new DatabaseEntry(rid.toArray());
Assert.IsTrue(heapDB.Exists(key));
heapDB.Close();
}
[Test]
public void TestAppendWithTxn() {
testName = "TestAppendWithTxn";
SetUpTest(true);
string heapDBFileName = testHome + "/" + testName + ".db";
string heapDBName =
Path.GetFileNameWithoutExtension(heapDBFileName);
DatabaseEnvironmentConfig envConfig =
new DatabaseEnvironmentConfig();
envConfig.Create = true;
envConfig.UseTxns = true;
envConfig.UseMPool = true;
DatabaseEnvironment env = DatabaseEnvironment.Open(
testHome, envConfig);
Transaction txn = env.BeginTransaction();
HeapDatabaseConfig heapConfig = new HeapDatabaseConfig();
heapConfig.Creation = CreatePolicy.ALWAYS;
heapConfig.Env = env;
/* If environmnet home is set, the file name in
* Open() is the relative path.
*/
HeapDatabase heapDB = HeapDatabase.Open(
heapDBName, heapConfig, txn);
DatabaseEntry data;
int i = 1000;
try {
while (i > 0) {
data = new DatabaseEntry(
BitConverter.GetBytes(i));
heapDB.Append(data, txn);
i--;
}
txn.Commit();
} catch {
txn.Abort();
} finally {
heapDB.Close();
env.Close();
}
}
[Test]
public void TestCursor() {
testName = "TestCursor";
SetUpTest(true);
GetCursur(testHome + "/" + testName + ".db", false);
}
[Test]
public void TestCursorWithConfig() {
testName = "TestCursorWithConfig";
SetUpTest(true);
GetCursur(testHome + "/" + testName + ".db", true);
}
public void GetCursur(string dbFileName, bool ifConfig) {
HeapDatabaseConfig dbConfig = new HeapDatabaseConfig();
dbConfig.Creation = CreatePolicy.IF_NEEDED;
HeapDatabase db = HeapDatabase.Open(dbFileName, dbConfig);
HeapRecordId rid = db.Append(
new DatabaseEntry(BitConverter.GetBytes((int)1)));
Cursor cursor;
CursorConfig cursorConfig = new CursorConfig();
cursorConfig.Priority = CachePriority.HIGH;
if (ifConfig == false)
cursor = db.Cursor();
else
cursor = db.Cursor(cursorConfig);
cursor.Add(new KeyValuePair<DatabaseEntry, DatabaseEntry>(
new DatabaseEntry(rid.toArray()),
new DatabaseEntry(BitConverter.GetBytes((int)2))));
Cursor dupCursor = cursor.Duplicate(false);
Assert.IsNull(dupCursor.Current.Key);
if (ifConfig)
Assert.AreEqual(CachePriority.HIGH, dupCursor.Priority);
dupCursor.Close();
cursor.Close();
db.Close();
}
[Test]
public void TestExists() {
testName = "TestExists";
SetUpTest(true);
DatabaseEnvironmentConfig envConfig =
new DatabaseEnvironmentConfig();
envConfig.Create = true;
envConfig.UseLocking = true;
envConfig.UseLogging = true;
envConfig.UseMPool = true;
envConfig.UseTxns = true;
DatabaseEnvironment env = DatabaseEnvironment.Open(
testHome, envConfig);
HeapDatabase db;
try {
Transaction openTxn = env.BeginTransaction();
try {
HeapDatabaseConfig heapConfig =
new HeapDatabaseConfig();
heapConfig.Creation = CreatePolicy.IF_NEEDED;
heapConfig.Env = env;
db = HeapDatabase.Open(testName + ".db",
heapConfig, openTxn);
openTxn.Commit();
} catch (DatabaseException e) {
openTxn.Abort();
throw e;
}
Transaction cursorTxn = env.BeginTransaction();
Cursor cursor;
HeapRecordId rid;
try {
/*
* Put a record into heap database with
* cursor and abort the operation.
*/
rid = db.Append(new DatabaseEntry(
ASCIIEncoding.ASCII.GetBytes("foo")), cursorTxn);
cursor = db.Cursor(cursorTxn);
KeyValuePair<DatabaseEntry, DatabaseEntry> pair;
pair = new KeyValuePair<DatabaseEntry, DatabaseEntry>(
new DatabaseEntry(rid.toArray()),
new DatabaseEntry(ASCIIEncoding.ASCII.GetBytes("bar")));
cursor.Add(pair);
cursor.Close();
Assert.IsTrue(db.Exists(
new DatabaseEntry(rid.toArray()), cursorTxn));
cursorTxn.Abort();
} catch (DatabaseException e) {
cursorTxn.Abort();
db.Close();
throw e;
}
Transaction delTxn = env.BeginTransaction();
try {
/*
* The put operation is aborted in the heap
* database so querying if the record still exists
* throws KeyEmptyException.
*/
Assert.IsFalse(db.Exists(
new DatabaseEntry(rid.toArray()), delTxn));
delTxn.Commit();
} catch (DatabaseException e) {
delTxn.Abort();
throw e;
} finally {
db.Close();
}
} finally {
env.Close();
}
}
[Test]
public void TestOpenExistingHeapDB() {
testName = "TestOpenExistingHeapDB";
SetUpTest(true);
string heapDBFileName = testHome + "/" + testName + ".db";
HeapDatabaseConfig heapConfig = new HeapDatabaseConfig();
heapConfig.Creation = CreatePolicy.ALWAYS;
HeapDatabase heapDB = HeapDatabase.Open(
heapDBFileName, heapConfig);
heapDB.Close();
DatabaseConfig dbConfig = new DatabaseConfig();
Database db = Database.Open(heapDBFileName, dbConfig);
Assert.AreEqual(db.Type, DatabaseType.HEAP);
db.Close();
}
[Test]
public void TestOpenNewHeapDB() {
testName = "TestOpenNewHeapDB";
SetUpTest(true);
string heapDBFileName = testHome + "/" + testName + ".db";
// Configure all fields/properties in heap database.
XmlElement xmlElem = Configuration.TestSetUp(
testFixtureName, testName);
HeapDatabaseConfig heapConfig = new HeapDatabaseConfig();
HeapDatabaseConfigTest.Config(xmlElem, ref heapConfig, true);
heapConfig.Feedback = new DatabaseFeedbackDelegate(DbFeedback);
// Open the heap database with above configuration.
HeapDatabase heapDB = HeapDatabase.Open(
heapDBFileName, heapConfig);
// Check the fields/properties in opened heap database.
Confirm(xmlElem, heapDB, true);
heapDB.Close();
}
private void DbFeedback(DatabaseFeedbackEvent opcode, int percent) {
if (opcode == DatabaseFeedbackEvent.UPGRADE)
Console.WriteLine("Update for %d%", percent);
if (opcode == DatabaseFeedbackEvent.VERIFY)
Console.WriteLine("Vertify for %d", percent);
}
[Test]
public void TestPutHeap() {
KeyValuePair<DatabaseEntry, DatabaseEntry> pair;
testName = "TestPutHeap";
SetUpTest(true);
string heapDBFileName = testHome + "/" +
testName + ".db";
HeapDatabaseConfig heapConfig =
new HeapDatabaseConfig();
heapConfig.Creation = CreatePolicy.ALWAYS;
using (HeapDatabase heapDB = HeapDatabase.Open(
heapDBFileName, heapConfig)) {
DatabaseEntry data = new DatabaseEntry(
BitConverter.GetBytes((int)1));
HeapRecordId rid = heapDB.Append(data);
DatabaseEntry key = new DatabaseEntry(rid.toArray());
data = new DatabaseEntry(BitConverter.GetBytes((int)2));
heapDB.Put(key, data);
pair = heapDB.GetBoth(key, data);
}
}
[Test]
public void TestStats() {
testName = "TestStats";
SetUpTest(true);
string dbFileName = testHome + "/" +
testName + ".db";
HeapDatabaseConfig dbConfig =
new HeapDatabaseConfig();
ConfigCase1(dbConfig);
HeapDatabase db = HeapDatabase.Open(dbFileName, dbConfig);
HeapStats stats = db.Stats();
ConfirmStatsPart1Case1(stats);
db.PrintFastStats(true);
// Put 500 records into the database.
PutRecordCase1(db, null);
stats = db.Stats();
ConfirmStatsPart2Case1(stats);
db.PrintFastStats();
db.Close();
}
[Test]
public void TestStatsInTxn() {
testName = "TestStatsInTxn";
SetUpTest(true);
StatsInTxn(testHome, testName, false);
}
[Test]
public void TestStatsWithIsolation() {
testName = "TestStatsWithIsolation";
SetUpTest(true);
StatsInTxn(testHome, testName, true);
}
public void StatsInTxn(string home, string name, bool ifIsolation) {
DatabaseEnvironmentConfig envConfig =
new DatabaseEnvironmentConfig();
EnvConfigCase1(envConfig);
DatabaseEnvironment env = DatabaseEnvironment.Open(
home, envConfig);
Transaction openTxn = env.BeginTransaction();
HeapDatabaseConfig dbConfig =
new HeapDatabaseConfig();
ConfigCase1(dbConfig);
dbConfig.Env = env;
HeapDatabase db = HeapDatabase.Open(name + ".db",
dbConfig, openTxn);
openTxn.Commit();
Transaction statsTxn = env.BeginTransaction();
HeapStats stats;
if (ifIsolation == false)
stats = db.Stats(statsTxn);
else
stats = db.Stats(statsTxn, Isolation.DEGREE_ONE);
ConfirmStatsPart1Case1(stats);
db.PrintStats(true);
// Put 500 records into the database.
PutRecordCase1(db, statsTxn);
if (ifIsolation == false)
stats = db.Stats(statsTxn);
else
stats = db.Stats(statsTxn, Isolation.DEGREE_TWO);
ConfirmStatsPart2Case1(stats);
db.PrintStats();
statsTxn.Commit();
db.Close();
env.Close();
}
public void EnvConfigCase1(DatabaseEnvironmentConfig cfg) {
cfg.Create = true;
cfg.UseTxns = true;
cfg.UseMPool = true;
cfg.UseLogging = true;
}
public void ConfigCase1(HeapDatabaseConfig dbConfig) {
dbConfig.Creation = CreatePolicy.IF_NEEDED;
dbConfig.PageSize = 4096;
}
public void PutRecordCase1(HeapDatabase db, Transaction txn) {
byte[] bigArray = new byte[4000];
HeapRecordId rid = new HeapRecordId(2, 0);
for (int i = 1; i <= 100; i++) {
if (txn == null)
rid = db.Append(
new DatabaseEntry(BitConverter.GetBytes(i)));
else
rid = db.Append(
new DatabaseEntry(BitConverter.GetBytes(i)), txn);
}
DatabaseEntry key = new DatabaseEntry(rid.toArray());
for (int i = 100; i <= 500; i++) {
if (txn == null)
db.Put(key, new DatabaseEntry(bigArray));
else
db.Put(key, new DatabaseEntry(bigArray), txn);
}
}
public void ConfirmStatsPart1Case1(HeapStats stats) {
Assert.AreNotEqual(0, stats.MagicNumber);
Assert.AreEqual(4096, stats.PageSize);
Assert.AreNotEqual(0, stats.Version);
}
public void ConfirmStatsPart2Case1(HeapStats stats) {
Assert.AreNotEqual(0, stats.nPages);
Assert.AreEqual(0, stats.MetadataFlags);
Assert.AreEqual(100, stats.nRecords);
Assert.AreNotEqual(0, stats.nRegions);
}
public static void Confirm(XmlElement xmlElem,
HeapDatabase heapDB, bool compulsory) {
DatabaseTest.Confirm(xmlElem, heapDB, compulsory);
// Confirm heap database specific field/property
Assert.AreEqual(DatabaseType.HEAP, heapDB.Type);
string type = heapDB.Type.ToString();
Assert.IsNotNull(type);
}
}
}