Browse Source
This commit introduces BEFORE_TEST_CLASS and AFTER_TEST_CLASS execution phases for @Sql. See gh-27285pull/29430/head
Andreas Ahlenstorf
5 years ago
committed by
Sam Brannen
8 changed files with 298 additions and 10 deletions
@ -0,0 +1,87 @@
@@ -0,0 +1,87 @@
|
||||
/* |
||||
* Copyright 2002-2023 the original author or authors. |
||||
* |
||||
* 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 |
||||
* |
||||
* https://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.springframework.test.context.jdbc; |
||||
|
||||
import javax.sql.DataSource; |
||||
|
||||
import org.junit.jupiter.api.Order; |
||||
import org.junit.jupiter.api.Test; |
||||
|
||||
import org.springframework.core.Ordered; |
||||
import org.springframework.jdbc.BadSqlGrammarException; |
||||
import org.springframework.jdbc.core.JdbcTemplate; |
||||
import org.springframework.test.annotation.Commit; |
||||
import org.springframework.test.annotation.DirtiesContext; |
||||
import org.springframework.test.context.TestContext; |
||||
import org.springframework.test.context.TestExecutionListener; |
||||
import org.springframework.test.context.TestExecutionListeners; |
||||
import org.springframework.test.context.junit.jupiter.SpringJUnitConfig; |
||||
import org.springframework.test.context.transaction.TestContextTransactionUtils; |
||||
|
||||
import static org.assertj.core.api.Assertions.assertThatExceptionOfType; |
||||
|
||||
/** |
||||
* Verifies that {@link Sql @Sql} with {@link Sql.ExecutionPhase#AFTER_TEST_CLASS} is run after all tests in the class
|
||||
* have been run. |
||||
* |
||||
* @author Andreas Ahlenstorf |
||||
* @since 6.1 |
||||
*/ |
||||
@SpringJUnitConfig(PopulatedSchemaDatabaseConfig.class) |
||||
@DirtiesContext(classMode = DirtiesContext.ClassMode.AFTER_CLASS) |
||||
@Sql(value = {"drop-schema.sql"}, executionPhase = Sql.ExecutionPhase.AFTER_TEST_CLASS) |
||||
@TestExecutionListeners( |
||||
value = AfterTestClassSqlScriptsTests.VerifyTestExecutionListener.class, |
||||
mergeMode = TestExecutionListeners.MergeMode.MERGE_WITH_DEFAULTS |
||||
) |
||||
class AfterTestClassSqlScriptsTests extends AbstractTransactionalTests { |
||||
|
||||
@Test |
||||
@Order(1) |
||||
@Sql(scripts = "data-add-catbert.sql") |
||||
@Commit |
||||
void databaseHasBeenInitialized() { |
||||
assertUsers("Catbert"); |
||||
} |
||||
|
||||
@Test |
||||
@Order(2) |
||||
@Sql(scripts = "data-add-dogbert.sql") |
||||
@Commit |
||||
void databaseIsNotWipedBetweenTests() { |
||||
assertUsers("Catbert", "Dogbert"); |
||||
} |
||||
|
||||
static class VerifyTestExecutionListener implements TestExecutionListener, Ordered { |
||||
|
||||
@Override |
||||
public void afterTestClass(TestContext testContext) throws Exception { |
||||
DataSource dataSource = TestContextTransactionUtils.retrieveDataSource(testContext, null); |
||||
JdbcTemplate jdbcTemplate = new JdbcTemplate(dataSource); |
||||
|
||||
assertThatExceptionOfType(BadSqlGrammarException.class) |
||||
.isThrownBy(() -> jdbcTemplate.queryForList("SELECT name FROM user", String.class)); |
||||
} |
||||
|
||||
@Override |
||||
public int getOrder() { |
||||
// Must run before DirtiesContextTestExecutionListener. Otherwise, the old data source will be removed and
|
||||
// replaced with a new one.
|
||||
return 3001; |
||||
} |
||||
} |
||||
} |
||||
@ -0,0 +1,58 @@
@@ -0,0 +1,58 @@
|
||||
/* |
||||
* Copyright 2002-2023 the original author or authors. |
||||
* |
||||
* 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 |
||||
* |
||||
* https://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.springframework.test.context.jdbc; |
||||
|
||||
import org.junit.jupiter.api.Test; |
||||
|
||||
import org.springframework.test.annotation.DirtiesContext; |
||||
import org.springframework.test.context.junit.jupiter.SpringJUnitConfig; |
||||
|
||||
import static org.springframework.test.context.jdbc.SqlMergeMode.MergeMode.MERGE; |
||||
import static org.springframework.test.context.jdbc.SqlMergeMode.MergeMode.OVERRIDE; |
||||
|
||||
/** |
||||
* Verifies that {@link Sql @Sql} with {@link Sql.ExecutionPhase#BEFORE_TEST_CLASS} is run before all tests in the class
|
||||
* have been run. |
||||
* |
||||
* @author Andreas Ahlenstorf |
||||
* @since 6.1 |
||||
*/ |
||||
@SpringJUnitConfig(classes = EmptyDatabaseConfig.class) |
||||
@DirtiesContext |
||||
@Sql(value = {"schema.sql", "data-add-catbert.sql"}, executionPhase = Sql.ExecutionPhase.BEFORE_TEST_CLASS) |
||||
class BeforeTestClassSqlScriptsTests extends AbstractTransactionalTests { |
||||
|
||||
@Test |
||||
void classLevelScriptsHaveBeenRun() { |
||||
assertUsers("Catbert"); |
||||
} |
||||
|
||||
@Test |
||||
@Sql("data-add-dogbert.sql") |
||||
@SqlMergeMode(MERGE) |
||||
void mergeDoesNotAffectClassLevelPhase() { |
||||
assertUsers("Catbert", "Dogbert"); |
||||
} |
||||
|
||||
@Test |
||||
@Sql({"data-add-dogbert.sql"}) |
||||
@SqlMergeMode(OVERRIDE) |
||||
void overrideDoesNotAffectClassLevelPhase() { |
||||
assertUsers("Dogbert", "Catbert"); |
||||
} |
||||
|
||||
} |
||||
Loading…
Reference in new issue