First AI generated version

integration_test.go

@@ -0,0 +1,85 @@
+//go:build integration
+
+package gollama_test
+
+import (
+	"context"
+	"os"
+	"testing"
+
+	gollama "gitea.neitzel.de/konrad/go-ollama"
+	"gitea.neitzel.de/konrad/go-ollama/internal/integrationtest"
+)
+
+func TestIntegrationPing(t *testing.T) {
+	client := integrationtest.RequireOllama(t)
+	cfg := integrationtest.ConfigForTest(t)
+	ctx, cancel := context.WithTimeout(context.Background(), cfg.Timeout)
+	defer cancel()
+	if err := client.Ping(ctx); err != nil {
+		t.Fatal(err)
+	}
+}
+
+func TestIntegrationModelNames(t *testing.T) {
+	client := integrationtest.RequireOllama(t)
+	cfg := integrationtest.ConfigForTest(t)
+	ctx, cancel := context.WithTimeout(context.Background(), cfg.Timeout)
+	defer cancel()
+	names, err := client.ModelNames(ctx)
+	if err != nil {
+		t.Fatal(err)
+	}
+	t.Logf("models: %v", names)
+}
+
+func TestIntegrationComplete(t *testing.T) {
+	client := integrationtest.RequireOllama(t)
+	cfg := integrationtest.ConfigForTest(t)
+	ctx, cancel := context.WithTimeout(context.Background(), cfg.Timeout)
+	defer cancel()
+
+	model := cfg.Model
+	names, err := client.ModelNames(ctx)
+	if err != nil {
+		t.Fatal(err)
+	}
+	if len(names) == 0 {
+		t.Skip("no models available on server")
+	}
+	if !contains(names, model) {
+		model = names[0]
+		t.Logf("configured model %q not found, using %q", cfg.Model, model)
+	}
+
+	result, err := client.Complete(ctx, "Reply with exactly one word: ok", gollama.WithModel(model))
+	if err != nil {
+		t.Fatal(err)
+	}
+	if result.Text == "" {
+		t.Fatal("empty response")
+	}
+	t.Logf("response: %q", result.Text)
+}
+
+func contains(ss []string, s string) bool {
+	for _, v := range ss {
+		if v == s {
+			return true
+		}
+	}
+	return false
+}
+
+func TestIntegrationPull(t *testing.T) {
+	if os.Getenv("GOLLAMA_TEST_PULL") != "1" {
+		t.Skip("set GOLLAMA_TEST_PULL=1 to run pull integration test")
+	}
+	client := integrationtest.RequireOllama(t)
+	cfg := integrationtest.ConfigForTest(t)
+	ctx, cancel := context.WithTimeout(context.Background(), cfg.Timeout*4)
+	defer cancel()
+	if err := client.Pull(ctx, cfg.Model); err != nil {
+		t.Fatal(err)
+	}
+}