feat : removes test case which doesnt adds context

rajadilipkolli · May 27, 2024 · 57be6b2 · 57be6b2
1 parent 434e4c5
commit 57be6b2
Show file tree

Hide file tree

Showing 3 changed files with 2 additions and 19 deletions.
diff --git a/chatbot-ollama-springai/pom.xml b/chatbot-ollama-springai/pom.xml
@@ -49,11 +49,6 @@
 			<artifactId>rest-assured</artifactId>
 			<scope>test</scope>
 		</dependency>
-		<dependency>
-			<groupId>org.springframework.boot</groupId>
-			<artifactId>spring-boot-testcontainers</artifactId>
-			<scope>test</scope>
-		</dependency>
 		<dependency>
 			<groupId>org.springframework.ai</groupId>
 			<artifactId>spring-ai-spring-boot-testcontainers</artifactId>

diff --git a/chatbot-ollama-springai/src/test/java/com/example/chatbot/ChatbotOllamaApplicationTests.java b/chatbot-ollama-springai/src/test/java/com/example/chatbot/ChatbotOllamaApplicationTests.java
@@ -27,24 +27,12 @@ void setUp() {
         RestAssured.port = localServerPort;
     }
 
-    @Test
-    void contextLoads() {
-        given().contentType(ContentType.JSON)
-                .body(new AIChatRequest("Hello?"))
-                .when()
-                .post("/api/ai/chat")
-                .then()
-                .statusCode(HttpStatus.SC_OK)
-                .contentType(ContentType.JSON)
-                .body("answer", containsString("help"));
-    }
-
     @Test
     void chat() {
         given().contentType(ContentType.JSON)
                 .body(
                         new AIChatRequest(
-                                "As a cricketer, how many centuries did Sachin Tendulkar scored adding up both One Day International (ODI)  and Test centuries ?"))
+                                "As a cricketer, how many centuries did Sachin Tendulkar scored adding up both One Day International (ODI) and Test centuries ?"))
                 .when()
                 .post("/api/ai/chat")
                 .then()

diff --git a/chatbot-ollama-springai/src/test/java/com/example/chatbot/TestChatbotOllamaApplication.java b/chatbot-ollama-springai/src/test/java/com/example/chatbot/TestChatbotOllamaApplication.java
@@ -21,7 +21,7 @@ OllamaContainer ollama() {
     @Bean
     @ServiceConnection
     ChromaDBContainer chromadb() {
-        return new ChromaDBContainer(DockerImageName.parse("chromadb/chroma:0.5.0"));
+        return new ChromaDBContainer(DockerImageName.parse("chromadb/chroma").withTag("0.5.0"));
     }
 
     public static void main(String[] args) {