Skip to content

Commit 5562739

Browse files
authored
refactor: replace DoclingServeClientBuilderFactory with DoclingServeApi in tests (#453)
Signed-off-by: Eric Deandrea <eric.deandrea@ibm.com>
1 parent 9028cee commit 5562739

1 file changed

Lines changed: 6 additions & 6 deletions

File tree

docling-testcontainers/src/test/java/ai/docling/testcontainers/serve/DoclingServeContainerAvailableTests.java

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -7,10 +7,10 @@
77
import org.testcontainers.junit.jupiter.Container;
88
import org.testcontainers.junit.jupiter.Testcontainers;
99

10+
import ai.docling.serve.api.DoclingServeApi;
1011
import ai.docling.serve.api.clear.request.ClearResultsRequest;
1112
import ai.docling.serve.api.clear.response.ClearResponse;
1213
import ai.docling.serve.api.health.HealthCheckResponse;
13-
import ai.docling.serve.client.DoclingServeClientBuilderFactory;
1414
import ai.docling.serve.client.DoclingServeClientException;
1515
import ai.docling.testcontainers.serve.config.DoclingServeContainerConfig;
1616

@@ -45,7 +45,7 @@ class DoclingServeContainerAvailableTests {
4545

4646
@Test
4747
void containerNoUI() {
48-
var client = DoclingServeClientBuilderFactory.newBuilder()
48+
var client = DoclingServeApi.builder()
4949
.baseUrl(this.noUiDoclingContainer.getApiUrl())
5050
.build();
5151

@@ -61,7 +61,7 @@ void containerNoUI() {
6161

6262
@Test
6363
void containerAvailable() {
64-
var client = DoclingServeClientBuilderFactory.newBuilder()
64+
var client = DoclingServeApi.builder()
6565
.baseUrl(this.noUiDoclingContainer.getApiUrl())
6666
.build();
6767

@@ -77,7 +77,7 @@ void containerAvailable() {
7777

7878
@Test
7979
void containerWithApiKeyClientWithout() {
80-
var client = DoclingServeClientBuilderFactory.newBuilder()
80+
var client = DoclingServeApi.builder()
8181
.baseUrl(this.withApiKeyDoclingContainer.getApiUrl())
8282
.logRequests()
8383
.logResponses()
@@ -93,7 +93,7 @@ void containerWithApiKeyClientWithout() {
9393

9494
@Test
9595
void containerWithoutApiKeyClientWith() {
96-
var client = DoclingServeClientBuilderFactory.newBuilder()
96+
var client = DoclingServeApi.builder()
9797
.baseUrl(this.doclingContainer.getApiUrl())
9898
.logRequests()
9999
.logResponses()
@@ -109,7 +109,7 @@ void containerWithoutApiKeyClientWith() {
109109

110110
@Test
111111
void containerWithApiKeyClientWith() {
112-
var client = DoclingServeClientBuilderFactory.newBuilder()
112+
var client = DoclingServeApi.builder()
113113
.baseUrl(this.withApiKeyDoclingContainer.getApiUrl())
114114
.logRequests()
115115
.logResponses()

0 commit comments

Comments
 (0)