package engine

import (
	"testing"

	"git.kingecg.top/kingecg/gomog/pkg/types"
)

func TestMatchFilter(t *testing.T) {
	tests := []struct {
		name     string
		doc      map[string]interface{}
		filter   types.Filter
		expected bool
	}{
		{
			name:     "empty filter",
			doc:      map[string]interface{}{"name": "Alice"},
			filter:   nil,
			expected: true,
		},
		{
			name:     "simple equality",
			doc:      map[string]interface{}{"name": "Alice", "age": 25},
			filter:   types.Filter{"name": "Alice"},
			expected: true,
		},
		{
			name:     "simple equality no match",
			doc:      map[string]interface{}{"name": "Alice", "age": 25},
			filter:   types.Filter{"name": "Bob"},
			expected: false,
		},
		{
			name:     "greater than",
			doc:      map[string]interface{}{"age": 30},
			filter:   types.Filter{"age": types.Filter{"$gt": 25}},
			expected: true,
		},
		{
			name:     "less than",
			doc:      map[string]interface{}{"age": 20},
			filter:   types.Filter{"age": types.Filter{"$lt": 25}},
			expected: true,
		},
		{
			name:     "in array",
			doc:      map[string]interface{}{"status": "active"},
			filter:   types.Filter{"status": types.Filter{"$in": []interface{}{"active", "pending"}}},
			expected: true,
		},
		{
			name:     "exists",
			doc:      map[string]interface{}{"name": "Alice"},
			filter:   types.Filter{"name": types.Filter{"$exists": true}},
			expected: true,
		},
		{
			name:     "not exists",
			doc:      map[string]interface{}{"name": "Alice"},
			filter:   types.Filter{"email": types.Filter{"$exists": false}},
			expected: true,
		},
	}

	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			result := MatchFilter(tt.doc, tt.filter)
			if result != tt.expected {
				t.Errorf("MatchFilter() = %v, want %v", result, tt.expected)
			}
		})
	}
}

func TestApplyUpdate(t *testing.T) {
	tests := []struct {
		name     string
		data     map[string]interface{}
		update   types.Update
		expected map[string]interface{}
	}{
		{
			name: "set field",
			data: map[string]interface{}{"name": "Alice", "age": 25},
			update: types.Update{
				Set: map[string]interface{}{"age": 26},
			},
			expected: map[string]interface{}{"name": "Alice", "age": 26},
		},
		{
			name: "unset field",
			data: map[string]interface{}{"name": "Alice", "email": "alice@example.com"},
			update: types.Update{
				Unset: map[string]interface{}{"email": 1},
			},
			expected: map[string]interface{}{"name": "Alice"},
		},
		{
			name: "increment field",
			data: map[string]interface{}{"count": 5},
			update: types.Update{
				Inc: map[string]interface{}{"count": 3},
			},
			expected: map[string]interface{}{"count": 8},
		},
		{
			name: "multiply field",
			data: map[string]interface{}{"price": 100},
			update: types.Update{
				Mul: map[string]interface{}{"price": 0.9},
			},
			expected: map[string]interface{}{"price": 90},
		},
	}

	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			result := applyUpdate(tt.data, tt.update, false)

			// 简单比较（实际应该深度比较）
			for k, v := range tt.expected {
				if result[k] != v {
					t.Errorf("applyUpdate()[%s] = %v, want %v", k, result[k], v)
				}
			}
		})
	}
}