OpenAiCompatibleV1ApiControllerTest : listModels() (#44) (#45)

Co-authored-by: Arnaud MENGUS <[email protected]>

backend/build.gradle

@@ -49,6 +49,8 @@ dependencies {
 	testImplementation 'org.awaitility:awaitility:4.2.1'
 	testImplementation 'org.springframework.boot:spring-boot-starter-test'
 	testImplementation 'org.springframework.security:spring-security-test'
+	testImplementation 'io.github.lambdua:client:0.20.9'
+	testImplementation 'io.github.lambdua:service:0.20.9'
 
 	compileOnly 'org.projectlombok:lombok:1.18.32'
 	annotationProcessor 'org.projectlombok:lombok:1.18.32'

backend/src/main/resources/application-test.yaml

@@ -15,6 +15,8 @@ org:
       poll-interval-in-seconds: 5 # Check for new work every 5 seconds
     job-scheduler:
       enabled: true
+    dashboard:
+      enabled: false
     jobs:
       default-number-of-retries: 0 # Number of retries for a failing job
 dragon:

backend/src/test/java/ai/dragon/controller/api/raag/OpenAiCompatibleV1ApiControllerTest.java

@@ -0,0 +1,29 @@
+package ai.dragon.controller.api.raag;
+
+import static org.junit.jupiter.api.Assertions.assertFalse;
+
+import java.util.List;
+
+import org.junit.jupiter.api.Test;
+import org.springframework.boot.test.context.SpringBootTest;
+import org.springframework.boot.test.context.SpringBootTest.WebEnvironment;
+import org.springframework.boot.test.web.server.LocalServerPort;
+import org.springframework.test.context.ActiveProfiles;
+
+import com.theokanning.openai.model.Model;
+import com.theokanning.openai.service.OpenAiService;
+
+@SpringBootTest(webEnvironment = WebEnvironment.RANDOM_PORT)
+@ActiveProfiles("test")
+public class OpenAiCompatibleV1ApiControllerTest {
+    @LocalServerPort
+    private int serverPort;
+
+    @Test
+    void listModels() throws Exception {
+        OpenAiService service = new OpenAiService("TODO_PUT_KEY_HERE",
+                String.format("http://localhost:%d/api/raag/v1/", serverPort));
+        List<Model> models = service.listModels();
+        assertFalse(models.isEmpty());
+    }
+}