Browse Source
Prior to this commit, it was only possible to declare SQL statements via @Sql within external script resources (i.e., classpath or file system resources); however, many developers have inquired about the ability to inline SQL statements with @Sql analogous to the support for inlined properties in @TestPropertySource. This commit introduces support for declaring _inlined SQL statements_ in `@Sql` via a new `statements` attribute. Inlined statements are executed after statements in scripts. Issue: SPR-13159pull/826/head
Sam Brannen
11 years ago
6 changed files with 181 additions and 34 deletions
@ -0,0 +1,86 @@
@@ -0,0 +1,86 @@
|
||||
/* |
||||
* Copyright 2002-2015 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 |
||||
* |
||||
* 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.springframework.test.context.jdbc; |
||||
|
||||
import javax.sql.DataSource; |
||||
|
||||
import org.junit.FixMethodOrder; |
||||
import org.junit.Test; |
||||
import org.junit.runner.RunWith; |
||||
import org.junit.runners.MethodSorters; |
||||
|
||||
import org.springframework.beans.factory.annotation.Autowired; |
||||
import org.springframework.jdbc.core.JdbcTemplate; |
||||
import org.springframework.test.annotation.DirtiesContext; |
||||
import org.springframework.test.context.ContextConfiguration; |
||||
import org.springframework.test.context.junit4.SpringJUnit4ClassRunner; |
||||
import org.springframework.test.jdbc.JdbcTestUtils; |
||||
import org.springframework.transaction.annotation.Transactional; |
||||
|
||||
import static org.junit.Assert.*; |
||||
|
||||
/** |
||||
* Transactional integration tests for {@link Sql @Sql} support with |
||||
* inlined SQL {@link Sql#statements statements}. |
||||
* |
||||
* @author Sam Brannen |
||||
* @since 4.2 |
||||
* @see TransactionalSqlScriptsTests |
||||
*/ |
||||
@RunWith(SpringJUnit4ClassRunner.class) |
||||
@FixMethodOrder(MethodSorters.NAME_ASCENDING) |
||||
@ContextConfiguration(classes = EmptyDatabaseConfig.class) |
||||
@Transactional |
||||
@Sql( |
||||
scripts = "schema.sql", |
||||
statements = "INSERT INTO user VALUES('Dilbert')" |
||||
) |
||||
@DirtiesContext |
||||
public class TransactionalInlinedStatementsSqlScriptsTests { |
||||
|
||||
protected JdbcTemplate jdbcTemplate; |
||||
|
||||
|
||||
@Autowired |
||||
public void setDataSource(DataSource dataSource) { |
||||
this.jdbcTemplate = new JdbcTemplate(dataSource); |
||||
} |
||||
|
||||
@Test |
||||
// test##_ prefix is required for @FixMethodOrder.
|
||||
public void test01_classLevelScripts() { |
||||
assertNumUsers(1); |
||||
} |
||||
|
||||
@Test |
||||
@Sql(statements = "DROP TABLE user IF EXISTS") |
||||
@Sql("schema.sql") |
||||
@Sql(statements = "INSERT INTO user VALUES ('Dilbert'), ('Dogbert'), ('Catbert')") |
||||
// test##_ prefix is required for @FixMethodOrder.
|
||||
public void test02_methodLevelScripts() { |
||||
assertNumUsers(3); |
||||
} |
||||
|
||||
protected int countRowsInTable(String tableName) { |
||||
return JdbcTestUtils.countRowsInTable(this.jdbcTemplate, tableName); |
||||
} |
||||
|
||||
protected void assertNumUsers(int expected) { |
||||
assertEquals("Number of rows in the 'user' table.", expected, countRowsInTable("user")); |
||||
} |
||||
|
||||
} |
||||
Loading…
Reference in new issue