//
//  Copyright 2026 The InfiniFlow Authors. All Rights Reserved.
//
//  Licensed under the Apache License, Version 2.0 (the "License");
//  you may not use this file except in compliance with the License.
//  You may obtain a copy of the License at
//
//      http://www.apache.org/licenses/LICENSE-2.0
//
//  Unless required by applicable law or agreed to in writing, software
//  distributed under the License is distributed on an "AS IS" BASIS,
//  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
//  See the License for the specific language governing permissions and
//  limitations under the License.
//

package service

import (
	"strings"
	"testing"

	"ragflow/internal/dao"
	"ragflow/internal/entity"
)

func setupSearchServiceTestDB(t *testing.T) {
	t.Helper()
	db := setupServiceTestDB(t)
	if err := db.AutoMigrate(&entity.Search{}); err != nil {
		t.Fatalf("failed to migrate search: %v", err)
	}
	pushServiceDB(t, db)
}

func TestSearchServiceCreateRejectsEmptyName(t *testing.T) {
	setupSearchServiceTestDB(t)

	_, err := NewSearchService().CreateSearch("tenant-1", "   ", nil)
	if err == nil {
		t.Fatal("expected empty name validation error")
	}
	if !strings.Contains(err.Error(), "empty") {
		t.Fatalf("unexpected error: %v", err)
	}
}

func TestSearchServiceUpdateRejectsUnauthorizedSearchID(t *testing.T) {
	setupSearchServiceTestDB(t)

	req := &UpdateSearchRequest{
		Name:         "New Name",
		SearchConfig: map[string]interface{}{},
	}
	_, err := NewSearchService().UpdateSearch("user-2", "invalid_search_id", req)
	if err == nil {
		t.Fatal("expected authorization error")
	}
	if err.Error() != "no authorization" {
		t.Fatalf("unexpected error: %v", err)
	}
}

func TestSearchServiceCreateAndUpdateRoundTrip(t *testing.T) {
	setupSearchServiceTestDB(t)

	created, err := NewSearchService().CreateSearch("tenant-1", "My Search", nil)
	if err != nil {
		t.Fatalf("CreateSearch failed: %v", err)
	}
	if created.SearchID == "" {
		t.Fatal("expected non-empty search_id")
	}

	// A different user must not be able to update it.
	req := &UpdateSearchRequest{
		Name:         "Hijacked Name",
		SearchConfig: map[string]interface{}{},
	}
	_, err = NewSearchService().UpdateSearch("user-2", created.SearchID, req)
	if err == nil || err.Error() != "no authorization" {
		t.Fatalf("expected no authorization, got %v", err)
	}

	// The owner can update name + merge config.
	req = &UpdateSearchRequest{
		Name:         "Updated Name",
		SearchConfig: map[string]interface{}{"summary": true},
	}
	updated, err := NewSearchService().UpdateSearch("tenant-1", created.SearchID, req)
	if err != nil {
		t.Fatalf("owner UpdateSearch failed: %v", err)
	}
	if updated.Name != "Updated Name" {
		t.Fatalf("expected updated name, got %q", updated.Name)
	}
	if updated.SearchConfig["summary"] != true {
		t.Fatalf("expected merged search_config, got %#v", updated.SearchConfig)
	}

	persisted, err := dao.NewSearchDAO().GetByID(created.SearchID)
	if err != nil {
		t.Fatalf("get updated search: %v", err)
	}
	if persisted.Name != "Updated Name" {
		t.Fatalf("expected persisted name, got %q", persisted.Name)
	}
}