/*
* eXist Open Source Native XML Database
* Copyright (C) 2001-04 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.io.InputStream;
import java.util.Optional;
import org.exist.EXistException;
import org.exist.collections.Collection;
import org.exist.collections.triggers.TriggerException;
import org.exist.dom.persistent.BinaryDocument;
import org.exist.security.PermissionDeniedException;
import org.exist.storage.lock.Lock.LockMode;
import org.exist.storage.txn.TransactionManager;
import org.exist.storage.txn.Txn;
import org.exist.test.ExistEmbeddedServer;
import org.exist.test.TestConstants;
import org.exist.util.Configuration;
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.fail;
import static org.junit.Assert.assertEquals;
/**
* 0 byte binary files cannot be retrieved from database. This test
* displays the error.
*
* @author wessels
*/
public class ResourceTest {
private final static String EMPTY_BINARY_FILE="";
private final static XmldbURI DOCUMENT_NAME_URI = XmldbURI.create("empty.txt");
// we don't use @ClassRule/@Rule as we want to force corruption in some tests
private ExistEmbeddedServer existEmbeddedServer = new ExistEmbeddedServer(true, false);
protected BrokerPool startDB() throws DatabaseConfigurationException, EXistException {
final Configuration config = new Configuration();
BrokerPool.configure(1, 5, config);
return BrokerPool.getInstance();
}
@After
public void tearDown() {
BrokerPool.stopAll(false);
}
@Test
public void storeAndRead() throws TriggerException, PermissionDeniedException, DatabaseConfigurationException, IOException, LockException, EXistException {
BrokerPool.FORCE_CORRUPTION = true;
BrokerPool pool = startDb();
store(pool);
stopDb();
BrokerPool.FORCE_CORRUPTION = false;
pool = startDb();
read(pool);
}
private void store(final BrokerPool pool) throws EXistException, DatabaseConfigurationException, PermissionDeniedException, IOException, TriggerException, LockException {
final TransactionManager transact = pool.getTransactionManager();
try(final DBBroker broker = pool.get(Optional.of(pool.getSecurityManager().getSystemSubject()));
final Txn transaction = transact.beginTransaction()) {
final Collection collection = broker
.getOrCreateCollection(transaction, TestConstants.TEST_COLLECTION_URI);
broker.saveCollection(transaction, collection);
@SuppressWarnings("unused")
final BinaryDocument doc =
collection.addBinaryResource(transaction, broker,
DOCUMENT_NAME_URI , EMPTY_BINARY_FILE.getBytes(), "text/text");
transact.commit(transaction);
}
}
private void read(final BrokerPool pool) throws EXistException, DatabaseConfigurationException, PermissionDeniedException, IOException, LockException, TriggerException {
final TransactionManager transact = pool.getTransactionManager();
byte[] data = null;
try(final DBBroker broker = pool.get(Optional.of(pool.getSecurityManager().getSystemSubject()));
final Txn transaction = transact.beginTransaction();) {
final XmldbURI docPath = TestConstants.TEST_COLLECTION_URI.append(DOCUMENT_NAME_URI);
BinaryDocument binDoc = null;
try {
binDoc = (BinaryDocument) broker.getXMLResource(docPath, LockMode.READ_LOCK);
// if document is not present, null is returned
if(binDoc == null) {
fail("Binary document '" + docPath + " does not exist.");
} else {
try(final InputStream is = broker.getBinaryResource(binDoc)) {
data = new byte[(int) broker.getBinaryResourceSize(binDoc)];
is.read(data);
}
}
} finally {
if(binDoc != null) {
binDoc.getUpdateLock().release(LockMode.READ_LOCK);
}
}
final Collection collection = broker.getCollection(TestConstants.TEST_COLLECTION_URI);
collection.removeBinaryResource(transaction, broker, binDoc);
broker.saveCollection(transaction, collection);
transact.commit(transaction);
}
assertEquals(0, data.length);
}
@Test
public void store2() throws TriggerException, PermissionDeniedException, DatabaseConfigurationException, IOException, LockException, EXistException {
BrokerPool.FORCE_CORRUPTION = false;
final BrokerPool pool = startDb();
store(pool);
}
@Test
public void removeCollection() throws EXistException, DatabaseConfigurationException, PermissionDeniedException, IOException, TriggerException {
BrokerPool.FORCE_CORRUPTION = false;
final BrokerPool pool = startDb();
final TransactionManager transact = pool.getTransactionManager();
byte[] data = null;
try(final DBBroker broker = pool.get(Optional.of(pool.getSecurityManager().getSystemSubject()));
final Txn transaction = transact.beginTransaction()) {
final XmldbURI docPath = TestConstants.TEST_COLLECTION_URI.append(DOCUMENT_NAME_URI);
BinaryDocument binDoc = null;
try {
binDoc = (BinaryDocument) broker.getXMLResource(docPath, LockMode.READ_LOCK);
// if document is not present, null is returned
if(binDoc == null) {
fail("Binary document '" + docPath + " does not exist.");
} else {
try(final InputStream is = broker.getBinaryResource(binDoc)) {
data = new byte[(int) broker.getBinaryResourceSize(binDoc)];
is.read(data);
}
}
} finally {
if(binDoc != null) {
binDoc.getUpdateLock().release(LockMode.READ_LOCK);
}
}
final Collection collection = broker.getCollection(TestConstants.TEST_COLLECTION_URI);
broker.removeCollection(transaction, collection);
transact.commit(transaction);
}
assertEquals(0, data.length);
}
private BrokerPool startDb() throws EXistException, IOException, DatabaseConfigurationException {
existEmbeddedServer.startDb();
return existEmbeddedServer.getBrokerPool();
}
@After
public void stopDb() {
existEmbeddedServer.stopDb();
}
}