X-Git-Url: http://git.argeo.org/?a=blobdiff_plain;ds=sidebyside;f=org.argeo.jcr%2Fsrc%2Forg%2Fargeo%2Fjcr%2Funit%2FAbstractJcrTestCase.java;fp=org.argeo.jcr%2Fsrc%2Forg%2Fargeo%2Fjcr%2Funit%2FAbstractJcrTestCase.java;h=1269a3ee5730243b172896b546ff6191e3974dcc;hb=e66b9893b0e511f8ab295e3cee42b7dc966f1597;hp=0000000000000000000000000000000000000000;hpb=8260f4470f514ea347ca53f5b4dfc632c4a4de66;p=lgpl%2Fargeo-commons.git

diff --git a/org.argeo.jcr/src/org/argeo/jcr/unit/AbstractJcrTestCase.java b/org.argeo.jcr/src/org/argeo/jcr/unit/AbstractJcrTestCase.java
new file mode 100644
index 000000000..1269a3ee5
--- /dev/null
+++ b/org.argeo.jcr/src/org/argeo/jcr/unit/AbstractJcrTestCase.java
@@ -0,0 +1,133 @@
+/*
+ * Copyright (C) 2007-2012 Argeo GmbH
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *         http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+package org.argeo.jcr.unit;
+
+import java.io.File;
+import java.security.AccessController;
+import java.security.PrivilegedAction;
+
+import javax.jcr.Repository;
+import javax.jcr.Session;
+import javax.jcr.SimpleCredentials;
+import javax.security.auth.Subject;
+import javax.security.auth.login.LoginContext;
+import javax.security.auth.login.LoginException;
+
+import org.apache.commons.io.FileUtils;
+import org.apache.commons.logging.Log;
+import org.apache.commons.logging.LogFactory;
+import org.argeo.jcr.ArgeoJcrException;
+
+import junit.framework.TestCase;
+
+public abstract class AbstractJcrTestCase extends TestCase {
+	private final static Log log = LogFactory.getLog(AbstractJcrTestCase.class);
+
+	private Repository repository;
+	private Session session = null;
+
+	public final static String LOGIN_CONTEXT_TEST_SYSTEM = "TEST_JACKRABBIT_ADMIN";
+
+	// protected abstract File getRepositoryFile() throws Exception;
+
+	protected abstract Repository createRepository() throws Exception;
+
+	protected abstract void clearRepository(Repository repository)
+			throws Exception;
+
+	@Override
+	protected void setUp() throws Exception {
+		File homeDir = getHomeDir();
+		FileUtils.deleteDirectory(homeDir);
+		repository = createRepository();
+	}
+
+	@Override
+	protected void tearDown() throws Exception {
+		if (session != null) {
+			session.logout();
+			if (log.isTraceEnabled())
+				log.trace("Logout session");
+		}
+		clearRepository(repository);
+	}
+
+	protected Session session() {
+		if (session != null && session.isLive())
+			return session;
+		Session session;
+		if (getLoginContext() != null) {
+			LoginContext lc;
+			try {
+				lc = new LoginContext(getLoginContext());
+				lc.login();
+			} catch (LoginException e) {
+				throw new ArgeoJcrException("JAAS login failed", e);
+			}
+			session = Subject.doAs(lc.getSubject(),
+					new PrivilegedAction<Session>() {
+
+						@Override
+						public Session run() {
+							return login();
+						}
+
+					});
+		} else
+			session = login();
+		this.session = session;
+		return this.session;
+	}
+
+	protected String getLoginContext() {
+		return null;
+	}
+
+	protected Session login() {
+		try {
+			if (log.isTraceEnabled())
+				log.trace("Login session");
+			Subject subject = Subject.getSubject(AccessController.getContext());
+			if (subject != null)
+				return getRepository().login();
+			else
+				return getRepository().login(
+						new SimpleCredentials("demo", "demo".toCharArray()));
+		} catch (Exception e) {
+			throw new ArgeoJcrException("Cannot login to repository", e);
+		}
+	}
+
+	protected Repository getRepository() {
+		return repository;
+	}
+
+	/**
+	 * enables children class to set an existing repository in case it is not
+	 * deleted on startup, to test migration by instance
+	 */
+	public void setRepository(Repository repository) {
+		this.repository = repository;
+	}
+
+	protected File getHomeDir() {
+		File homeDir = new File(System.getProperty("java.io.tmpdir"),
+				AbstractJcrTestCase.class.getSimpleName() + "-"
+						+ System.getProperty("user.name"));
+		return homeDir;
+	}
+
+}