org.tentackle.pdo.testng.AbstractPdoTest Maven / Gradle / Ivy
Show all versions of tentackle-test-pdo Show documentation
/*
* Tentackle - https://tentackle.org.
*
* This library 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.1 of the License, or (at your option) any later version.
*
* This library 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 library; if not, write to the Free Software
* Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
*
*/
package org.tentackle.pdo.testng;
import org.testng.ITestResult;
import org.testng.Reporter;
import org.testng.SkipException;
import org.testng.annotations.AfterClass;
import org.testng.annotations.AfterMethod;
import org.testng.annotations.AfterSuite;
import org.testng.annotations.BeforeClass;
import org.testng.annotations.BeforeMethod;
import org.testng.annotations.BeforeSuite;
import org.tentackle.dbms.CommitTxRunnable;
import org.tentackle.dbms.Db;
import org.tentackle.dbms.RollbackTxRunnable;
import org.tentackle.model.Model;
import org.tentackle.model.ModelException;
import org.tentackle.pdo.DomainContext;
import org.tentackle.pdo.DomainContextProvider;
import org.tentackle.pdo.Pdo;
import org.tentackle.pdo.test.DbTestUtilities;
import org.tentackle.script.ScriptFactory;
import org.tentackle.script.ScriptingLanguage;
import org.tentackle.session.ModificationTracker;
import org.tentackle.session.PersistenceException;
import org.tentackle.session.Session;
import org.tentackle.sql.Backend;
import java.io.BufferedReader;
import java.io.File;
import java.io.IOException;
import java.io.InputStreamReader;
import java.io.Reader;
import java.util.Arrays;
import java.util.Collection;
import java.util.Set;
import static org.testng.Assert.fail;
/**
* Base class for TestNG tests on PDO level.
*
* @author harald
*/
public abstract class AbstractPdoTest implements DomainContextProvider {
/**
* The transaction handling for the test class.
*/
public enum TransactionType {
/** transaction per test class. */
CLASS,
/** transaction per test method. */
METHOD,
/** no automatic transaction handling. */
NONE
}
private static Session session; // session for all methods in all test classes
private static DomainContext context; // domain context for all methods in all test classes
private final TransactionType txType; // transaction handling
private final boolean commit; // true if commit transaction, else rollback (default)
private boolean rollbackLogged; // true to log the rollback, even if test succeeds
private long txVoucher; // transaction voucher for current test class
/**
* Creates a test.
*
* @param txType the transaction type
* @param commit true if commit transaction, else rollback (default for regular tests)
*/
public AbstractPdoTest(TransactionType txType, boolean commit) {
this.txType = txType;
this.commit = commit;
}
/**
* Creates a test with rollback.
*
* @param txType the transaction type
*/
public AbstractPdoTest(TransactionType txType) {
this(txType, false);
}
/**
* Creates a test with rollback and one transaction per class.
*/
public AbstractPdoTest() {
this(TransactionType.CLASS, false); // per class
}
/**
* Returns whether statements rolled back are always logged.
*
* @return true to log even if test succeeds, false is default
*/
public boolean isRollbackLogged() {
return rollbackLogged;
}
/**
* Sets whether to log the statements rolled back.
* By default, the statements are not logged if the test succeeds.
* Sometimes, however, it is nice to log the statements that were executed during the test.
*
* Notice that the flag is reset to false when the transaction ends.
*
* @param rollbackLogged true to log the statements, default is false
*/
public void setRollbackLogged(boolean rollbackLogged) {
this.rollbackLogged = rollbackLogged;
}
/**
* Opens the sessions and starts the modification tracker.
* If the database is in-memory, it will be populated before the tracker is started.
*
* @throws Exception if failed
* @see #createDatabaseTables(Db)
*/
@BeforeSuite(alwaysRun = true)
public void openSessionsAndStartModificationTracker() throws Exception {
session = openSession();
context = createDomainContext();
boolean populate = false;
if (session instanceof Db db) {
Backend backend = db.getBackend();
if (backend.isDatabaseInMemory(session.getUrl())) {
createDatabaseTables(db);
populate = true;
}
}
// set the default scripting language, if not already set and exactly one provided
if (ScriptFactory.getInstance().getDefaultLanguage() == null) {
Set languages = ScriptFactory.getInstance().getLanguages();
if (languages.size() == 1) {
ScriptFactory.getInstance().setDefaultLanguage(languages.iterator().next());
}
}
ModificationTracker tracker = ModificationTracker.getInstance();
tracker.setSession(session.clone(null));
tracker.setSleepInterval(500); // fast polling for tests
tracker.start();
if (populate) {
populateDatabase();
}
}
/**
* Terminates the modification tracker and closes the sessions.
*
* @throws Exception if failed
*/
@AfterSuite(alwaysRun = true)
public void closeSessionsAndTerminateModificationTracker() throws Exception {
if (session != null) {
ModificationTracker.getInstance().terminate();
session.close();
}
}
/**
* Begins the transaction if type is {@link TransactionType#CLASS}.
*
* @throws Exception if failed
*/
@BeforeClass(alwaysRun = true)
public void beforeClass() throws Exception {
if (txType == TransactionType.CLASS) {
beginTransaction();
}
}
/**
* Ends the transaction if type is {@link TransactionType#CLASS}.
*
* @throws Exception if failed
*/
@AfterClass(alwaysRun = true)
public void afterClass() throws Exception {
if (txType == TransactionType.CLASS) {
endTransaction();
}
}
/**
* Begins the transaction if type is {@link TransactionType#METHOD}.
*
* @throws Exception if failed
*/
@BeforeMethod(alwaysRun = true)
public void beforeMethod() throws Exception {
if (txType == TransactionType.METHOD) {
beginTransaction();
}
}
/**
* Ends the transaction if type is {@link TransactionType#METHOD}.
*
* @throws Exception if failed
*/
@AfterMethod(alwaysRun = true)
public void afterMethod(ITestResult result) throws Exception {
if (!result.isSuccess()) {
rollbackLogged = true; // log statements rolled back, if test failed
}
if (txType == TransactionType.METHOD) {
endTransaction();
}
}
/**
* Gets the session.
*
* @return the session
*/
public Session getSession() {
return session;
}
/**
* Returns whether the transaction should be committed or rolled back.
*
* @return true if commit, false if rollback (default for regular tests)
*/
public boolean isCommit() {
return commit;
}
/**
* Gets the domain context.
*
* @return the domain context
*/
@Override
public DomainContext getDomainContext() {
return context;
}
/**
* Runs the given class in another JVM.
* The test class must have a main method.
*
* @param testClass the test class
* @return the exit value
* @throws IOException if some IO operation failed
*/
public int runInOtherJVM(Class testClass) throws IOException {
Process process = runClass(testClass);
waitForProcess(process);
return process.exitValue();
}
/**
* Opens the session.
*
* @return the thread-local session
*/
protected Session openSession() {
Session s;
try {
s = Pdo.createSession();
}
catch (PersistenceException ex) {
// no database? wrong database? whatever: testing environment incomplete
throw new SkipException("no backend found -> no tests", ex);
}
s.makeCurrent();
return s;
}
/**
* Creates the database tables.
* The default implementation loads the model from the classpath.
* Override this method to provide additional model sources.
*
* @param db the low-level database session
* @see Model#loadFromResources(boolean)
*/
protected void createDatabaseTables(Db db) {
try {
String script = DbTestUtilities.getInstance().createPopulateScript(db);
DbTestUtilities.getInstance().runScript(db, script);
}
catch (ModelException e) {
fail("populating the database failed", e);
}
}
/**
* Populates the database with test data.
*/
protected void populateDatabase() {
// default does nothing
}
/**
* Creates the domain context.
*
* @return the context (usually thread-local)
*/
protected DomainContext createDomainContext() {
return Pdo.createDomainContext(); // thread-local
}
/**
* Begins a transaction.
*/
protected void beginTransaction() {
if (session != null) {
txVoucher = session.begin("test");
}
}
/**
* Commits or rolls back a transaction.
*/
protected void endTransaction() {
if (session != null && session.isTxRunning()) {
if (commit) {
session.commit(txVoucher);
}
else {
if (rollbackLogged) {
session.rollback(txVoucher);
}
else {
// don't log rollback if test succeeds
session.rollbackSilently(txVoucher);
}
}
txVoucher = 0;
rollbackLogged = false;
}
}
/**
* Invokes the {@link CommitTxRunnable}s, if any.
* Useful for special test scenarios.
* Must be invoked explicitly from within the test.
*/
protected void executeCommitTxRunnables() {
if (session instanceof Db db) {
Collection runnables = db.getCommitTxRunnables();
if (runnables != null) {
runnables.forEach(runnable -> runnable.commit(db));
}
}
}
/**
* Invokes the {@link RollbackTxRunnable}s, if any.
* Useful for special test scenarios.
* Must be invoked explicitly from within the test.
*/
protected void executeRollbackTxRunnables() {
if (session instanceof Db db) {
Collection runnables = db.getRollbackTxRunnables();
if (runnables != null) {
runnables.forEach(runnable -> runnable.rollback(db));
}
}
}
/**
* Runs the given class in another JVM and waits for termination.
* The test class must have a main method.
*
* @param testClass the test class
* @param args optional arguments
* @return the process object
* @throws IOException if some IO operation failed
*/
public static Process runClass(Class testClass, String... args) throws IOException {
// add node and key in other jvm
String javaHome = System.getProperty("java.home");
String classPath = System.getProperty("java.class.path");
String modulePath = System.getProperty("jdk.module.path");
if (modulePath != null) {
if (!classPath.endsWith(":")) {
classPath += ":";
}
classPath += modulePath; // just append the module path to the classpath
}
String[] cmd = {
javaHome + File.separator + "bin" + File.separator + "java",
"-cp",
classPath,
testClass.getName()
};
if (args != null && args.length > 0) {
String[] aCmd = Arrays.copyOf(cmd, cmd.length + args.length);
System.arraycopy(args, 0, aCmd, cmd.length, args.length);
cmd = aCmd;
}
Reporter.log("running: " + Arrays.toString(cmd) + "
");
return Runtime.getRuntime().exec(cmd);
}
/**
* Waits for process to terminate and write stdout and stderr to the reporter log.
*
* @param process the process
* @throws IOException if some IO failed
*/
public static void waitForProcess(Process process) throws IOException {
try {
process.waitFor();
}
catch (InterruptedException ex) {
fail("executing " + process + " failed
", ex);
}
// collect stdout
Reader r = new InputStreamReader(process.getInputStream());
try (BufferedReader in = new BufferedReader(r)) {
String line;
while ((line = in.readLine()) != null) {
Reporter.log(line + "
");
}
}
// collect stderr
r = new InputStreamReader(process.getErrorStream());
try (BufferedReader in = new BufferedReader(r)) {
String line;
while ((line = in.readLine()) != null) {
Reporter.log(line + "
");
}
}
}
}