Browse Source

Concurrency and exception message refinements for test transactions

pull/1717/merge
Juergen Hoeller 7 years ago
parent
commit
a0cc80063d
  1. 8
      spring-test/src/main/java/org/springframework/test/context/transaction/AfterTransaction.java
  2. 8
      spring-test/src/main/java/org/springframework/test/context/transaction/BeforeTransaction.java
  3. 61
      spring-test/src/main/java/org/springframework/test/context/transaction/TestContextTransactionUtils.java
  4. 21
      spring-test/src/main/java/org/springframework/test/context/transaction/TestTransaction.java
  5. 24
      spring-test/src/main/java/org/springframework/test/context/transaction/TransactionContext.java
  6. 12
      spring-test/src/main/java/org/springframework/test/context/transaction/TransactionContextHolder.java
  7. 20
      spring-test/src/main/java/org/springframework/test/context/transaction/TransactionalTestExecutionListener.java
  8. 76
      spring-test/src/test/java/org/springframework/test/context/transaction/PrimaryTransactionManagerTests.java
  9. 194
      spring-test/src/test/java/org/springframework/test/context/transaction/TransactionalTestExecutionListenerTests.java

8
spring-test/src/main/java/org/springframework/test/context/transaction/AfterTransaction.java

@ -1,5 +1,5 @@
/* /*
* Copyright 2002-2016 the original author or authors. * Copyright 2002-2018 the original author or authors.
* *
* Licensed under the Apache License, Version 2.0 (the "License"); * Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License. * you may not use this file except in compliance with the License.
@ -28,15 +28,15 @@ import java.lang.annotation.Target;
* configured to run within a transaction via Spring's {@code @Transactional} * configured to run within a transaction via Spring's {@code @Transactional}
* annotation. * annotation.
* *
* <p>As of Spring Framework 4.3, {@code @AfterTransaction} may be declared on
* Java 8 based interface default methods.
*
* <p>{@code @AfterTransaction} methods declared in superclasses or as interface * <p>{@code @AfterTransaction} methods declared in superclasses or as interface
* default methods will be executed after those of the current test class. * default methods will be executed after those of the current test class.
* *
* <p>As of Spring Framework 4.0, this annotation may be used as a * <p>As of Spring Framework 4.0, this annotation may be used as a
* <em>meta-annotation</em> to create custom <em>composed annotations</em>. * <em>meta-annotation</em> to create custom <em>composed annotations</em>.
* *
* <p>As of Spring Framework 4.3, {@code @AfterTransaction} may also be
* declared on Java 8 based interface default methods.
*
* @author Sam Brannen * @author Sam Brannen
* @since 2.5 * @since 2.5
* @see org.springframework.transaction.annotation.Transactional * @see org.springframework.transaction.annotation.Transactional

8
spring-test/src/main/java/org/springframework/test/context/transaction/BeforeTransaction.java

@ -1,5 +1,5 @@
/* /*
* Copyright 2002-2016 the original author or authors. * Copyright 2002-2018 the original author or authors.
* *
* Licensed under the Apache License, Version 2.0 (the "License"); * Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License. * you may not use this file except in compliance with the License.
@ -28,15 +28,15 @@ import java.lang.annotation.Target;
* configured to run within a transaction via Spring's {@code @Transactional} * configured to run within a transaction via Spring's {@code @Transactional}
* annotation. * annotation.
* *
* <p>As of Spring Framework 4.3, {@code @BeforeTransaction} may be declared on
* Java 8 based interface default methods.
*
* <p>{@code @BeforeTransaction} methods declared in superclasses or as interface * <p>{@code @BeforeTransaction} methods declared in superclasses or as interface
* default methods will be executed before those of the current test class. * default methods will be executed before those of the current test class.
* *
* <p>As of Spring Framework 4.0, this annotation may be used as a * <p>As of Spring Framework 4.0, this annotation may be used as a
* <em>meta-annotation</em> to create custom <em>composed annotations</em>. * <em>meta-annotation</em> to create custom <em>composed annotations</em>.
* *
* <p>As of Spring Framework 4.3, {@code @BeforeTransaction} may also be
* declared on Java 8 based interface default methods.
*
* @author Sam Brannen * @author Sam Brannen
* @since 2.5 * @since 2.5
* @see org.springframework.transaction.annotation.Transactional * @see org.springframework.transaction.annotation.Transactional

61
spring-test/src/main/java/org/springframework/test/context/transaction/TestContextTransactionUtils.java

@ -1,5 +1,5 @@
/* /*
* Copyright 2002-2016 the original author or authors. * Copyright 2002-2018 the original author or authors.
* *
* Licensed under the Apache License, Version 2.0 (the "License"); * Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License. * you may not use this file except in compliance with the License.
@ -17,7 +17,6 @@
package org.springframework.test.context.transaction; package org.springframework.test.context.transaction;
import java.util.Map; import java.util.Map;
import javax.sql.DataSource; import javax.sql.DataSource;
import org.apache.commons.logging.Log; import org.apache.commons.logging.Log;
@ -40,6 +39,7 @@ import org.springframework.util.StringUtils;
/** /**
* Utility methods for working with transactions and data access related beans * Utility methods for working with transactions and data access related beans
* within the <em>Spring TestContext Framework</em>. * within the <em>Spring TestContext Framework</em>.
*
* <p>Mainly for internal use within the framework. * <p>Mainly for internal use within the framework.
* *
* @author Sam Brannen * @author Sam Brannen
@ -48,8 +48,6 @@ import org.springframework.util.StringUtils;
*/ */
public abstract class TestContextTransactionUtils { public abstract class TestContextTransactionUtils {
private static final Log logger = LogFactory.getLog(TestContextTransactionUtils.class);
/** /**
* Default bean name for a {@link DataSource}: {@code "dataSource"}. * Default bean name for a {@link DataSource}: {@code "dataSource"}.
*/ */
@ -62,9 +60,8 @@ public abstract class TestContextTransactionUtils {
public static final String DEFAULT_TRANSACTION_MANAGER_NAME = "transactionManager"; public static final String DEFAULT_TRANSACTION_MANAGER_NAME = "transactionManager";
private TestContextTransactionUtils() { private static final Log logger = LogFactory.getLog(TestContextTransactionUtils.class);
/* prevent instantiation */
}
/** /**
* Retrieve the {@link DataSource} to use for the supplied {@linkplain TestContext * Retrieve the {@link DataSource} to use for the supplied {@linkplain TestContext
@ -82,8 +79,8 @@ public abstract class TestContextTransactionUtils {
* {@linkplain #DEFAULT_DATA_SOURCE_NAME default data source name}. * {@linkplain #DEFAULT_DATA_SOURCE_NAME default data source name}.
* @param testContext the test context for which the {@code DataSource} * @param testContext the test context for which the {@code DataSource}
* should be retrieved; never {@code null} * should be retrieved; never {@code null}
* @param name the name of the {@code DataSource} to retrieve; may be {@code null} * @param name the name of the {@code DataSource} to retrieve
* or <em>empty</em> * (may be {@code null} or <em>empty</em>)
* @return the {@code DataSource} to use, or {@code null} if not found * @return the {@code DataSource} to use, or {@code null} if not found
* @throws BeansException if an error occurs while retrieving an explicitly * @throws BeansException if an error occurs while retrieving an explicitly
* named {@code DataSource} * named {@code DataSource}
@ -94,14 +91,14 @@ public abstract class TestContextTransactionUtils {
BeanFactory bf = testContext.getApplicationContext().getAutowireCapableBeanFactory(); BeanFactory bf = testContext.getApplicationContext().getAutowireCapableBeanFactory();
try { try {
// look up by type and explicit name // Look up by type and explicit name
if (StringUtils.hasText(name)) { if (StringUtils.hasText(name)) {
return bf.getBean(name, DataSource.class); return bf.getBean(name, DataSource.class);
} }
} }
catch (BeansException ex) { catch (BeansException ex) {
logger.error( logger.error(String.format("Failed to retrieve DataSource named '%s' for test context %s",
String.format("Failed to retrieve DataSource named '%s' for test context %s", name, testContext), ex); name, testContext), ex);
throw ex; throw ex;
} }
@ -109,9 +106,9 @@ public abstract class TestContextTransactionUtils {
if (bf instanceof ListableBeanFactory) { if (bf instanceof ListableBeanFactory) {
ListableBeanFactory lbf = (ListableBeanFactory) bf; ListableBeanFactory lbf = (ListableBeanFactory) bf;
// look up single bean by type // Look up single bean by type
Map<String, DataSource> dataSources = BeanFactoryUtils.beansOfTypeIncludingAncestors(lbf, Map<String, DataSource> dataSources =
DataSource.class); BeanFactoryUtils.beansOfTypeIncludingAncestors(lbf, DataSource.class);
if (dataSources.size() == 1) { if (dataSources.size() == 1) {
return dataSources.values().iterator().next(); return dataSources.values().iterator().next();
} }
@ -153,8 +150,8 @@ public abstract class TestContextTransactionUtils {
* name}. * name}.
* @param testContext the test context for which the transaction manager * @param testContext the test context for which the transaction manager
* should be retrieved; never {@code null} * should be retrieved; never {@code null}
* @param name the name of the transaction manager to retrieve; may be * @param name the name of the transaction manager to retrieve
* {@code null} or <em>empty</em> * (may be {@code null} or <em>empty</em>)
* @return the transaction manager to use, or {@code null} if not found * @return the transaction manager to use, or {@code null} if not found
* @throws BeansException if an error occurs while retrieving an explicitly * @throws BeansException if an error occurs while retrieving an explicitly
* named transaction manager * named transaction manager
@ -167,14 +164,14 @@ public abstract class TestContextTransactionUtils {
BeanFactory bf = testContext.getApplicationContext().getAutowireCapableBeanFactory(); BeanFactory bf = testContext.getApplicationContext().getAutowireCapableBeanFactory();
try { try {
// look up by type and explicit name // Look up by type and explicit name
if (StringUtils.hasText(name)) { if (StringUtils.hasText(name)) {
return bf.getBean(name, PlatformTransactionManager.class); return bf.getBean(name, PlatformTransactionManager.class);
} }
} }
catch (BeansException ex) { catch (BeansException ex) {
logger.error(String.format("Failed to retrieve transaction manager named '%s' for test context %s", name, logger.error(String.format("Failed to retrieve transaction manager named '%s' for test context %s",
testContext), ex); name, testContext), ex);
throw ex; throw ex;
} }
@ -182,24 +179,24 @@ public abstract class TestContextTransactionUtils {
if (bf instanceof ListableBeanFactory) { if (bf instanceof ListableBeanFactory) {
ListableBeanFactory lbf = (ListableBeanFactory) bf; ListableBeanFactory lbf = (ListableBeanFactory) bf;
// look up single bean by type // Look up single bean by type
Map<String, PlatformTransactionManager> txMgrs = BeanFactoryUtils.beansOfTypeIncludingAncestors(lbf, Map<String, PlatformTransactionManager> txMgrs =
PlatformTransactionManager.class); BeanFactoryUtils.beansOfTypeIncludingAncestors(lbf, PlatformTransactionManager.class);
if (txMgrs.size() == 1) { if (txMgrs.size() == 1) {
return txMgrs.values().iterator().next(); return txMgrs.values().iterator().next();
} }
try { try {
// look up single bean by type, with support for 'primary' beans // Look up single bean by type, with support for 'primary' beans
return bf.getBean(PlatformTransactionManager.class); return bf.getBean(PlatformTransactionManager.class);
} }
catch (BeansException ex) { catch (BeansException ex) {
logBeansException(testContext, ex, PlatformTransactionManager.class); logBeansException(testContext, ex, PlatformTransactionManager.class);
} }
// look up single TransactionManagementConfigurer // Look up single TransactionManagementConfigurer
Map<String, TransactionManagementConfigurer> configurers = BeanFactoryUtils.beansOfTypeIncludingAncestors( Map<String, TransactionManagementConfigurer> configurers =
lbf, TransactionManagementConfigurer.class); BeanFactoryUtils.beansOfTypeIncludingAncestors(lbf, TransactionManagementConfigurer.class);
Assert.state(configurers.size() <= 1, Assert.state(configurers.size() <= 1,
"Only one TransactionManagementConfigurer may exist in the ApplicationContext"); "Only one TransactionManagementConfigurer may exist in the ApplicationContext");
if (configurers.size() == 1) { if (configurers.size() == 1) {
@ -227,13 +224,13 @@ public abstract class TestContextTransactionUtils {
* Create a delegating {@link TransactionAttribute} for the supplied target * Create a delegating {@link TransactionAttribute} for the supplied target
* {@link TransactionAttribute} and {@link TestContext}, using the names of * {@link TransactionAttribute} and {@link TestContext}, using the names of
* the test class and test method to build the name of the transaction. * the test class and test method to build the name of the transaction.
* * @param testContext the {@code TestContext} upon which to base the name
* @param testContext the {@code TestContext} upon which to base the name; never {@code null} * @param targetAttribute the {@code TransactionAttribute} to delegate to
* @param targetAttribute the {@code TransactionAttribute} to delegate to; never {@code null}
* @return the delegating {@code TransactionAttribute} * @return the delegating {@code TransactionAttribute}
*/ */
public static TransactionAttribute createDelegatingTransactionAttribute(TestContext testContext, public static TransactionAttribute createDelegatingTransactionAttribute(
TransactionAttribute targetAttribute) { TestContext testContext, TransactionAttribute targetAttribute) {
Assert.notNull(testContext, "TestContext must not be null"); Assert.notNull(testContext, "TestContext must not be null");
Assert.notNull(targetAttribute, "Target TransactionAttribute must not be null"); Assert.notNull(targetAttribute, "Target TransactionAttribute must not be null");
return new TestContextTransactionAttribute(targetAttribute, testContext); return new TestContextTransactionAttribute(targetAttribute, testContext);

21
spring-test/src/main/java/org/springframework/test/context/transaction/TestTransaction.java

@ -1,5 +1,5 @@
/* /*
* Copyright 2002-2016 the original author or authors. * Copyright 2002-2018 the original author or authors.
* *
* Licensed under the Apache License, Version 2.0 (the "License"); * Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License. * you may not use this file except in compliance with the License.
@ -50,10 +50,8 @@ public class TestTransaction {
TransactionContext transactionContext = TransactionContextHolder.getCurrentTransactionContext(); TransactionContext transactionContext = TransactionContextHolder.getCurrentTransactionContext();
if (transactionContext != null) { if (transactionContext != null) {
TransactionStatus transactionStatus = transactionContext.getTransactionStatus(); TransactionStatus transactionStatus = transactionContext.getTransactionStatus();
return (transactionStatus != null) && (!transactionStatus.isCompleted()); return (transactionStatus != null && !transactionStatus.isCompleted());
} }
// else
return false; return false;
} }
@ -80,8 +78,7 @@ public class TestTransaction {
* Rather, the value of this flag will be used to determine whether or not * Rather, the value of this flag will be used to determine whether or not
* the current test-managed transaction should be rolled back or committed * the current test-managed transaction should be rolled back or committed
* once it is {@linkplain #end ended}. * once it is {@linkplain #end ended}.
* @throws IllegalStateException if a transaction is not active for the * @throws IllegalStateException if no transaction is active for the current test
* current test
* @see #isActive() * @see #isActive()
* @see #isFlaggedForRollback() * @see #isFlaggedForRollback()
* @see #start() * @see #start()
@ -97,8 +94,7 @@ public class TestTransaction {
* Rather, the value of this flag will be used to determine whether or not * Rather, the value of this flag will be used to determine whether or not
* the current test-managed transaction should be rolled back or committed * the current test-managed transaction should be rolled back or committed
* once it is {@linkplain #end ended}. * once it is {@linkplain #end ended}.
* @throws IllegalStateException if a transaction is not active for the * @throws IllegalStateException if no transaction is active for the current test
* current test
* @see #isActive() * @see #isActive()
* @see #isFlaggedForRollback() * @see #isFlaggedForRollback()
* @see #start() * @see #start()
@ -122,9 +118,9 @@ public class TestTransaction {
} }
/** /**
* Immediately force a <em>commit</em> or <em>rollback</em> of the current * Immediately force a <em>commit</em> or <em>rollback</em> of the
* test-managed transaction, according to the {@linkplain #isFlaggedForRollback * current test-managed transaction, according to the
* rollback flag}. * {@linkplain #isFlaggedForRollback rollback flag}.
* @throws IllegalStateException if the transaction context could not be * @throws IllegalStateException if the transaction context could not be
* retrieved or if a transaction is not active for the current test * retrieved or if a transaction is not active for the current test
* @see #isActive() * @see #isActive()
@ -134,6 +130,7 @@ public class TestTransaction {
requireCurrentTransactionContext().endTransaction(); requireCurrentTransactionContext().endTransaction();
} }
private static TransactionContext requireCurrentTransactionContext() { private static TransactionContext requireCurrentTransactionContext() {
TransactionContext txContext = TransactionContextHolder.getCurrentTransactionContext(); TransactionContext txContext = TransactionContextHolder.getCurrentTransactionContext();
Assert.state(txContext != null, "TransactionContext is not active"); Assert.state(txContext != null, "TransactionContext is not active");
@ -144,4 +141,4 @@ public class TestTransaction {
requireCurrentTransactionContext().setFlaggedForRollback(flag); requireCurrentTransactionContext().setFlaggedForRollback(flag);
} }
} }

24
spring-test/src/main/java/org/springframework/test/context/transaction/TransactionContext.java

@ -1,5 +1,5 @@
/* /*
* Copyright 2002-2017 the original author or authors. * Copyright 2002-2018 the original author or authors.
* *
* Licensed under the Apache License, Version 2.0 (the "License"); * Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License. * you may not use this file except in compliance with the License.
@ -16,6 +16,8 @@
package org.springframework.test.context.transaction; package org.springframework.test.context.transaction;
import java.util.concurrent.atomic.AtomicInteger;
import org.apache.commons.logging.Log; import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory; import org.apache.commons.logging.LogFactory;
@ -53,7 +55,7 @@ class TransactionContext {
@Nullable @Nullable
private TransactionStatus transactionStatus; private TransactionStatus transactionStatus;
private volatile int transactionsStarted = 0; private final AtomicInteger transactionsStarted = new AtomicInteger(0);
TransactionContext(TestContext testContext, PlatformTransactionManager transactionManager, TransactionContext(TestContext testContext, PlatformTransactionManager transactionManager,
@ -82,8 +84,8 @@ class TransactionContext {
} }
void setFlaggedForRollback(boolean flaggedForRollback) { void setFlaggedForRollback(boolean flaggedForRollback) {
Assert.state(this.transactionStatus != null, () -> String.format( Assert.state(this.transactionStatus != null, () ->
"Failed to set rollback flag for test context %s: transaction does not exist.", this.testContext)); "Failed to set rollback flag - transaction does not exist: " + this.testContext);
this.flaggedForRollback = flaggedForRollback; this.flaggedForRollback = flaggedForRollback;
} }
@ -95,16 +97,16 @@ class TransactionContext {
*/ */
void startTransaction() { void startTransaction() {
Assert.state(this.transactionStatus == null, Assert.state(this.transactionStatus == null,
"Cannot start a new transaction without ending the existing transaction first."); "Cannot start a new transaction without ending the existing transaction first");
this.flaggedForRollback = this.defaultRollback; this.flaggedForRollback = this.defaultRollback;
this.transactionStatus = this.transactionManager.getTransaction(this.transactionDefinition); this.transactionStatus = this.transactionManager.getTransaction(this.transactionDefinition);
++this.transactionsStarted; int transactionsStarted = this.transactionsStarted.incrementAndGet();
if (logger.isInfoEnabled()) { if (logger.isInfoEnabled()) {
logger.info(String.format( logger.info(String.format(
"Began transaction (%s) for test context %s; transaction manager [%s]; rollback [%s]", "Began transaction (%s) for test context %s; transaction manager [%s]; rollback [%s]",
this.transactionsStarted, this.testContext, this.transactionManager, flaggedForRollback)); transactionsStarted, this.testContext, this.transactionManager, flaggedForRollback));
} }
} }
@ -118,8 +120,8 @@ class TransactionContext {
"Ending transaction for test context %s; transaction status [%s]; rollback [%s]", "Ending transaction for test context %s; transaction status [%s]; rollback [%s]",
this.testContext, this.transactionStatus, this.flaggedForRollback)); this.testContext, this.transactionStatus, this.flaggedForRollback));
} }
Assert.state(this.transactionStatus != null, () -> String.format( Assert.state(this.transactionStatus != null,
"Failed to end transaction for test context %s: transaction does not exist.", this.testContext)); () -> "Failed to end transaction - transaction does not exist: " + this.testContext);
try { try {
if (this.flaggedForRollback) { if (this.flaggedForRollback) {
@ -134,8 +136,8 @@ class TransactionContext {
} }
if (logger.isInfoEnabled()) { if (logger.isInfoEnabled()) {
logger.info(String.format("%s transaction for test context %s.", logger.info((this.flaggedForRollback ? "Rolled back" : "Committed") +
(this.flaggedForRollback ? "Rolled back" : "Committed"), this.testContext)); " transaction for test: " + this.testContext);
} }
} }

12
spring-test/src/main/java/org/springframework/test/context/transaction/TransactionContextHolder.java

@ -1,5 +1,5 @@
/* /*
* Copyright 2002-2017 the original author or authors. * Copyright 2002-2018 the original author or authors.
* *
* Licensed under the Apache License, Version 2.0 (the "License"); * Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License. * you may not use this file except in compliance with the License.
@ -31,7 +31,7 @@ class TransactionContextHolder {
new NamedInheritableThreadLocal<>("Test Transaction Context"); new NamedInheritableThreadLocal<>("Test Transaction Context");
static void setCurrentTransactionContext(@Nullable TransactionContext transactionContext) { static void setCurrentTransactionContext(TransactionContext transactionContext) {
currentTransactionContext.set(transactionContext); currentTransactionContext.set(transactionContext);
} }
@ -42,11 +42,9 @@ class TransactionContextHolder {
@Nullable @Nullable
static TransactionContext removeCurrentTransactionContext() { static TransactionContext removeCurrentTransactionContext() {
synchronized (currentTransactionContext) { TransactionContext transactionContext = currentTransactionContext.get();
TransactionContext transactionContext = currentTransactionContext.get(); currentTransactionContext.remove();
currentTransactionContext.remove(); return transactionContext;
return transactionContext;
}
} }
} }

20
spring-test/src/main/java/org/springframework/test/context/transaction/TransactionalTestExecutionListener.java

@ -1,5 +1,5 @@
/* /*
* Copyright 2002-2017 the original author or authors. * Copyright 2002-2018 the original author or authors.
* *
* Licensed under the Apache License, Version 2.0 (the "License"); * Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License. * you may not use this file except in compliance with the License.
@ -136,6 +136,7 @@ public class TransactionalTestExecutionListener extends AbstractTestExecutionLis
// Do not require @Transactional test methods to be public. // Do not require @Transactional test methods to be public.
protected final TransactionAttributeSource attributeSource = new AnnotationTransactionAttributeSource(false); protected final TransactionAttributeSource attributeSource = new AnnotationTransactionAttributeSource(false);
/** /**
* Returns {@code 4000}. * Returns {@code 4000}.
*/ */
@ -159,10 +160,10 @@ public class TransactionalTestExecutionListener extends AbstractTestExecutionLis
public void beforeTestMethod(final TestContext testContext) throws Exception { public void beforeTestMethod(final TestContext testContext) throws Exception {
Method testMethod = testContext.getTestMethod(); Method testMethod = testContext.getTestMethod();
Class<?> testClass = testContext.getTestClass(); Class<?> testClass = testContext.getTestClass();
Assert.notNull(testMethod, "The test method of the supplied TestContext must not be null"); Assert.notNull(testMethod, "Test method of supplied TestContext must not be null");
TransactionContext txContext = TransactionContextHolder.removeCurrentTransactionContext(); TransactionContext txContext = TransactionContextHolder.removeCurrentTransactionContext();
Assert.state(txContext == null, "Cannot start a new transaction without ending the existing transaction."); Assert.state(txContext == null, "Cannot start new transaction without ending existing transaction");
PlatformTransactionManager tm = null; PlatformTransactionManager tm = null;
TransactionAttribute transactionAttribute = this.attributeSource.getTransactionAttribute(testMethod, testClass); TransactionAttribute transactionAttribute = this.attributeSource.getTransactionAttribute(testMethod, testClass);
@ -172,8 +173,8 @@ public class TransactionalTestExecutionListener extends AbstractTestExecutionLis
transactionAttribute); transactionAttribute);
if (logger.isDebugEnabled()) { if (logger.isDebugEnabled()) {
logger.debug("Explicit transaction definition [" + transactionAttribute + "] found for test context " + logger.debug("Explicit transaction definition [" + transactionAttribute +
testContext); "] found for test context " + testContext);
} }
if (transactionAttribute.getPropagationBehavior() == TransactionDefinition.PROPAGATION_NOT_SUPPORTED) { if (transactionAttribute.getPropagationBehavior() == TransactionDefinition.PROPAGATION_NOT_SUPPORTED) {
@ -181,9 +182,8 @@ public class TransactionalTestExecutionListener extends AbstractTestExecutionLis
} }
tm = getTransactionManager(testContext, transactionAttribute.getQualifier()); tm = getTransactionManager(testContext, transactionAttribute.getQualifier());
Assert.state(tm != null, () -> String.format( Assert.state(tm != null,
"Failed to retrieve PlatformTransactionManager for @Transactional test for test context %s.", () -> "Failed to retrieve PlatformTransactionManager for @Transactional test: " + testContext);
testContext));
} }
if (tm != null) { if (tm != null) {
@ -368,8 +368,8 @@ public class TransactionalTestExecutionListener extends AbstractTestExecutionLis
if (rollbackPresent) { if (rollbackPresent) {
boolean defaultRollback = rollback.value(); boolean defaultRollback = rollback.value();
if (logger.isDebugEnabled()) { if (logger.isDebugEnabled()) {
logger.debug(String.format("Retrieved default @Rollback(%s) for test class [%s].", defaultRollback, logger.debug(String.format("Retrieved default @Rollback(%s) for test class [%s].",
testClass.getName())); defaultRollback, testClass.getName()));
} }
return defaultRollback; return defaultRollback;
} }

76
spring-test/src/test/java/org/springframework/test/context/transaction/PrimaryTransactionManagerTests.java

@ -1,5 +1,5 @@
/* /*
* Copyright 2002-2016 the original author or authors. * Copyright 2002-2018 the original author or authors.
* *
* Licensed under the Apache License, Version 2.0 (the "License"); * Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License. * you may not use this file except in compliance with the License.
@ -53,38 +53,6 @@ import static org.junit.Assert.*;
@DirtiesContext @DirtiesContext
public class PrimaryTransactionManagerTests { public class PrimaryTransactionManagerTests {
@Configuration
static class Config {
@Primary
@Bean
public PlatformTransactionManager primaryTransactionManager() {
return new DataSourceTransactionManager(dataSource1());
}
@Bean
public PlatformTransactionManager additionalTransactionManager() {
return new DataSourceTransactionManager(dataSource2());
}
@Bean
public DataSource dataSource1() {
// @formatter:off
return new EmbeddedDatabaseBuilder()
.generateUniqueName(true)
.addScript("classpath:/org/springframework/test/context/jdbc/schema.sql")
.build();
// @formatter:on
}
@Bean
public DataSource dataSource2() {
return new EmbeddedDatabaseBuilder().generateUniqueName(true).build();
}
}
private JdbcTemplate jdbcTemplate; private JdbcTemplate jdbcTemplate;
@ -93,11 +61,17 @@ public class PrimaryTransactionManagerTests {
this.jdbcTemplate = new JdbcTemplate(dataSource1); this.jdbcTemplate = new JdbcTemplate(dataSource1);
} }
@BeforeTransaction @BeforeTransaction
public void beforeTransaction() { public void beforeTransaction() {
assertNumUsers(0); assertNumUsers(0);
} }
@AfterTransaction
public void afterTransaction() {
assertNumUsers(0);
}
@Test @Test
@Transactional @Transactional
public void transactionalTest() { public void transactionalTest() {
@ -109,14 +83,38 @@ public class PrimaryTransactionManagerTests {
assertNumUsers(1); assertNumUsers(1);
} }
@AfterTransaction private void assertNumUsers(int expected) {
public void afterTransaction() { assertEquals("Number of rows in the 'user' table", expected,
assertNumUsers(0); JdbcTestUtils.countRowsInTable(this.jdbcTemplate, "user"));
} }
private void assertNumUsers(int expected) {
assertEquals("Number of rows in the 'user' table.", expected, @Configuration
JdbcTestUtils.countRowsInTable(this.jdbcTemplate, "user")); static class Config {
@Primary
@Bean
public PlatformTransactionManager primaryTransactionManager() {
return new DataSourceTransactionManager(dataSource1());
}
@Bean
public PlatformTransactionManager additionalTransactionManager() {
return new DataSourceTransactionManager(dataSource2());
}
@Bean
public DataSource dataSource1() {
return new EmbeddedDatabaseBuilder()
.generateUniqueName(true)
.addScript("classpath:/org/springframework/test/context/jdbc/schema.sql")
.build();
}
@Bean
public DataSource dataSource2() {
return new EmbeddedDatabaseBuilder().generateUniqueName(true).build();
}
} }
} }

194
spring-test/src/test/java/org/springframework/test/context/transaction/TransactionalTestExecutionListenerTests.java

@ -23,8 +23,8 @@ import org.junit.After;
import org.junit.Rule; import org.junit.Rule;
import org.junit.Test; import org.junit.Test;
import org.junit.rules.ExpectedException; import org.junit.rules.ExpectedException;
import org.mockito.BDDMockito; import org.mockito.BDDMockito;
import org.springframework.beans.BeanUtils; import org.springframework.beans.BeanUtils;
import org.springframework.core.annotation.AliasFor; import org.springframework.core.annotation.AliasFor;
import org.springframework.test.annotation.Commit; import org.springframework.test.annotation.Commit;
@ -51,7 +51,6 @@ public class TransactionalTestExecutionListenerTests {
private final PlatformTransactionManager tm = mock(PlatformTransactionManager.class); private final PlatformTransactionManager tm = mock(PlatformTransactionManager.class);
private final TransactionalTestExecutionListener listener = new TransactionalTestExecutionListener() { private final TransactionalTestExecutionListener listener = new TransactionalTestExecutionListener() {
@Override @Override
protected PlatformTransactionManager getTransactionManager(TestContext testContext, String qualifier) { protected PlatformTransactionManager getTransactionManager(TestContext testContext, String qualifier) {
return tm; return tm;
@ -64,100 +63,21 @@ public class TransactionalTestExecutionListenerTests {
public ExpectedException exception = ExpectedException.none(); public ExpectedException exception = ExpectedException.none();
private void assertBeforeTestMethod(Class<? extends Invocable> clazz) throws Exception {
assertBeforeTestMethodWithTransactionalTestMethod(clazz);
assertBeforeTestMethodWithNonTransactionalTestMethod(clazz);
}
private void assertBeforeTestMethodWithTransactionalTestMethod(Class<? extends Invocable> clazz) throws Exception {
assertBeforeTestMethodWithTransactionalTestMethod(clazz, true);
}
private void assertBeforeTestMethodWithTransactionalTestMethod(Class<? extends Invocable> clazz, boolean invokedInTx)
throws Exception {
BDDMockito.<Class<?>> given(testContext.getTestClass()).willReturn(clazz);
Invocable instance = BeanUtils.instantiateClass(clazz);
given(testContext.getTestInstance()).willReturn(instance);
given(testContext.getTestMethod()).willReturn(clazz.getDeclaredMethod("transactionalTest"));
assertFalse("callback should not have been invoked", instance.invoked());
TransactionContextHolder.removeCurrentTransactionContext();
listener.beforeTestMethod(testContext);
assertEquals(invokedInTx, instance.invoked());
}
private void assertBeforeTestMethodWithNonTransactionalTestMethod(Class<? extends Invocable> clazz)
throws Exception {
BDDMockito.<Class<?>> given(testContext.getTestClass()).willReturn(clazz);
Invocable instance = BeanUtils.instantiateClass(clazz);
given(testContext.getTestInstance()).willReturn(instance);
given(testContext.getTestMethod()).willReturn(clazz.getDeclaredMethod("nonTransactionalTest"));
assertFalse("callback should not have been invoked", instance.invoked());
TransactionContextHolder.removeCurrentTransactionContext();
listener.beforeTestMethod(testContext);
assertFalse("callback should not have been invoked", instance.invoked());
}
private void assertAfterTestMethod(Class<? extends Invocable> clazz) throws Exception {
assertAfterTestMethodWithTransactionalTestMethod(clazz);
assertAfterTestMethodWithNonTransactionalTestMethod(clazz);
}
private void assertAfterTestMethodWithTransactionalTestMethod(Class<? extends Invocable> clazz) throws Exception {
BDDMockito.<Class<?>> given(testContext.getTestClass()).willReturn(clazz);
Invocable instance = BeanUtils.instantiateClass(clazz);
given(testContext.getTestInstance()).willReturn(instance);
given(testContext.getTestMethod()).willReturn(clazz.getDeclaredMethod("transactionalTest"));
given(tm.getTransaction(BDDMockito.any(TransactionDefinition.class))).willReturn(new SimpleTransactionStatus());
assertFalse("callback should not have been invoked", instance.invoked());
TransactionContextHolder.removeCurrentTransactionContext();
listener.beforeTestMethod(testContext);
assertFalse("callback should not have been invoked", instance.invoked());
listener.afterTestMethod(testContext);
assertTrue("callback should have been invoked", instance.invoked());
}
private void assertAfterTestMethodWithNonTransactionalTestMethod(Class<? extends Invocable> clazz) throws Exception {
BDDMockito.<Class<?>> given(testContext.getTestClass()).willReturn(clazz);
Invocable instance = BeanUtils.instantiateClass(clazz);
given(testContext.getTestInstance()).willReturn(instance);
given(testContext.getTestMethod()).willReturn(clazz.getDeclaredMethod("nonTransactionalTest"));
assertFalse("callback should not have been invoked", instance.invoked());
TransactionContextHolder.removeCurrentTransactionContext();
listener.beforeTestMethod(testContext);
listener.afterTestMethod(testContext);
assertFalse("callback should not have been invoked", instance.invoked());
}
private void assertIsRollback(Class<?> clazz, boolean rollback) throws NoSuchMethodException, Exception {
BDDMockito.<Class<?>> given(testContext.getTestClass()).willReturn(clazz);
given(testContext.getTestMethod()).willReturn(clazz.getDeclaredMethod("test"));
assertEquals(rollback, listener.isRollback(testContext));
}
@After @After
public void cleanUpThreadLocalStateForSubsequentTestClassesInSuite() { public void cleanUpThreadLocalStateForSubsequentTestClassesInSuite() {
TransactionContextHolder.removeCurrentTransactionContext(); TransactionContextHolder.removeCurrentTransactionContext();
} }
/**
* SPR-13895 @Test // SPR-13895
*/
@Test
public void transactionalTestWithoutTransactionManager() throws Exception { public void transactionalTestWithoutTransactionManager() throws Exception {
TransactionalTestExecutionListener listener = new TransactionalTestExecutionListener() { TransactionalTestExecutionListener listener = new TransactionalTestExecutionListener() {
protected PlatformTransactionManager getTransactionManager(TestContext testContext, String qualifier) { protected PlatformTransactionManager getTransactionManager(TestContext testContext, String qualifier) {
return null; return null;
} }
}; };
Class<? extends Invocable> clazz = TransactionalDeclaredOnClassLocallyTestCase.class; Class<? extends Invocable> clazz = TransactionalDeclaredOnClassLocallyTestCase.class;
BDDMockito.<Class<?>> given(testContext.getTestClass()).willReturn(clazz); BDDMockito.<Class<?>> given(testContext.getTestClass()).willReturn(clazz);
Invocable instance = BeanUtils.instantiateClass(clazz); Invocable instance = BeanUtils.instantiateClass(clazz);
given(testContext.getTestInstance()).willReturn(instance); given(testContext.getTestInstance()).willReturn(instance);
@ -172,7 +92,7 @@ public class TransactionalTestExecutionListenerTests {
} }
catch (IllegalStateException e) { catch (IllegalStateException e) {
assertTrue(e.getMessage().startsWith( assertTrue(e.getMessage().startsWith(
"Failed to retrieve PlatformTransactionManager for @Transactional test for test context")); "Failed to retrieve PlatformTransactionManager for @Transactional test"));
} }
} }
@ -191,7 +111,7 @@ public class TransactionalTestExecutionListenerTests {
// Note: not actually invoked within a transaction since the test class is // Note: not actually invoked within a transaction since the test class is
// annotated with @MetaTxWithOverride(propagation = NOT_SUPPORTED) // annotated with @MetaTxWithOverride(propagation = NOT_SUPPORTED)
assertBeforeTestMethodWithTransactionalTestMethod( assertBeforeTestMethodWithTransactionalTestMethod(
TransactionalDeclaredOnClassViaMetaAnnotationWithOverrideTestCase.class, false); TransactionalDeclaredOnClassViaMetaAnnotationWithOverrideTestCase.class, false);
} }
@Test @Test
@ -199,7 +119,7 @@ public class TransactionalTestExecutionListenerTests {
// Note: not actually invoked within a transaction since the method is // Note: not actually invoked within a transaction since the method is
// annotated with @MetaTxWithOverride(propagation = NOT_SUPPORTED) // annotated with @MetaTxWithOverride(propagation = NOT_SUPPORTED)
assertBeforeTestMethodWithTransactionalTestMethod( assertBeforeTestMethodWithTransactionalTestMethod(
TransactionalDeclaredOnMethodViaMetaAnnotationWithOverrideTestCase.class, false); TransactionalDeclaredOnMethodViaMetaAnnotationWithOverrideTestCase.class, false);
assertBeforeTestMethodWithNonTransactionalTestMethod(TransactionalDeclaredOnMethodViaMetaAnnotationWithOverrideTestCase.class); assertBeforeTestMethodWithNonTransactionalTestMethod(TransactionalDeclaredOnMethodViaMetaAnnotationWithOverrideTestCase.class);
} }
@ -289,11 +209,84 @@ public class TransactionalTestExecutionListenerTests {
} }
// ------------------------------------------------------------------------- private void assertBeforeTestMethod(Class<? extends Invocable> clazz) throws Exception {
assertBeforeTestMethodWithTransactionalTestMethod(clazz);
assertBeforeTestMethodWithNonTransactionalTestMethod(clazz);
}
private void assertBeforeTestMethodWithTransactionalTestMethod(Class<? extends Invocable> clazz) throws Exception {
assertBeforeTestMethodWithTransactionalTestMethod(clazz, true);
}
private void assertBeforeTestMethodWithTransactionalTestMethod(Class<? extends Invocable> clazz, boolean invokedInTx)
throws Exception {
BDDMockito.<Class<?>> given(testContext.getTestClass()).willReturn(clazz);
Invocable instance = BeanUtils.instantiateClass(clazz);
given(testContext.getTestInstance()).willReturn(instance);
given(testContext.getTestMethod()).willReturn(clazz.getDeclaredMethod("transactionalTest"));
assertFalse("callback should not have been invoked", instance.invoked());
TransactionContextHolder.removeCurrentTransactionContext();
listener.beforeTestMethod(testContext);
assertEquals(invokedInTx, instance.invoked());
}
private void assertBeforeTestMethodWithNonTransactionalTestMethod(Class<? extends Invocable> clazz) throws Exception {
BDDMockito.<Class<?>> given(testContext.getTestClass()).willReturn(clazz);
Invocable instance = BeanUtils.instantiateClass(clazz);
given(testContext.getTestInstance()).willReturn(instance);
given(testContext.getTestMethod()).willReturn(clazz.getDeclaredMethod("nonTransactionalTest"));
assertFalse("callback should not have been invoked", instance.invoked());
TransactionContextHolder.removeCurrentTransactionContext();
listener.beforeTestMethod(testContext);
assertFalse("callback should not have been invoked", instance.invoked());
}
private void assertAfterTestMethod(Class<? extends Invocable> clazz) throws Exception {
assertAfterTestMethodWithTransactionalTestMethod(clazz);
assertAfterTestMethodWithNonTransactionalTestMethod(clazz);
}
private void assertAfterTestMethodWithTransactionalTestMethod(Class<? extends Invocable> clazz) throws Exception {
BDDMockito.<Class<?>> given(testContext.getTestClass()).willReturn(clazz);
Invocable instance = BeanUtils.instantiateClass(clazz);
given(testContext.getTestInstance()).willReturn(instance);
given(testContext.getTestMethod()).willReturn(clazz.getDeclaredMethod("transactionalTest"));
given(tm.getTransaction(BDDMockito.any(TransactionDefinition.class))).willReturn(new SimpleTransactionStatus());
assertFalse("callback should not have been invoked", instance.invoked());
TransactionContextHolder.removeCurrentTransactionContext();
listener.beforeTestMethod(testContext);
assertFalse("callback should not have been invoked", instance.invoked());
listener.afterTestMethod(testContext);
assertTrue("callback should have been invoked", instance.invoked());
}
private void assertAfterTestMethodWithNonTransactionalTestMethod(Class<? extends Invocable> clazz) throws Exception {
BDDMockito.<Class<?>> given(testContext.getTestClass()).willReturn(clazz);
Invocable instance = BeanUtils.instantiateClass(clazz);
given(testContext.getTestInstance()).willReturn(instance);
given(testContext.getTestMethod()).willReturn(clazz.getDeclaredMethod("nonTransactionalTest"));
assertFalse("callback should not have been invoked", instance.invoked());
TransactionContextHolder.removeCurrentTransactionContext();
listener.beforeTestMethod(testContext);
listener.afterTestMethod(testContext);
assertFalse("callback should not have been invoked", instance.invoked());
}
private void assertIsRollback(Class<?> clazz, boolean rollback) throws Exception {
BDDMockito.<Class<?>> given(testContext.getTestClass()).willReturn(clazz);
given(testContext.getTestMethod()).willReturn(clazz.getDeclaredMethod("test"));
assertEquals(rollback, listener.isRollback(testContext));
}
@Transactional @Transactional
@Retention(RetentionPolicy.RUNTIME) @Retention(RetentionPolicy.RUNTIME)
private static @interface MetaTransactional { private @interface MetaTransactional {
} }
@Transactional @Transactional
@ -308,12 +301,12 @@ public class TransactionalTestExecutionListenerTests {
@BeforeTransaction @BeforeTransaction
@Retention(RetentionPolicy.RUNTIME) @Retention(RetentionPolicy.RUNTIME)
private static @interface MetaBeforeTransaction { private @interface MetaBeforeTransaction {
} }
@AfterTransaction @AfterTransaction
@Retention(RetentionPolicy.RUNTIME) @Retention(RetentionPolicy.RUNTIME)
private static @interface MetaAfterTransaction { private @interface MetaAfterTransaction {
} }
private interface Invocable { private interface Invocable {
@ -348,7 +341,6 @@ public class TransactionalTestExecutionListenerTests {
} }
public void transactionalTest() { public void transactionalTest() {
/* no-op */
} }
} }
@ -361,11 +353,9 @@ public class TransactionalTestExecutionListenerTests {
@Transactional @Transactional
public void transactionalTest() { public void transactionalTest() {
/* no-op */
} }
public void nonTransactionalTest() { public void nonTransactionalTest() {
/* no-op */
} }
} }
@ -378,7 +368,6 @@ public class TransactionalTestExecutionListenerTests {
} }
public void transactionalTest() { public void transactionalTest() {
/* no-op */
} }
} }
@ -391,11 +380,9 @@ public class TransactionalTestExecutionListenerTests {
@MetaTransactional @MetaTransactional
public void transactionalTest() { public void transactionalTest() {
/* no-op */
} }
public void nonTransactionalTest() { public void nonTransactionalTest() {
/* no-op */
} }
} }
@ -408,7 +395,6 @@ public class TransactionalTestExecutionListenerTests {
} }
public void transactionalTest() { public void transactionalTest() {
/* no-op */
} }
} }
@ -421,11 +407,9 @@ public class TransactionalTestExecutionListenerTests {
@MetaTxWithOverride(propagation = NOT_SUPPORTED) @MetaTxWithOverride(propagation = NOT_SUPPORTED)
public void transactionalTest() { public void transactionalTest() {
/* no-op */
} }
public void nonTransactionalTest() { public void nonTransactionalTest() {
/* no-op */
} }
} }
@ -438,11 +422,9 @@ public class TransactionalTestExecutionListenerTests {
@Transactional @Transactional
public void transactionalTest() { public void transactionalTest() {
/* no-op */
} }
public void nonTransactionalTest() { public void nonTransactionalTest() {
/* no-op */
} }
} }
@ -455,11 +437,9 @@ public class TransactionalTestExecutionListenerTests {
@Transactional @Transactional
public void transactionalTest() { public void transactionalTest() {
/* no-op */
} }
public void nonTransactionalTest() { public void nonTransactionalTest() {
/* no-op */
} }
} }
@ -472,11 +452,9 @@ public class TransactionalTestExecutionListenerTests {
@Transactional @Transactional
public void transactionalTest() { public void transactionalTest() {
/* no-op */
} }
public void nonTransactionalTest() { public void nonTransactionalTest() {
/* no-op */
} }
} }
@ -489,11 +467,9 @@ public class TransactionalTestExecutionListenerTests {
@Transactional @Transactional
public void transactionalTest() { public void transactionalTest() {
/* no-op */
} }
public void nonTransactionalTest() { public void nonTransactionalTest() {
/* no-op */
} }
} }
@ -518,11 +494,9 @@ public class TransactionalTestExecutionListenerTests {
@Transactional @Transactional
public void transactionalTest() { public void transactionalTest() {
/* no-op */
} }
public void nonTransactionalTest() { public void nonTransactionalTest() {
/* no-op */
} }
} }
@ -531,11 +505,9 @@ public class TransactionalTestExecutionListenerTests {
@Transactional @Transactional
public void transactionalTest() { public void transactionalTest() {
/* no-op */
} }
public void nonTransactionalTest() { public void nonTransactionalTest() {
/* no-op */
} }
} }

Loading…
Cancel
Save