package embeddings

import (
	"context"
	"net/http"
	"net/http/httptest"
	"testing"
)

func TestNewOllama_Defaults(t *testing.T) {
	e, err := NewOllama(Config{})
	if err != nil {
		t.Fatalf("unexpected error: %v", err)
	}
	if e.baseURL != "http://localhost:11434" {
		t.Errorf("expected 'http://localhost:11434', got %q", e.baseURL)
	}
	if e.model != "nomic-embed-text" {
		t.Errorf("expected 'nomic-embed-text', got %q", e.model)
	}
}

func TestNewOllama_Custom(t *testing.T) {
	e, err := NewOllama(Config{
		BaseURL: "http://custom:8080",
		Model:   "my-embedder",
	})
	if err != nil {
		t.Fatalf("unexpected error: %v", err)
	}
	if e.baseURL != "http://custom:8080" {
		t.Errorf("expected 'http://custom:8080', got %q", e.baseURL)
	}
	if e.model != "my-embedder" {
		t.Errorf("expected 'my-embedder', got %q", e.model)
	}
}

func TestOllama_Dimensions(t *testing.T) {
	e, _ := NewOllama(Config{})
	if dims := e.Dimensions(); dims != 768 {
		t.Errorf("expected 768 dimensions, got %d", dims)
	}
}

func TestOllama_Embed_Success(t *testing.T) {
	server := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
		w.Header().Set("Content-Type", "application/json")
		w.WriteHeader(http.StatusOK)
		w.Write([]byte(`{"embedding": [0.1, 0.2, 0.3]}`))
	}))
	defer server.Close()

	e, _ := NewOllama(Config{BaseURL: server.URL})
	ctx := context.Background()
	vector, err := e.Embed(ctx, "test input")
	if err != nil {
		t.Fatalf("unexpected error: %v", err)
	}
	if len(vector) != 3 {
		t.Errorf("expected 3 dimensions, got %d", len(vector))
	}
	if vector[0] != 0.1 || vector[1] != 0.2 || vector[2] != 0.3 {
		t.Errorf("unexpected vector: %v", vector)
	}
}

func TestOllama_Embed_Error(t *testing.T) {
	server := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
		w.WriteHeader(http.StatusInternalServerError)
		w.Write([]byte("internal error"))
	}))
	defer server.Close()

	e, _ := NewOllama(Config{BaseURL: server.URL})
	ctx := context.Background()
	_, err := e.Embed(ctx, "test input")
	if err == nil {
		t.Fatal("expected error, got nil")
	}
}

func TestMockEmbedder_Default(t *testing.T) {
	m := &MockEmbedder{}
	vector, err := m.Embed(context.Background(), "test")
	if err != nil {
		t.Fatalf("unexpected error: %v", err)
	}
	if len(vector) != 3 {
		t.Errorf("expected 3 dimensions, got %d", len(vector))
	}
	if dims := m.Dimensions(); dims != 3 {
		t.Errorf("expected 3 dimensions, got %d", dims)
	}
}

func TestMockEmbedder_Custom(t *testing.T) {
	m := &MockEmbedder{
		EmbedFunc: func(ctx context.Context, text string) ([]float32, error) {
			return []float32{1.0, 2.0}, nil
		},
		DimensionsFn: func() int {
			return 2
		},
	}
	vector, _ := m.Embed(context.Background(), "test")
	if len(vector) != 2 {
		t.Errorf("expected 2 dimensions, got %d", len(vector))
	}
	if dims := m.Dimensions(); dims != 2 {
		t.Errorf("expected 2 dimensions, got %d", dims)
	}
}