Skip to content

Commit 5ce51ce

Browse files
committed
Generate test app when spring ai chat memory cassandra is selected
1 parent 5e92b36 commit 5ce51ce

File tree

4 files changed

+13
-3
lines changed

4 files changed

+13
-3
lines changed

start-site/src/main/java/io/spring/start/site/extension/dependency/cassandra/CassandraProjectGenerationConfiguration.java

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -68,7 +68,8 @@ ComposeFileCustomizer cassandraComposeFileCustomizer(Build build, DockerServiceR
6868

6969
private boolean isCassandraEnabled(Build build) {
7070
return build.dependencies().has("data-cassandra") || build.dependencies().has("data-cassandra-reactive")
71-
|| build.dependencies().has("spring-ai-vectordb-cassandra");
71+
|| build.dependencies().has("spring-ai-vectordb-cassandra")
72+
|| build.dependencies().has("spring-ai-chat-memory-repository-cassandra");
7273
}
7374

7475
}

start-site/src/main/java/io/spring/start/site/extension/dependency/testcontainers/TestcontainersModuleRegistry.java

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -57,7 +57,8 @@ static Iterable<ImplicitDependency> create(Testcontainers testcontainers) {
5757
.customizeHelpDocument(addReferenceLink("Consul Module", "consul/")));
5858
builders.add(onDependencies("cloud-starter-vault-config").customizeBuild(addModule("vault", testcontainers))
5959
.customizeHelpDocument(addReferenceLink("Vault Module", "vault/")));
60-
builders.add(onDependencies("data-cassandra", "data-cassandra-reactive", "spring-ai-vectordb-cassandra")
60+
builders.add(onDependencies("data-cassandra", "data-cassandra-reactive", "spring-ai-vectordb-cassandra",
61+
"spring-ai-chat-memory-repository-cassandra")
6162
.customizeBuild(addModule("cassandra", testcontainers, SupportedContainer.CASSANDRA))
6263
.customizeHelpDocument(addReferenceLink("Cassandra Module", "databases/cassandra/")));
6364
builders.add(onDependencies("data-couchbase", "data-couchbase-reactive")

start-site/src/test/java/io/spring/start/site/extension/dependency/cassandra/CassandraProjectGenerationConfigurationTests.java

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -69,4 +69,11 @@ void doesNotFailWhenBothCassandraAndReactiveCassandraAreSelected() {
6969
assertThat(composeFile(request)).hasSameContentAs(new ClassPathResource("compose/cassandra.yaml"));
7070
}
7171

72+
@Test
73+
void createsCassandraServiceWhenSpringAiChatMemoryIsSelected() {
74+
ProjectRequest request = createProjectRequest(SupportedBootVersion.V3_5, "docker-compose",
75+
"spring-ai-chat-memory-repository-cassandra");
76+
assertThat(composeFile(request)).hasSameContentAs(new ClassPathResource("compose/cassandra.yaml"));
77+
}
78+
7279
}

start-site/src/test/java/io/spring/start/site/extension/dependency/testcontainers/TestcontainersProjectGenerationConfigurationTests.java

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -80,7 +80,8 @@ static Stream<Arguments> supportedTestcontainersActiveMQEntriesBuild() {
8080
}
8181

8282
static Stream<Arguments> supportedTestcontainersSpringAiEntriesBuild() {
83-
return Stream.of(Arguments.arguments("spring-ai-vectordb-chroma", "chromadb"),
83+
return Stream.of(Arguments.arguments("spring-ai-chat-memory-repository-cassandra", "cassandra"),
84+
Arguments.arguments("spring-ai-vectordb-chroma", "chromadb"),
8485
Arguments.arguments("spring-ai-vectordb-milvus", "milvus"),
8586
Arguments.arguments("spring-ai-vectordb-mongodb-atlas", "mongodb"),
8687
Arguments.arguments("spring-ai-vectordb-qdrant", "qdrant"),

0 commit comments

Comments
 (0)