LowLevelText.java example

Explorer
exist-master
package org.exist.storage;

import org.exist.EXistException;
import org.exist.security.PermissionDeniedException;
import org.exist.source.StringSource;
import org.exist.test.ExistEmbeddedServer;
import org.exist.util.DatabaseConfigurationException;
import org.exist.xquery.CompiledXQuery;
import org.exist.xquery.XPathException;
import org.exist.xquery.XQuery;
import org.exist.xquery.XQueryContext;
import org.junit.After;
import org.junit.Before;
import org.junit.ClassRule;
import org.junit.Test;

import java.io.IOException;
import java.util.Optional;

import static org.junit.Assert.assertEquals;
import static org.junit.Assert.assertNotNull;

/**
 * Currently, tests for the {@link org.exist.storage.XQueryPool}
 * with the {@link StringSource}. 
 */
public class LowLevelText {

	private static final String TEST_XQUERY_SOURCE = "/test";

	@ClassRule
	public static final ExistEmbeddedServer existEmbeddedServer = new ExistEmbeddedServer(true, false);

	private DBBroker broker;

	private XQueryPool xqueryPool;

	private StringSource stringSource;

	private CompiledXQuery preCompiledXQuery;

	@Before
	public void setUp() throws DatabaseConfigurationException, EXistException, XPathException, PermissionDeniedException, IOException {
		final BrokerPool pool = existEmbeddedServer.getBrokerPool();
		broker = pool.get(Optional.of(pool.getSecurityManager().getSystemSubject()));
		xqueryPool = pool.getXQueryPool();
		stringSource = new StringSource(TEST_XQUERY_SOURCE);

		final XQuery xquery = pool.getXQueryService();
		final XQueryContext context = new XQueryContext(broker.getBrokerPool());
		preCompiledXQuery = xquery.compile(broker, context, stringSource);
	}

	@After
	public void tearDown() {
		if(broker != null) {
			broker.close();
		}
	}

	@Test
	public void borrowCompiledXQuery1() throws PermissionDeniedException {
		// put the preCompiledXQuery in cache - NOTE: returnCompiledXQuery() is not a good name
		xqueryPool.returnCompiledXQuery(stringSource, preCompiledXQuery);
		callAndTestBorrowCompiledXQuery(stringSource);
	}

	@Test
	public void borrowCompiledXQuery2() throws PermissionDeniedException {
		xqueryPool.returnCompiledXQuery(stringSource, preCompiledXQuery);

		callAndTestBorrowCompiledXQuery(stringSource);
		callAndTestBorrowCompiledXQuery(stringSource);
	}

	@Test
	public void borrowCompiledXQuery3() throws PermissionDeniedException {
		xqueryPool.returnCompiledXQuery(stringSource, preCompiledXQuery);

		callAndTestBorrowCompiledXQuery(stringSource);
		callAndTestBorrowCompiledXQuery(stringSource);
		callAndTestBorrowCompiledXQuery(stringSource);
	}

	/**
	 * test with a new StringSource object having same content
	 */
	@Test
	public void borrowCompiledXQueryNewStringSource() throws PermissionDeniedException {
		xqueryPool.returnCompiledXQuery(stringSource, preCompiledXQuery);
		StringSource localStringSource = new StringSource(TEST_XQUERY_SOURCE);

		callAndTestBorrowCompiledXQuery(localStringSource);
	}

	/**
	 * test with a new StringSource object having same content
	 */
	@Test
	public void borrowCompiledXQueryNewStringSource2() throws PermissionDeniedException {
		xqueryPool.returnCompiledXQuery(stringSource, preCompiledXQuery);
		StringSource localStringSource = new StringSource(TEST_XQUERY_SOURCE);

		callAndTestBorrowCompiledXQuery(localStringSource);
		callAndTestBorrowCompiledXQuery(localStringSource);
	}

	private void callAndTestBorrowCompiledXQuery(StringSource stringSourceArg) throws PermissionDeniedException {
		final CompiledXQuery compiledXQuery = xqueryPool.borrowCompiledXQuery(broker, stringSourceArg);
		assertNotNull(
				"borrowCompiledXQuery should retrieve something for this stringSource",
				compiledXQuery);
		assertEquals(
				"borrowCompiledXQuery should retrieve the preCompiled XQuery for this stringSource",
				compiledXQuery, preCompiledXQuery);
		xqueryPool.returnCompiledXQuery(stringSourceArg, compiledXQuery);
	}
}