feat: edu-search-service migriert, voice-service/geo-service entfernt
All checks were successful
CI / go-lint (push) Has been skipped
CI / python-lint (push) Has been skipped
CI / nodejs-lint (push) Has been skipped
CI / test-go-school (push) Successful in 28s
CI / test-go-edu-search (push) Successful in 27s
CI / test-python-klausur (push) Successful in 1m45s
CI / test-python-agent-core (push) Successful in 16s
CI / test-nodejs-website (push) Successful in 21s
All checks were successful
CI / go-lint (push) Has been skipped
CI / python-lint (push) Has been skipped
CI / nodejs-lint (push) Has been skipped
CI / test-go-school (push) Successful in 28s
CI / test-go-edu-search (push) Successful in 27s
CI / test-python-klausur (push) Successful in 1m45s
CI / test-python-agent-core (push) Successful in 16s
CI / test-nodejs-website (push) Successful in 21s
- edu-search-service von breakpilot-pwa nach breakpilot-lehrer kopiert (ohne vendor) - opensearch + edu-search-service in docker-compose.yml hinzugefuegt - voice-service aus docker-compose.yml entfernt (jetzt in breakpilot-core) - geo-service aus docker-compose.yml entfernt (nicht mehr benoetigt) - CI/CD: edu-search-service zu Gitea Actions und Woodpecker hinzugefuegt (Go lint, test mit go mod download, build, SBOM) Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
This commit is contained in:
319
edu-search-service/internal/embedding/embedding_test.go
Normal file
319
edu-search-service/internal/embedding/embedding_test.go
Normal file
@@ -0,0 +1,319 @@
|
||||
package embedding
|
||||
|
||||
import (
|
||||
"context"
|
||||
"encoding/json"
|
||||
"net/http"
|
||||
"net/http/httptest"
|
||||
"testing"
|
||||
"time"
|
||||
)
|
||||
|
||||
func TestNewService_Disabled(t *testing.T) {
|
||||
service, err := NewService("none", "", "", "", 1536, false)
|
||||
if err != nil {
|
||||
t.Fatalf("NewService failed: %v", err)
|
||||
}
|
||||
|
||||
if service.IsEnabled() {
|
||||
t.Error("Service should not be enabled")
|
||||
}
|
||||
|
||||
if service.Dimension() != 1536 {
|
||||
t.Errorf("Expected dimension 1536, got %d", service.Dimension())
|
||||
}
|
||||
}
|
||||
|
||||
func TestNewService_DisabledByProvider(t *testing.T) {
|
||||
service, err := NewService("none", "", "", "", 1536, true)
|
||||
if err != nil {
|
||||
t.Fatalf("NewService failed: %v", err)
|
||||
}
|
||||
|
||||
if service.IsEnabled() {
|
||||
t.Error("Service should not be enabled when provider is 'none'")
|
||||
}
|
||||
}
|
||||
|
||||
func TestNewService_OpenAIMissingKey(t *testing.T) {
|
||||
_, err := NewService("openai", "", "", "", 1536, true)
|
||||
if err == nil {
|
||||
t.Error("Expected error for missing OpenAI API key")
|
||||
}
|
||||
}
|
||||
|
||||
func TestNewService_UnknownProvider(t *testing.T) {
|
||||
_, err := NewService("unknown", "", "", "", 1536, true)
|
||||
if err == nil {
|
||||
t.Error("Expected error for unknown provider")
|
||||
}
|
||||
}
|
||||
|
||||
func TestService_EmbedWhenDisabled(t *testing.T) {
|
||||
service, _ := NewService("none", "", "", "", 1536, false)
|
||||
|
||||
_, err := service.Embed(context.Background(), "test text")
|
||||
if err == nil {
|
||||
t.Error("Expected error when embedding with disabled service")
|
||||
}
|
||||
}
|
||||
|
||||
func TestService_EmbedBatchWhenDisabled(t *testing.T) {
|
||||
service, _ := NewService("none", "", "", "", 1536, false)
|
||||
|
||||
_, err := service.EmbedBatch(context.Background(), []string{"test1", "test2"})
|
||||
if err == nil {
|
||||
t.Error("Expected error when embedding batch with disabled service")
|
||||
}
|
||||
}
|
||||
|
||||
// =====================================================
|
||||
// OpenAI Provider Tests with Mock Server
|
||||
// =====================================================
|
||||
|
||||
func TestOpenAIProvider_Embed(t *testing.T) {
|
||||
// Create mock server
|
||||
server := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
||||
// Verify request
|
||||
if r.Method != "POST" {
|
||||
t.Errorf("Expected POST, got %s", r.Method)
|
||||
}
|
||||
if r.Header.Get("Authorization") != "Bearer test-api-key" {
|
||||
t.Errorf("Expected correct Authorization header")
|
||||
}
|
||||
if r.Header.Get("Content-Type") != "application/json" {
|
||||
t.Errorf("Expected Content-Type application/json")
|
||||
}
|
||||
|
||||
// Parse request body
|
||||
var reqBody openAIEmbeddingRequest
|
||||
if err := json.NewDecoder(r.Body).Decode(&reqBody); err != nil {
|
||||
t.Fatalf("Failed to parse request body: %v", err)
|
||||
}
|
||||
|
||||
if reqBody.Model != "text-embedding-3-small" {
|
||||
t.Errorf("Expected model text-embedding-3-small, got %s", reqBody.Model)
|
||||
}
|
||||
|
||||
// Send mock response
|
||||
resp := openAIEmbeddingResponse{
|
||||
Data: []struct {
|
||||
Embedding []float32 `json:"embedding"`
|
||||
Index int `json:"index"`
|
||||
}{
|
||||
{
|
||||
Embedding: make([]float32, 1536),
|
||||
Index: 0,
|
||||
},
|
||||
},
|
||||
}
|
||||
resp.Data[0].Embedding[0] = 0.1
|
||||
resp.Data[0].Embedding[1] = 0.2
|
||||
|
||||
w.Header().Set("Content-Type", "application/json")
|
||||
json.NewEncoder(w).Encode(resp)
|
||||
}))
|
||||
defer server.Close()
|
||||
|
||||
// Create provider with mock server (we need to override the URL)
|
||||
provider := &OpenAIProvider{
|
||||
apiKey: "test-api-key",
|
||||
model: "text-embedding-3-small",
|
||||
dimension: 1536,
|
||||
httpClient: &http.Client{
|
||||
Timeout: 10 * time.Second,
|
||||
},
|
||||
}
|
||||
|
||||
// Note: This test won't actually work with the mock server because
|
||||
// the provider hardcodes the OpenAI URL. This is a structural test.
|
||||
// For real testing, we'd need to make the URL configurable.
|
||||
|
||||
if provider.Dimension() != 1536 {
|
||||
t.Errorf("Expected dimension 1536, got %d", provider.Dimension())
|
||||
}
|
||||
}
|
||||
|
||||
func TestOpenAIProvider_EmbedBatch_EmptyInput(t *testing.T) {
|
||||
provider := NewOpenAIProvider("test-key", "text-embedding-3-small", 1536)
|
||||
|
||||
result, err := provider.EmbedBatch(context.Background(), []string{})
|
||||
if err != nil {
|
||||
t.Errorf("Empty input should not cause error: %v", err)
|
||||
}
|
||||
if result != nil {
|
||||
t.Errorf("Expected nil result for empty input, got %v", result)
|
||||
}
|
||||
}
|
||||
|
||||
// =====================================================
|
||||
// Ollama Provider Tests with Mock Server
|
||||
// =====================================================
|
||||
|
||||
func TestOllamaProvider_Embed(t *testing.T) {
|
||||
// Create mock server
|
||||
mockEmbedding := make([]float32, 384)
|
||||
mockEmbedding[0] = 0.5
|
||||
mockEmbedding[1] = 0.3
|
||||
|
||||
server := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
||||
if r.Method != "POST" {
|
||||
t.Errorf("Expected POST, got %s", r.Method)
|
||||
}
|
||||
if r.URL.Path != "/api/embeddings" {
|
||||
t.Errorf("Expected path /api/embeddings, got %s", r.URL.Path)
|
||||
}
|
||||
|
||||
// Parse request
|
||||
var reqBody ollamaEmbeddingRequest
|
||||
if err := json.NewDecoder(r.Body).Decode(&reqBody); err != nil {
|
||||
t.Fatalf("Failed to parse request: %v", err)
|
||||
}
|
||||
|
||||
if reqBody.Model != "nomic-embed-text" {
|
||||
t.Errorf("Expected model nomic-embed-text, got %s", reqBody.Model)
|
||||
}
|
||||
|
||||
// Send response
|
||||
resp := ollamaEmbeddingResponse{
|
||||
Embedding: mockEmbedding,
|
||||
}
|
||||
|
||||
w.Header().Set("Content-Type", "application/json")
|
||||
json.NewEncoder(w).Encode(resp)
|
||||
}))
|
||||
defer server.Close()
|
||||
|
||||
provider, err := NewOllamaProvider(server.URL, "nomic-embed-text", 384)
|
||||
if err != nil {
|
||||
t.Fatalf("Failed to create provider: %v", err)
|
||||
}
|
||||
|
||||
ctx := context.Background()
|
||||
embedding, err := provider.Embed(ctx, "Test text für Embedding")
|
||||
|
||||
if err != nil {
|
||||
t.Fatalf("Embed failed: %v", err)
|
||||
}
|
||||
|
||||
if len(embedding) != 384 {
|
||||
t.Errorf("Expected 384 dimensions, got %d", len(embedding))
|
||||
}
|
||||
|
||||
if embedding[0] != 0.5 {
|
||||
t.Errorf("Expected first value 0.5, got %f", embedding[0])
|
||||
}
|
||||
}
|
||||
|
||||
func TestOllamaProvider_EmbedBatch(t *testing.T) {
|
||||
callCount := 0
|
||||
|
||||
server := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
||||
callCount++
|
||||
|
||||
mockEmbedding := make([]float32, 384)
|
||||
mockEmbedding[0] = float32(callCount) * 0.1
|
||||
|
||||
resp := ollamaEmbeddingResponse{
|
||||
Embedding: mockEmbedding,
|
||||
}
|
||||
|
||||
w.Header().Set("Content-Type", "application/json")
|
||||
json.NewEncoder(w).Encode(resp)
|
||||
}))
|
||||
defer server.Close()
|
||||
|
||||
provider, err := NewOllamaProvider(server.URL, "nomic-embed-text", 384)
|
||||
if err != nil {
|
||||
t.Fatalf("Failed to create provider: %v", err)
|
||||
}
|
||||
|
||||
ctx := context.Background()
|
||||
texts := []string{"Text 1", "Text 2", "Text 3"}
|
||||
embeddings, err := provider.EmbedBatch(ctx, texts)
|
||||
|
||||
if err != nil {
|
||||
t.Fatalf("EmbedBatch failed: %v", err)
|
||||
}
|
||||
|
||||
if len(embeddings) != 3 {
|
||||
t.Errorf("Expected 3 embeddings, got %d", len(embeddings))
|
||||
}
|
||||
|
||||
// Verify each embedding was called
|
||||
if callCount != 3 {
|
||||
t.Errorf("Expected 3 API calls, got %d", callCount)
|
||||
}
|
||||
}
|
||||
|
||||
func TestOllamaProvider_EmbedServerError(t *testing.T) {
|
||||
server := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
||||
w.WriteHeader(http.StatusInternalServerError)
|
||||
w.Write([]byte("Internal server error"))
|
||||
}))
|
||||
defer server.Close()
|
||||
|
||||
provider, _ := NewOllamaProvider(server.URL, "nomic-embed-text", 384)
|
||||
|
||||
_, err := provider.Embed(context.Background(), "test")
|
||||
if err == nil {
|
||||
t.Error("Expected error for server error response")
|
||||
}
|
||||
}
|
||||
|
||||
func TestOllamaProvider_Dimension(t *testing.T) {
|
||||
provider, _ := NewOllamaProvider("http://localhost:11434", "nomic-embed-text", 768)
|
||||
|
||||
if provider.Dimension() != 768 {
|
||||
t.Errorf("Expected dimension 768, got %d", provider.Dimension())
|
||||
}
|
||||
}
|
||||
|
||||
// =====================================================
|
||||
// Text Truncation Tests
|
||||
// =====================================================
|
||||
|
||||
func TestOllamaProvider_TextTruncation(t *testing.T) {
|
||||
receivedText := ""
|
||||
|
||||
server := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
||||
var reqBody ollamaEmbeddingRequest
|
||||
json.NewDecoder(r.Body).Decode(&reqBody)
|
||||
receivedText = reqBody.Prompt
|
||||
|
||||
resp := ollamaEmbeddingResponse{
|
||||
Embedding: make([]float32, 384),
|
||||
}
|
||||
w.Header().Set("Content-Type", "application/json")
|
||||
json.NewEncoder(w).Encode(resp)
|
||||
}))
|
||||
defer server.Close()
|
||||
|
||||
provider, _ := NewOllamaProvider(server.URL, "nomic-embed-text", 384)
|
||||
|
||||
// Create very long text
|
||||
longText := ""
|
||||
for i := 0; i < 40000; i++ {
|
||||
longText += "a"
|
||||
}
|
||||
|
||||
provider.Embed(context.Background(), longText)
|
||||
|
||||
// Text should be truncated to 30000 chars
|
||||
if len(receivedText) > 30000 {
|
||||
t.Errorf("Expected truncated text <= 30000 chars, got %d", len(receivedText))
|
||||
}
|
||||
}
|
||||
|
||||
// =====================================================
|
||||
// Integration Tests (require actual service)
|
||||
// =====================================================
|
||||
|
||||
func TestOpenAIProvider_Integration(t *testing.T) {
|
||||
// Skip in CI/CD - only run manually with real API key
|
||||
t.Skip("Integration test - requires OPENAI_API_KEY environment variable")
|
||||
|
||||
// provider := NewOpenAIProvider(os.Getenv("OPENAI_API_KEY"), "text-embedding-3-small", 1536)
|
||||
// embedding, err := provider.Embed(context.Background(), "Lehrplan Mathematik Bayern")
|
||||
// ...
|
||||
}
|
||||
Reference in New Issue
Block a user