org.springframework.test.jdbc.JdbcTestUtils Java Examples

@Test
public void testZeroCustomTransactionManagerConfiguration() {
	ApplicationContextRunner applicationContextRunner = new ApplicationContextRunner()
			.withConfiguration(
					AutoConfigurations.of(PropertyPlaceholderAutoConfiguration.class,
							SimpleTaskAutoConfiguration.class,
							ZeroTransactionManagerConfiguration.class))
			.withPropertyValues("application.name=transactionManagerTask");

	applicationContextRunner.run((context) -> {
		DataSource dataSource = context.getBean("dataSource", DataSource.class);

		int taskExecutionCount = JdbcTestUtils
				.countRowsInTable(new JdbcTemplate(dataSource), "TASK_EXECUTION");

		assertThat(taskExecutionCount).isEqualTo(1);
	});
}
 

@Test
public void checkUserNotDuplicated() throws Exception {
    final SavedQuery savedQuery1 = new SavedQuery.Builder()
            .setTitle("title1")
            .setMinScore(0)
            .build();

    final SavedQuery savedQuery2 = new SavedQuery.Builder()
            .setTitle("title2")
            .setMinScore(0)
            .build();

    createSavedQuery(savedQuery1);
    createSavedQuery(savedQuery2);

    final int userRows = JdbcTestUtils.countRowsInTable(jdbcTemplate, "find." + UserEntity.Table.NAME);
    assertThat(userRows, is(1));
}
 

@Test
@Sql("data.sql")
public void dataSourceTest() {
	TransactionTestUtils.assertInTransaction(false);
	assertEquals("Number of rows in the 'user' table.", 1,
		JdbcTestUtils.countRowsInTable(this.jdbcTemplate, "user"));
}
 

public static void executeScript(DataSource dataSource, String... sqlResourcePaths) throws DataAccessException {
	JdbcTemplate jdbcTemplate = new JdbcTemplate(dataSource);
	for (String sqlResourcePath : sqlResourcePaths) {
		Resource resource = resourceLoader.getResource(sqlResourcePath);
		JdbcTestUtils.executeSqlScript(jdbcTemplate, new EncodedResource(resource, DEFAULT_ENCODING), true);
	}
}
 

private void testConfiguration(Class configurationClass) {
	ApplicationContextRunner applicationContextRunner = new ApplicationContextRunner()
			.withConfiguration(
					AutoConfigurations.of(PropertyPlaceholderAutoConfiguration.class,
							SimpleTaskAutoConfiguration.class, configurationClass))
			.withPropertyValues("application.name=transactionManagerTask");

	applicationContextRunner.run((context) -> {
		DataSource dataSource = context.getBean("dataSource", DataSource.class);

		int taskExecutionCount = JdbcTestUtils
				.countRowsInTable(new JdbcTemplate(dataSource), "TASK_EXECUTION");

		// Verify that the create call was rolled back
		assertThat(taskExecutionCount).isEqualTo(0);

		// Execute a new create call so that things close cleanly
		TaskRepository taskRepository = context.getBean("taskRepository",
				TaskRepository.class);

		TaskExecution taskExecution = taskRepository
				.createTaskExecution("transactionManagerTask");
		taskExecution = taskRepository.startTaskExecution(
				taskExecution.getExecutionId(), taskExecution.getTaskName(),
				new Date(), new ArrayList<>(0), null);

		TaskLifecycleListener listener = context.getBean(TaskLifecycleListener.class);

		ReflectionTestUtils.setField(listener, "taskExecution", taskExecution);
	});
}
 

@Test
@Sql("data.sql")
public void dataSourceTest() {
	TransactionTestUtils.assertInTransaction(false);
	assertEquals("Number of rows in the 'user' table.", 1,
		JdbcTestUtils.countRowsInTable(this.jdbcTemplate, "user"));
}
 

protected int deleteFromTables(String... names) {
	return JdbcTestUtils.deleteFromTables(this.jdbcTemplate, names);
}
 

protected int countRowsInTable(String tableName) {
	return JdbcTestUtils.countRowsInTable(this.jdbcTemplate, tableName);
}
 

private int countNumEvents() {
	return JdbcTestUtils.countRowsInTable(jdbcTemplate, "event");
}
 

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,
		JdbcTestUtils.countRowsInTable(jdbcTemplate, "user"));
}
 

protected void assertNumUsers(int expected) {
	assertEquals("Number of rows in the 'user' table.", expected,
		JdbcTestUtils.countRowsInTable(jdbcTemplate, "user"));
}
 

private void assertNumUsers(int expected) {
	assertEquals("Number of rows in the 'user' table", expected,
			JdbcTestUtils.countRowsInTable(this.jdbcTemplate, "user"));
}
 

@Before
public void setUp() {
    JdbcTestUtils.deleteFromTables(simpleJdbcTemplate, "SERVICETICKET");
    JdbcTestUtils.deleteFromTables(simpleJdbcTemplate, "TICKETGRANTINGTICKET");
}
 

private int countRowsInTable(String tableName) {
	return JdbcTestUtils.countRowsInTable(this.jdbcTemplate, tableName);
}
 

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,
		JdbcTestUtils.countRowsInTable(jdbcTemplate, "user"));
}
 

protected int countRowsInTable(String tableName) {
	return JdbcTestUtils.countRowsInTable(this.jdbcTemplate, tableName);
}
 

protected int deleteFromTables(String... names) {
	return JdbcTestUtils.deleteFromTables(this.jdbcTemplate, names);
}
 

@After
public void tearDown() {
    JdbcTestUtils.dropTables(jdbcTemplate, "test.person");
}
 

protected void assertNumUsers(int expected) {
	assertEquals("Number of rows in the 'user' table.", expected,
		JdbcTestUtils.countRowsInTable(jdbcTemplate, "user"));
}
 

@Before
public void setUp() {
    JdbcTestUtils.deleteFromTables(simpleJdbcTemplate, "SERVICETICKET");
    JdbcTestUtils.deleteFromTables(simpleJdbcTemplate, "TICKETGRANTINGTICKET");
}
 

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,
		JdbcTestUtils.countRowsInTable(jdbcTemplate, "user"));
}
 

protected void assertNumUsers(int expected) {
	assertEquals("Number of rows in the 'user' table.", expected,
		JdbcTestUtils.countRowsInTable(jdbcTemplate, "user"));
}
 

protected int countRowsInTable(String tableName) {
	return JdbcTestUtils.countRowsInTable(this.jdbcTemplate, tableName);
}
 

private void assertNumUsers(int expected) {
	assertEquals("Number of rows in the 'user' table", expected,
			JdbcTestUtils.countRowsInTable(this.jdbcTemplate, "user"));
}
 

protected int deleteFromTables(String... names) {
	return JdbcTestUtils.deleteFromTables(this.jdbcTemplate, names);
}
 

private int countRowsInTable(String tableName) {
	return JdbcTestUtils.countRowsInTable(this.jdbcTemplate, tableName);
}
 

private int countRowsInTable(String tableName) {
	return JdbcTestUtils.countRowsInTable(this.jdbcTemplate, tableName);
}