SpringCloud/spring-cloud-task
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Integration tests now work with Boot 3.2

Browse Source
This commit is contained in:
Glenn Renfro
2023-09-11 14:56:40 -04:00
parent 40566a5bc7
commit 3997a161f2
4 changed files with 89 additions and 130 deletions
Download Patch File Download Diff File Expand all files Collapse all files

95
spring-cloud-task-integration-tests/src/test/java/org/springframework/cloud/task/executionid/TaskStartTests.java
View File

@@ -28,15 +28,15 @@ import java.util.UUID;
import javax.sql.DataSource;
import org.h2.tools.Server;
import org.junit.jupiter.api.AfterEach;
import org.junit.jupiter.api.BeforeEach;
import org.junit.jupiter.api.Test;
import org.junit.jupiter.api.extension.ExtendWith;
import org.testcontainers.containers.MariaDBContainer;
import org.testcontainers.junit.jupiter.Container;
import org.testcontainers.junit.jupiter.Testcontainers;
import org.testcontainers.utility.DockerImageName;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.boot.SpringApplication;
import org.springframework.boot.test.context.SpringBootTest;
import org.springframework.cloud.task.repository.TaskExecution;
import org.springframework.cloud.task.repository.TaskExplorer;
import org.springframework.cloud.task.repository.TaskRepository;
@@ -45,8 +45,6 @@ import org.springframework.cloud.task.repository.support.SimpleTaskRepository;
import org.springframework.cloud.task.repository.support.TaskExecutionDaoFactoryBean;
import org.springframework.context.ApplicationContextException;
import org.springframework.context.ConfigurableApplicationContext;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.core.env.ConfigurableEnvironment;
import org.springframework.core.env.MapPropertySource;
import org.springframework.core.env.MutablePropertySources;
@@ -59,37 +57,26 @@ import org.springframework.jdbc.core.simple.SimpleJdbcInsert;
import org.springframework.jdbc.datasource.DriverManagerDataSource;
import org.springframework.jdbc.datasource.init.DataSourceInitializer;
import org.springframework.jdbc.datasource.init.ResourceDatabasePopulator;
import org.springframework.test.context.junit.jupiter.SpringExtension;
import org.springframework.test.util.TestSocketUtils;
import static org.assertj.core.api.Assertions.assertThat;
import static org.assertj.core.api.Assertions.assertThatExceptionOfType;
@ExtendWith(SpringExtension.class)
@SpringBootTest(classes = { TaskStartTests.TaskLauncherConfiguration.class })
@Testcontainers
public class TaskStartTests {
private final static int WAIT_INTERVAL = 500;
private final static int MAX_WAIT_TIME = 5000;
private final static String DATASOURCE_URL;
private final static String DATASOURCE_USER_NAME = "SA";
private final static String DATASOURCE_USER_PASSWORD = "''";
private final static String DATASOURCE_DRIVER_CLASS_NAME = "org.h2.Driver";
private final static String TASK_NAME = "TASK_LAUNCHER_SINK_TEST";
private static int randomPort;
private static final DockerImageName MARIADB_IMAGE = DockerImageName.parse("mariadb:10.9.3");
static {
randomPort = TestSocketUtils.findAvailableTcpPort();
DATASOURCE_URL = "jdbc:h2:tcp://localhost:" + randomPort + "/mem:dataflow;DB_CLOSE_DELAY=-1;"
+ "DB_CLOSE_ON_EXIT=FALSE";
}
/**
* Provide mariadb test container for tests.
*/
@Container
public static MariaDBContainer<?> mariaDBContainer = new MariaDBContainer<>(MARIADB_IMAGE);
private DataSource dataSource;
@@ -108,21 +95,23 @@ public class TaskStartTests {
}
}
@Autowired
public void setDataSource(DataSource dataSource) {
@BeforeEach
public void setup() {
DriverManagerDataSource dataSource = new DriverManagerDataSource();
dataSource.setDriverClassName(mariaDBContainer.getDriverClassName());
dataSource.setUrl(mariaDBContainer.getJdbcUrl());
dataSource.setUsername(mariaDBContainer.getUsername());
dataSource.setPassword(mariaDBContainer.getPassword());
this.dataSource = dataSource;
TaskExecutionDaoFactoryBean factoryBean = new TaskExecutionDaoFactoryBean(dataSource);
this.taskExplorer = new SimpleTaskExplorer(factoryBean);
this.taskRepository = new SimpleTaskRepository(factoryBean);
}
@BeforeEach
public void setup() {
this.properties = new HashMap<>();
this.properties.put("spring.datasource.url", DATASOURCE_URL);
this.properties.put("spring.datasource.username", DATASOURCE_USER_NAME);
this.properties.put("spring.datasource.password", DATASOURCE_USER_PASSWORD);
this.properties.put("spring.datasource.driverClassName", DATASOURCE_DRIVER_CLASS_NAME);
this.properties.put("spring.datasource.url", mariaDBContainer.getJdbcUrl());
this.properties.put("spring.datasource.username", mariaDBContainer.getUsername());
this.properties.put("spring.datasource.password", mariaDBContainer.getPassword());
this.properties.put("spring.datasource.driverClassName", mariaDBContainer.getDriverClassName());
this.properties.put("spring.application.name", TASK_NAME);
this.properties.put("spring.cloud.task.initialize-enabled", "false");
@@ -147,7 +136,7 @@ public class TaskStartTests {
initializer.setDataSource(this.dataSource);
ResourceDatabasePopulator databasePopulator = new ResourceDatabasePopulator();
databasePopulator.addScript(new ClassPathResource("/org/springframework/cloud/task/schema-h2.sql"));
databasePopulator.addScript(new ClassPathResource("/org/springframework/cloud/task/schema-mariadb.sql"));
initializer.setDatabasePopulator(databasePopulator);
initializer.afterPropertiesSet();
}
@@ -260,6 +249,11 @@ public class TaskStartTests {
ConfigurableEnvironment environment = new StandardEnvironment();
MutablePropertySources propertySources = environment.getPropertySources();
myMap.put("spring.cloud.task.executionid", executionId);
myMap.put("spring.datasource.url", mariaDBContainer.getJdbcUrl());
myMap.put("spring.datasource.username", mariaDBContainer.getUsername());
myMap.put("spring.datasource.password", mariaDBContainer.getPassword());
myMap.put("spring.datasource.driverClassName", mariaDBContainer.getDriverClassName());
propertySources.addFirst(new MapPropertySource("EnvrionmentTestPropsource", myMap));
myapp.setEnvironment(environment);
return myapp;
@@ -296,37 +290,4 @@ public class TaskStartTests {
taskLockInsert.execute(taskLockParams);
}
@Configuration
public static class TaskLauncherConfiguration {
private static Server defaultServer;
@Bean
public Server initH2TCPServer() {
Server server = null;
try {
if (defaultServer == null) {
server = Server.createTcpServer("-ifNotExists", "-tcp", "-tcpAllowOthers", "-tcpPort",
String.valueOf(randomPort)).start();
defaultServer = server;
}
}
catch (SQLException e) {
throw new IllegalStateException(e);
}
return defaultServer;
}
@Bean
public DataSource dataSource() {
DriverManagerDataSource dataSource = new DriverManagerDataSource();
dataSource.setDriverClassName(DATASOURCE_DRIVER_CLASS_NAME);
dataSource.setUrl(DATASOURCE_URL);
dataSource.setUsername(DATASOURCE_USER_NAME);
dataSource.setPassword(DATASOURCE_USER_PASSWORD);
return dataSource;
}
}
}

106
spring-cloud-task-integration-tests/src/test/java/org/springframework/cloud/task/initializer/TaskInitializerTests.java
View File

@@ -19,18 +19,21 @@ package org.springframework.cloud.task.initializer;
import java.sql.Connection;
import java.sql.ResultSet;
import java.sql.SQLException;
import java.util.HashMap;
import java.util.Map;
import javax.sql.DataSource;
import org.h2.tools.Server;
import org.junit.jupiter.api.AfterEach;
import org.junit.jupiter.api.BeforeEach;
import org.junit.jupiter.api.Test;
import org.junit.jupiter.api.extension.ExtendWith;
import org.testcontainers.containers.MariaDBContainer;
import org.testcontainers.junit.jupiter.Container;
import org.testcontainers.junit.jupiter.Testcontainers;
import org.testcontainers.utility.DockerImageName;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.boot.SpringApplication;
import org.springframework.boot.test.context.SpringBootTest;
import org.springframework.cloud.task.executionid.TaskStartApplication;
import org.springframework.cloud.task.repository.TaskExecution;
import org.springframework.cloud.task.repository.TaskExplorer;
@@ -38,52 +41,39 @@ import org.springframework.cloud.task.repository.support.SimpleTaskExplorer;
import org.springframework.cloud.task.repository.support.TaskExecutionDaoFactoryBean;
import org.springframework.context.ApplicationContextException;
import org.springframework.context.ConfigurableApplicationContext;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.core.env.ConfigurableEnvironment;
import org.springframework.core.env.MapPropertySource;
import org.springframework.core.env.MutablePropertySources;
import org.springframework.core.env.StandardEnvironment;
import org.springframework.data.domain.Page;
import org.springframework.data.domain.PageRequest;
import org.springframework.jdbc.core.JdbcTemplate;
import org.springframework.jdbc.datasource.DriverManagerDataSource;
import org.springframework.test.context.junit.jupiter.SpringExtension;
import org.springframework.test.util.TestSocketUtils;
import static org.assertj.core.api.Assertions.assertThat;
import static org.assertj.core.api.Assertions.assertThatExceptionOfType;
@ExtendWith(SpringExtension.class)
@SpringBootTest(classes = { TaskInitializerTests.TaskLauncherConfiguration.class })
@Testcontainers
public class TaskInitializerTests {
private final static int WAIT_INTERVAL = 500;
private final static int MAX_WAIT_TIME = 5000;
private final static String URL = "maven://io.spring.cloud:" + "timestamp-task:jar:1.1.0.RELEASE";
private final static String DATASOURCE_URL;
private final static String DATASOURCE_USER_NAME = "SA";
private final static String DATASOURCE_USER_PASSWORD = "''";
private final static String DATASOURCE_DRIVER_CLASS_NAME = "org.h2.Driver";
private final static String TASK_NAME = "TASK_LAUNCHER_SINK_TEST";
private static int randomPort;
static {
randomPort = TestSocketUtils.findAvailableTcpPort();
DATASOURCE_URL = "jdbc:h2:tcp://localhost:" + randomPort + "/mem:dataflow;DB_CLOSE_DELAY=-1;"
+ "DB_CLOSE_ON_EXIT=FALSE";
}
private DataSource dataSource;
private TaskExplorer taskExplorer;
private ConfigurableApplicationContext applicationContext;
private static final DockerImageName MARIADB_IMAGE = DockerImageName.parse("mariadb:10.9.3");
/**
* Provide mariadb test container for tests.
*/
@Container
public static MariaDBContainer<?> mariaDBContainer = new MariaDBContainer<>(MARIADB_IMAGE);
@AfterEach
public void tearDown() {
if (this.applicationContext != null && this.applicationContext.isActive()) {
@@ -100,6 +90,14 @@ public class TaskInitializerTests {
@BeforeEach
public void setup() {
DriverManagerDataSource dataSource = new DriverManagerDataSource();
dataSource.setDriverClassName(mariaDBContainer.getDriverClassName());
dataSource.setUrl(mariaDBContainer.getJdbcUrl());
dataSource.setUsername(mariaDBContainer.getUsername());
dataSource.setPassword(mariaDBContainer.getPassword());
this.dataSource = dataSource;
TaskExecutionDaoFactoryBean factoryBean = new TaskExecutionDaoFactoryBean(dataSource);
this.taskExplorer = new SimpleTaskExplorer(factoryBean);
JdbcTemplate template = new JdbcTemplate(this.dataSource);
template.execute("DROP TABLE IF EXISTS TASK_TASK_BATCH");
@@ -121,7 +119,7 @@ public class TaskInitializerTests {
@Test
public void testNotInitialized() throws Exception {
SpringApplication myapp = new SpringApplication(TaskStartApplication.class);
SpringApplication myapp = getTaskApplication();
String[] properties = { "--spring.cloud.task.initialize-enabled=false" };
assertThatExceptionOfType(ApplicationContextException.class).isThrownBy(() -> {
this.applicationContext = myapp.run(properties);
@@ -130,7 +128,7 @@ public class TaskInitializerTests {
@Test
public void testWithInitialized() throws Exception {
SpringApplication myapp = new SpringApplication(TaskStartApplication.class);
SpringApplication myapp = getTaskApplication();
String[] properties = { "--spring.cloud.task.initialize-enabled=true" };
this.applicationContext = myapp.run(properties);
assertThat(waitForDBToBePopulated()).isTrue();
@@ -144,7 +142,7 @@ public class TaskInitializerTests {
@Test
public void testNotInitializedOriginalProperty() throws Exception {
SpringApplication myapp = new SpringApplication(TaskStartApplication.class);
SpringApplication myapp = getTaskApplication();
String[] properties = { "--spring.cloud.task.initialize.enable=false" };
assertThatExceptionOfType(ApplicationContextException.class).isThrownBy(() -> {
this.applicationContext = myapp.run(properties);
@@ -153,7 +151,7 @@ public class TaskInitializerTests {
@Test
public void testWithInitializedOriginalProperty() throws Exception {
SpringApplication myapp = new SpringApplication(TaskStartApplication.class);
SpringApplication myapp = getTaskApplication();
String[] properties = { "--spring.cloud.task.initialize.enable=true" };
this.applicationContext = myapp.run(properties);
assertThat(waitForDBToBePopulated()).isTrue();
@@ -186,37 +184,19 @@ public class TaskInitializerTests {
return isDbPopulated;
}
@Configuration
public static class TaskLauncherConfiguration {
private static Server defaultServer;
@Bean
public Server initH2TCPServer() {
Server server = null;
try {
if (defaultServer == null) {
server = Server.createTcpServer("-ifNotExists", "-tcp", "-tcpAllowOthers", "-tcpPort",
String.valueOf(randomPort)).start();
defaultServer = server;
}
}
catch (SQLException e) {
throw new IllegalStateException(e);
}
return defaultServer;
}
@Bean
public DataSource dataSource() {
DriverManagerDataSource dataSource = new DriverManagerDataSource();
dataSource.setDriverClassName(DATASOURCE_DRIVER_CLASS_NAME);
dataSource.setUrl(DATASOURCE_URL);
dataSource.setUsername(DATASOURCE_USER_NAME);
dataSource.setPassword(DATASOURCE_USER_PASSWORD);
return dataSource;
}
private SpringApplication getTaskApplication() {
SpringApplication myapp = new SpringApplication(TaskStartApplication.class);
Map<String, Object> myMap = new HashMap<>();
ConfigurableEnvironment environment = new StandardEnvironment();
MutablePropertySources propertySources = environment.getPropertySources();
myMap.put("spring.datasource.url", mariaDBContainer.getJdbcUrl());
myMap.put("spring.datasource.username", mariaDBContainer.getUsername());
myMap.put("spring.datasource.password", mariaDBContainer.getPassword());
myMap.put("spring.datasource.driverClassName", mariaDBContainer.getDriverClassName());
propertySources.addFirst(new MapPropertySource("EnvrionmentTestPropsource", myMap));
myapp.setEnvironment(environment);
return myapp;
}
}

3
spring-cloud-task-integration-tests/src/test/java/org/springframework/cloud/task/listener/BatchExecutionEventTests.java
View File

@@ -37,6 +37,7 @@ import org.springframework.cloud.stream.binder.test.OutputDestination;
import org.springframework.cloud.stream.binder.test.TestChannelBinderConfiguration;
import org.springframework.cloud.task.batch.listener.support.JobExecutionEvent;
import org.springframework.cloud.task.batch.listener.support.StepExecutionEvent;
import org.springframework.cloud.task.configuration.EnableTask;
import org.springframework.context.ConfigurableApplicationContext;
import org.springframework.context.annotation.Import;
import org.springframework.messaging.Message;
@@ -223,12 +224,14 @@ public class BatchExecutionEventTests {
}
@SpringBootApplication
@EnableTask
@Import(JobConfiguration.class)
public static class BatchEventsApplication {
}
@SpringBootApplication
@EnableTask
@Import(JobSkipConfiguration.class)
public static class BatchSkipEventsApplication {