package org.exist.storage;
import static org.junit.Assert.*;
import java.io.File;
import org.exist.collections.*;
import org.exist.storage.txn.*;
import org.exist.util.Configuration;
import org.exist.xmldb.XmldbURI;
import org.junit.*;
import org.xml.sax.InputSource;
public class RemoveRootCollectionTest {
private BrokerPool pool;
private DBBroker broker;
private TransactionManager transact;
Collection root;
@Test public void removeEmptyRootCollection() throws Exception {
Txn transaction = transact.beginTransaction();
broker.removeCollection(transaction, root);
transact.commit(transaction);
assertEquals(0, root.getChildCollectionCount(broker));
assertEquals(0, root.getDocumentCount(broker));
}
@Test public void removeRootCollectionWithChildCollection() throws Exception {
addChildToRoot();
Txn transaction = transact.beginTransaction();
broker.removeCollection(transaction, root);
transact.commit(transaction);
assertEquals(0, root.getChildCollectionCount(broker));
assertEquals(0, root.getDocumentCount(broker));
}
@Ignore @Test public void removeRootCollectionWithDocument() throws Exception {
addDocumentToRoot();
Txn transaction = transact.beginTransaction();
broker.removeCollection(transaction, root);
transact.commit(transaction);
assertEquals(0, root.getChildCollectionCount(broker));
assertEquals(0, root.getDocumentCount(broker));
}
@Before public void startDB() throws Exception {
Configuration config = new Configuration();
BrokerPool.configure(1, 1, config);
pool = BrokerPool.getInstance(); assertNotNull(pool);
broker = pool.get(pool.getSecurityManager().getSystemSubject()); assertNotNull(broker);
transact = pool.getTransactionManager(); assertNotNull(transact);
root = broker.getCollection(XmldbURI.ROOT_COLLECTION_URI); assertNotNull(root);
}
@After public void stopDB() {
pool.release(broker);
BrokerPool.stopAll(false);
}
private void addDocumentToRoot() throws Exception {
Txn transaction = transact.beginTransaction();
InputSource is = new InputSource(new File("samples/shakespeare/hamlet.xml").toURI().toASCIIString());
assertNotNull(is);
IndexInfo info = root.validateXMLResource(transaction, broker, XmldbURI.create("hamlet.xml"), is);
assertNotNull(info);
root.store(transaction, broker, info, is, false);
transact.commit(transaction);
}
private void addChildToRoot() throws Exception {
Txn transaction = transact.beginTransaction();
Collection child = broker.getOrCreateCollection(transaction, XmldbURI.ROOT_COLLECTION_URI.append("child"));
broker.saveCollection(transaction, child);
transact.commit(transaction);
}
}