From 84ea3d359b2c69e8e3920b6fe0e6a811889bb5dc Mon Sep 17 00:00:00 2001
From: Madhura Bhave <mbhave@pivotal.io>
Date: Mon, 27 Jan 2020 16:37:50 -0800
Subject: [PATCH] Increase timeout for Cassandra tests

Closes gh-19802
---
 .../CassandraDataAutoConfigurationIntegrationTests.java         | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/spring-boot-project/spring-boot-autoconfigure/src/test/java/org/springframework/boot/autoconfigure/data/cassandra/CassandraDataAutoConfigurationIntegrationTests.java b/spring-boot-project/spring-boot-autoconfigure/src/test/java/org/springframework/boot/autoconfigure/data/cassandra/CassandraDataAutoConfigurationIntegrationTests.java
index 1338fb25f9d..376503067e5 100644
--- a/spring-boot-project/spring-boot-autoconfigure/src/test/java/org/springframework/boot/autoconfigure/data/cassandra/CassandraDataAutoConfigurationIntegrationTests.java
+++ b/spring-boot-project/spring-boot-autoconfigure/src/test/java/org/springframework/boot/autoconfigure/data/cassandra/CassandraDataAutoConfigurationIntegrationTests.java
@@ -55,6 +55,7 @@ class CassandraDataAutoConfigurationIntegrationTests {
 	@BeforeEach
 	void setUp() {
 		this.context = new AnnotationConfigApplicationContext();
+		System.setProperty("datastax-java-driver.advanced.connection.init-query-timeout", "1 second");
 		TestPropertyValues
 				.of("spring.data.cassandra.contact-points:localhost:" + cassandra.getFirstMappedPort(),
 						"spring.data.cassandra.local-datacenter=datacenter1",
@@ -64,6 +65,7 @@ class CassandraDataAutoConfigurationIntegrationTests {
 
 	@AfterEach
 	void close() {
+		System.clearProperty("datastax-java-driver.advanced.connection.init-query-timeout");
 		if (this.context != null) {
 			this.context.close();
 		}