Skip to content

Commit 8a6e39d

Browse files
committed
Improve Oracle Vector Store IT container startup retry logic
Signed-off-by: Ilayaperumal Gopinathan <ilayaperumal.gopinathan@broadcom.com>
1 parent 920e6f4 commit 8a6e39d

File tree

2 files changed

+14
-4
lines changed

2 files changed

+14
-4
lines changed

vector-stores/spring-ai-oracle-store/src/test/java/org/springframework/ai/vectorstore/oracle/OracleVectorStoreIT.java

Lines changed: 7 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -18,6 +18,7 @@
1818

1919
import java.io.IOException;
2020
import java.nio.charset.StandardCharsets;
21+
import java.time.Duration;
2122
import java.util.Collections;
2223
import java.util.Iterator;
2324
import java.util.List;
@@ -71,8 +72,12 @@
7172
public class OracleVectorStoreIT extends BaseVectorStoreTests {
7273

7374
@Container
74-
static OracleContainer oracle23aiContainer = new OracleContainer(OracleImage.DEFAULT_IMAGE).withCopyFileToContainer(
75-
MountableFile.forClasspathResource("/initialize.sql"), "/container-entrypoint-initdb.d/initialize.sql");
75+
static OracleContainer oracle23aiContainer = new OracleContainer(OracleImage.DEFAULT_IMAGE)
76+
.withCopyFileToContainer(MountableFile.forClasspathResource("/initialize.sql"),
77+
"/container-entrypoint-initdb.d/initialize.sql")
78+
.withStartupTimeout(Duration.ofMinutes(5))
79+
.withStartupAttempts(3)
80+
.withSharedMemorySize(2L * 1024L * 1024L * 1024L); // 2GB shared memory
7681

7782
final List<Document> documents = List.of(
7883
new Document(getText("classpath:/test/data/spring.ai.txt"), Map.of("meta1", "meta1")),

vector-stores/spring-ai-oracle-store/src/test/java/org/springframework/ai/vectorstore/oracle/OracleVectorStoreObservationIT.java

Lines changed: 7 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -18,6 +18,7 @@
1818

1919
import java.io.IOException;
2020
import java.nio.charset.StandardCharsets;
21+
import java.time.Duration;
2122
import java.util.List;
2223
import java.util.Map;
2324

@@ -68,8 +69,12 @@
6869
public class OracleVectorStoreObservationIT {
6970

7071
@Container
71-
static OracleContainer oracle23aiContainer = new OracleContainer(OracleImage.DEFAULT_IMAGE).withCopyFileToContainer(
72-
MountableFile.forClasspathResource("/initialize.sql"), "/container-entrypoint-initdb.d/initialize.sql");
72+
static OracleContainer oracle23aiContainer = new OracleContainer(OracleImage.DEFAULT_IMAGE)
73+
.withCopyFileToContainer(MountableFile.forClasspathResource("/initialize.sql"),
74+
"/container-entrypoint-initdb.d/initialize.sql")
75+
.withStartupTimeout(Duration.ofMinutes(5))
76+
.withStartupAttempts(3)
77+
.withSharedMemorySize(2L * 1024L * 1024L * 1024L); // 2GB shared memory
7378

7479
private final ApplicationContextRunner contextRunner = new ApplicationContextRunner()
7580
.withUserConfiguration(Config.class)

0 commit comments

Comments
 (0)