Spring/spring-ai
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Polish Qdrant tests

Browse Source 
Use helper methods provided by Testcontainers.
This commit is contained in:
Eddú Meléndez
2024-06-19 16:45:18 -05:00
committed by Christian Tzolov
parent 5543643452
commit 5844f9bf39
2 changed files with 4 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
spring-ai-spring-boot-autoconfigure/src/test/java/org/springframework/ai/autoconfigure/vectorstore/qdrant/QdrantVectorStoreAutoConfigurationIT.java
View File

@@ -46,8 +46,6 @@ import static org.assertj.core.api.Assertions.assertThat;
@Testcontainers
public class QdrantVectorStoreAutoConfigurationIT {
private static final int QDRANT_GRPC_PORT = 6334;
@Container
static QdrantContainer qdrantContainer = new QdrantContainer("qdrant/qdrant:v1.9.2");
@@ -59,7 +57,7 @@ public class QdrantVectorStoreAutoConfigurationIT {
private final ApplicationContextRunner contextRunner = new ApplicationContextRunner()
.withConfiguration(AutoConfigurations.of(QdrantVectorStoreAutoConfiguration.class))
.withUserConfiguration(Config.class)
.withPropertyValues("spring.ai.vectorstore.qdrant.port=" + qdrantContainer.getMappedPort(QDRANT_GRPC_PORT),
.withPropertyValues("spring.ai.vectorstore.qdrant.port=" + qdrantContainer.getGrpcPort(),
"spring.ai.vectorstore.qdrant.host=" + qdrantContainer.getHost());
@Test

7
vector-stores/spring-ai-qdrant-store/src/test/java/org/springframework/ai/vectorstore/qdrant/QdrantVectorStoreIT.java
View File

@@ -49,6 +49,7 @@ import static org.assertj.core.api.Assertions.assertThat;
/**
* @author Anush Shetty
* @author Josh Long
* @author Eddú Meléndez
* @since 0.8.1
*/
@Testcontainers
@@ -60,8 +61,6 @@ public class QdrantVectorStoreIT {
private static final int EMBEDDING_DIMENSION = 1536;
private static final int QDRANT_GRPC_PORT = 6334;
@Container
static QdrantContainer qdrantContainer = new QdrantContainer("qdrant/qdrant:v1.9.2");
@@ -81,7 +80,7 @@ public class QdrantVectorStoreIT {
static void setup() throws InterruptedException, ExecutionException {
String host = qdrantContainer.getHost();
int port = qdrantContainer.getMappedPort(QDRANT_GRPC_PORT);
int port = qdrantContainer.getGrpcPort();
QdrantClient client = new QdrantClient(QdrantGrpcClient.newBuilder(host, port, false).build());
client
@@ -245,7 +244,7 @@ public class QdrantVectorStoreIT {
@Bean
public QdrantClient qdrantClient() {
String host = qdrantContainer.getHost();
int port = qdrantContainer.getMappedPort(QDRANT_GRPC_PORT);
int port = qdrantContainer.getGrpcPort();
QdrantClient qdrantClient = new QdrantClient(QdrantGrpcClient.newBuilder(host, port, false).build());
return qdrantClient;
}