/*
* eXist Open Source Native XML Database
* Copyright (C) 2001-2014 The eXist Project
* http://exist-db.org
*
* This program is free software; you can redistribute it and/or
* modify it under the terms of the GNU Lesser General Public License
* as published by the Free Software Foundation; either version 2
* of the License, or (at your option) any later version.
*
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU Lesser General Public License for more details.
*
* You should have received a copy of the GNU Lesser General Public License
* along with this program; if not, write to the Free Software
* Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
*
* $Id$
*/
package org.exist.storage;
import java.io.IOException;
import java.nio.file.Path;
import java.util.Optional;
import org.exist.EXistException;
import org.exist.TestUtils;
import org.exist.collections.Collection;
import org.exist.collections.IndexInfo;
import org.exist.dom.persistent.DocumentImpl;
import org.exist.security.PermissionDeniedException;
import org.exist.storage.lock.Lock.LockMode;
import org.exist.storage.serializers.Serializer;
import org.exist.storage.txn.TransactionManager;
import org.exist.storage.txn.Txn;
import org.exist.test.ExistEmbeddedServer;
import org.exist.util.DatabaseConfigurationException;
import org.exist.util.LockException;
import org.exist.xmldb.XmldbURI;
import org.junit.After;
import org.junit.Test;
import static org.junit.Assert.assertNotNull;
import static org.junit.Assert.assertNull;
import org.xml.sax.InputSource;
import org.xml.sax.SAXException;
/**
* @author wolf
*
*/
public class CopyResourceTest {
// we don't use @ClassRule/@Rule as we want to force corruption in some tests
private ExistEmbeddedServer existEmbeddedServer = new ExistEmbeddedServer(true, false);
@Test
public void storeAndRead() throws PermissionDeniedException, DatabaseConfigurationException, IOException, LockException, SAXException, EXistException {
final String testCollectionName = "copyResource";
final String subCollection = "storeAndRead";
BrokerPool.FORCE_CORRUPTION = true;
BrokerPool pool = startDb();
store(pool, testCollectionName, subCollection);
stopDb();
BrokerPool.FORCE_CORRUPTION = false;
pool = startDb();
read(pool, testCollectionName);
}
@Test
public void storeAndReadAborted() throws PermissionDeniedException, DatabaseConfigurationException, IOException, LockException, SAXException, EXistException {
final String testCollectionName = "copyResource";
final String subCollection = "storeAndReadAborted";
BrokerPool.FORCE_CORRUPTION = true;
BrokerPool pool = startDb();
storeAborted(pool, testCollectionName, subCollection);
stopDb();
pool = startDb();
readAborted(pool, testCollectionName, subCollection);
}
private void store(final BrokerPool pool, final String testCollectionName, final String subCollection) throws EXistException, PermissionDeniedException, IOException, SAXException, LockException, DatabaseConfigurationException {
final TransactionManager transact = pool.getTransactionManager();
try(final DBBroker broker = pool.get(Optional.of(pool.getSecurityManager().getSystemSubject()))) {
Collection testCollection;
IndexInfo info;
try (final Txn transaction = transact.beginTransaction()) {
final Collection root = broker.getOrCreateCollection(transaction, XmldbURI.ROOT_COLLECTION_URI.append("test"));
assertNotNull(root);
broker.saveCollection(transaction, root);
testCollection = broker.getOrCreateCollection(transaction, XmldbURI.ROOT_COLLECTION_URI.append("test").append(testCollectionName));
assertNotNull(testCollection);
broker.saveCollection(transaction, testCollection);
final Collection subTestCollection = broker.getOrCreateCollection(transaction, XmldbURI.ROOT_COLLECTION_URI.append("test").append(testCollectionName).append(subCollection));
assertNotNull(subTestCollection);
broker.saveCollection(transaction, subTestCollection);
final Path f = TestUtils.resolveShakespeareSample("r_and_j.xml");
assertNotNull(f);
info = subTestCollection.validateXMLResource(transaction, broker, XmldbURI.create("test.xml"), new InputSource(f.toUri().toASCIIString()));
assertNotNull(info);
subTestCollection.store(transaction, broker, info, new InputSource(f.toUri().toASCIIString()));
transact.commit(transaction);
}
try (final Txn transaction = transact.beginTransaction()) {
broker.copyResource(transaction, info.getDocument(), testCollection, XmldbURI.create("new_test.xml"));
broker.saveCollection(transaction, testCollection);
transact.commit(transaction);
}
}
}
private void read(final BrokerPool pool, final String testCollectionName) throws EXistException, DatabaseConfigurationException, PermissionDeniedException, SAXException {
try(final DBBroker broker = pool.get(Optional.of(pool.getSecurityManager().getSystemSubject()))) {
final Serializer serializer = broker.getSerializer();
serializer.reset();
DocumentImpl doc = null;
try {
doc = broker.getXMLResource(XmldbURI.ROOT_COLLECTION_URI.append("test").append(testCollectionName).append("new_test.xml"), LockMode.READ_LOCK);
assertNotNull("Document should not be null", doc);
final String data = serializer.serialize(doc);
assertNotNull(data);
} finally {
if(doc != null) {
doc.getUpdateLock().release(LockMode.READ_LOCK);
}
}
}
}
private void storeAborted(final BrokerPool pool, final String testCollectionName, final String subCollection) throws EXistException, DatabaseConfigurationException, PermissionDeniedException, IOException, SAXException, LockException {
final TransactionManager transact = pool.getTransactionManager();
try(final DBBroker broker = pool.get(Optional.of(pool.getSecurityManager().getSystemSubject()))) {
Collection testCollection;
IndexInfo info;
try(final Txn transaction = transact.beginTransaction()) {
final Collection root = broker.getOrCreateCollection(transaction, XmldbURI.ROOT_COLLECTION_URI.append("test"));
assertNotNull(root);
broker.saveCollection(transaction, root);
testCollection = broker.getOrCreateCollection(transaction, XmldbURI.ROOT_COLLECTION_URI.append("test").append(testCollectionName));
assertNotNull(root);
broker.saveCollection(transaction, root);
final Collection subTestCollection = broker.getOrCreateCollection(transaction, XmldbURI.ROOT_COLLECTION_URI.append("test").append(testCollectionName).append(subCollection));
assertNotNull(subTestCollection);
broker.saveCollection(transaction, subTestCollection);
final Path f = TestUtils.resolveShakespeareSample("r_and_j.xml");
assertNotNull(f);
info = subTestCollection.validateXMLResource(transaction, broker, XmldbURI.create("test2.xml"), new InputSource(f.toUri().toASCIIString()));
assertNotNull(info);
subTestCollection.store(transaction, broker, info, new InputSource(f.toUri().toASCIIString()));
transact.commit(transaction);
}
final Txn transaction = transact.beginTransaction();
broker.copyResource(transaction, info.getDocument(), testCollection, XmldbURI.create("new_test2.xml"));
broker.saveCollection(transaction, testCollection);
//DO NOT COMMIT TRANSACTION
pool.getJournalManager().get().flush(true, false);
}
}
private void readAborted(final BrokerPool pool, final String testCollectionName, final String subCollection) throws EXistException, DatabaseConfigurationException, PermissionDeniedException, SAXException {
try(final DBBroker broker = pool.get(Optional.of(pool.getSecurityManager().getSystemSubject()))) {
final Serializer serializer = broker.getSerializer();
serializer.reset();
DocumentImpl doc = null;
try {
doc = broker.getXMLResource(XmldbURI.ROOT_COLLECTION_URI.append("test").append(testCollectionName).append(subCollection).append("test2.xml"), LockMode.READ_LOCK);
assertNotNull("Document should not be null", doc);
final String data = serializer.serialize(doc);
assertNotNull(data);
} finally {
if(doc != null) {
doc.getUpdateLock().release(LockMode.READ_LOCK);
}
}
doc = broker.getXMLResource(XmldbURI.ROOT_COLLECTION_URI.append("test").append(testCollectionName).append("new_test2.xml"), LockMode.READ_LOCK);
assertNull("Document should not exist", doc);
}
}
private BrokerPool startDb() throws EXistException, IOException, DatabaseConfigurationException {
existEmbeddedServer.startDb();
return existEmbeddedServer.getBrokerPool();
}
@After
public void stopDb() {
existEmbeddedServer.stopDb();
}
}