Ollama: ut for OllamaApiBuilder

Signed-off-by: lambochen <[email protected]>

Author: lambochen

auto-configurations/models/spring-ai-autoconfigure-model-ollama/src/test/java/org/springframework/ai/model/ollama/autoconfigure/OllamaApiBuilderTests.java

@@ -0,0 +1,68 @@
+package org.springframework.ai.model.ollama.autoconfigure;
+
+import org.junit.jupiter.api.Test;
+import org.springframework.ai.ollama.api.OllamaApi;
+
+import static org.assertj.core.api.Assertions.assertThat;
+import static org.assertj.core.api.AssertionsForClassTypes.assertThatThrownBy;
+
+/**
+ * @author lambochen
+ */
+public class OllamaApiBuilderTests {
+
+	@Test
+	void fullBuilder() {
+		OllamaApi ollamaApi = OllamaApi.builder()
+			.baseUrl("http://localhost:11434")
+			.chatPath("/api/chat")
+			.embedPath("/api/embeddings")
+			.listModelsPath("/api/tags")
+			.showModelPath("/api/show")
+			.copyModelPath("/api/copy")
+			.deleteModelPath("/api/delete")
+			.pullModelPath("/api/pull")
+			.build();
+
+		assertThat(ollamaApi).isNotNull();
+	}
+
+	@Test
+	void defaultBuilder() {
+		OllamaApi ollamaApi = OllamaApi.builder().build();
+		assertThat(ollamaApi).isNotNull();
+	}
+
+	@Test
+	void invalidPath() {
+		assertThatThrownBy(() -> OllamaApi.builder().chatPath("").build()).isInstanceOf(IllegalArgumentException.class);
+		assertThatThrownBy(() -> OllamaApi.builder().embedPath("").build())
+			.isInstanceOf(IllegalArgumentException.class);
+		assertThatThrownBy(() -> OllamaApi.builder().listModelsPath("").build())
+			.isInstanceOf(IllegalArgumentException.class);
+		assertThatThrownBy(() -> OllamaApi.builder().showModelPath("").build())
+			.isInstanceOf(IllegalArgumentException.class);
+		assertThatThrownBy(() -> OllamaApi.builder().copyModelPath("").build())
+			.isInstanceOf(IllegalArgumentException.class);
+		assertThatThrownBy(() -> OllamaApi.builder().deleteModelPath("").build())
+			.isInstanceOf(IllegalArgumentException.class);
+		assertThatThrownBy(() -> OllamaApi.builder().pullModelPath("").build())
+			.isInstanceOf(IllegalArgumentException.class);
+
+		assertThatThrownBy(() -> OllamaApi.builder().chatPath(null).build())
+			.isInstanceOf(IllegalArgumentException.class);
+		assertThatThrownBy(() -> OllamaApi.builder().embedPath(null).build())
+			.isInstanceOf(IllegalArgumentException.class);
+		assertThatThrownBy(() -> OllamaApi.builder().listModelsPath(null).build())
+			.isInstanceOf(IllegalArgumentException.class);
+		assertThatThrownBy(() -> OllamaApi.builder().showModelPath(null).build())
+			.isInstanceOf(IllegalArgumentException.class);
+		assertThatThrownBy(() -> OllamaApi.builder().copyModelPath(null).build())
+			.isInstanceOf(IllegalArgumentException.class);
+		assertThatThrownBy(() -> OllamaApi.builder().deleteModelPath(null).build())
+			.isInstanceOf(IllegalArgumentException.class);
+		assertThatThrownBy(() -> OllamaApi.builder().pullModelPath(null).build())
+			.isInstanceOf(IllegalArgumentException.class);
+	}
+
+}