package analysis

import (
	"testing"

	metricsapi "github.com/keptn/lifecycle-toolkit/metrics-operator/api/v1"
	"github.com/keptn/lifecycle-toolkit/metrics-operator/controllers/common/analysis/fake"
	"github.com/keptn/lifecycle-toolkit/metrics-operator/controllers/common/analysis/types"
	"github.com/stretchr/testify/require"
)

func TestAnalysisEvaluator_Evaluate(t *testing.T) {
	tests := []struct {
		name            string
		values          map[string]metricsapi.ProviderResult
		a               metricsapi.AnalysisDefinition
		want            types.AnalysisResult
		mockedEvaluator IObjectiveEvaluator
	}{
		{
			name:   "no objectives",
			values: map[string]metricsapi.ProviderResult{},
			a: metricsapi.AnalysisDefinition{
				Spec: metricsapi.AnalysisDefinitionSpec{
					Objectives: []metricsapi.Objective{},
				},
			},
			want: types.AnalysisResult{
				TotalScore:       0.0,
				MaximumScore:     0.0,
				Pass:             true,
				Warning:          false,
				ObjectiveResults: []types.ObjectiveResult{},
			},
			mockedEvaluator: nil,
		},
		{
			name:   "pass scenario",
			values: map[string]metricsapi.ProviderResult{},
			a: metricsapi.AnalysisDefinition{
				Spec: metricsapi.AnalysisDefinitionSpec{
					Objectives: []metricsapi.Objective{
						{
							Weight: 10,
						},
					},
					TotalScore: metricsapi.TotalScore{
						PassPercentage:    80,
						WarningPercentage: 50,
					},
				},
			},
			want: types.AnalysisResult{
				TotalScore:   10.0,
				MaximumScore: 10.0,
				Pass:         true,
				Warning:      false,
				ObjectiveResults: []types.ObjectiveResult{
					{
						Result: types.TargetResult{},
						Value:  5.0,
						Query:  "query",
						Score:  10.0,
						Error:  nil,
					},
				},
			},
			mockedEvaluator: &fake.IObjectiveEvaluatorMock{
				EvaluateFunc: func(values map[string]metricsapi.ProviderResult, objective *metricsapi.Objective) types.ObjectiveResult {
					return types.ObjectiveResult{
						Result: types.TargetResult{},
						Query:  "query",
						Value:  5.0,
						Score:  10.0,
						Error:  nil,
					}
				},
			},
		},
		{
			name:   "pass scenario - multiple objectives",
			values: map[string]metricsapi.ProviderResult{},
			a: metricsapi.AnalysisDefinition{
				Spec: metricsapi.AnalysisDefinitionSpec{
					Objectives: []metricsapi.Objective{
						{
							Weight: 10,
						},
						{
							Weight: 10,
						},
					},
					TotalScore: metricsapi.TotalScore{
						PassPercentage:    80,
						WarningPercentage: 50,
					},
				},
			},
			want: types.AnalysisResult{
				TotalScore:   20.0,
				MaximumScore: 20.0,
				Pass:         true,
				Warning:      false,
				ObjectiveResults: []types.ObjectiveResult{
					{
						Result: types.TargetResult{},
						Value:  5.0,
						Query:  "query",
						Score:  10.0,
						Error:  nil,
					},
					{
						Result: types.TargetResult{},
						Value:  5.0,
						Query:  "query",
						Score:  10.0,
						Error:  nil,
					},
				},
			},
			mockedEvaluator: &fake.IObjectiveEvaluatorMock{
				EvaluateFunc: func(values map[string]metricsapi.ProviderResult, objective *metricsapi.Objective) types.ObjectiveResult {
					return types.ObjectiveResult{
						Result: types.TargetResult{},
						Value:  5.0,
						Query:  "query",
						Score:  10.0,
						Error:  nil,
					}
				},
			},
		},
		{
			name:   "warning scenario",
			values: map[string]metricsapi.ProviderResult{},
			a: metricsapi.AnalysisDefinition{
				Spec: metricsapi.AnalysisDefinitionSpec{
					Objectives: []metricsapi.Objective{
						{
							Weight: 10,
						},
					},
					TotalScore: metricsapi.TotalScore{
						PassPercentage:    80,
						WarningPercentage: 50,
					},
				},
			},
			want: types.AnalysisResult{
				TotalScore:   5.0,
				MaximumScore: 10.0,
				Pass:         false,
				Warning:      true,
				ObjectiveResults: []types.ObjectiveResult{
					{
						Result: types.TargetResult{},
						Query:  "query",
						Value:  5.0,
						Score:  5.0,
						Error:  nil,
					},
				},
			},
			mockedEvaluator: &fake.IObjectiveEvaluatorMock{
				EvaluateFunc: func(values map[string]metricsapi.ProviderResult, objective *metricsapi.Objective) types.ObjectiveResult {
					return types.ObjectiveResult{
						Result: types.TargetResult{},
						Query:  "query",
						Value:  5.0,
						Score:  5.0,
						Error:  nil,
					}
				},
			},
		},
		{
			name:   "fail scenario",
			values: map[string]metricsapi.ProviderResult{},
			a: metricsapi.AnalysisDefinition{
				Spec: metricsapi.AnalysisDefinitionSpec{
					Objectives: []metricsapi.Objective{
						{
							Weight: 10,
						},
					},
					TotalScore: metricsapi.TotalScore{
						PassPercentage:    80,
						WarningPercentage: 50,
					},
				},
			},
			want: types.AnalysisResult{
				TotalScore:   0.0,
				MaximumScore: 10.0,
				Pass:         false,
				Warning:      false,
				ObjectiveResults: []types.ObjectiveResult{
					{
						Result: types.TargetResult{},
						Value:  5.0,
						Query:  "query",
						Score:  0.0,
						Error:  nil,
					},
				},
			},
			mockedEvaluator: &fake.IObjectiveEvaluatorMock{
				EvaluateFunc: func(values map[string]metricsapi.ProviderResult, objective *metricsapi.Objective) types.ObjectiveResult {
					return types.ObjectiveResult{
						Result: types.TargetResult{},
						Value:  5.0,
						Query:  "query",
						Score:  0.0,
						Error:  nil,
					}
				},
			},
		},
		{
			name:   "fail scenario - key objective failed",
			values: map[string]metricsapi.ProviderResult{},
			a: metricsapi.AnalysisDefinition{
				Spec: metricsapi.AnalysisDefinitionSpec{
					Objectives: []metricsapi.Objective{
						{
							Weight:       10,
							KeyObjective: false,
						},
						{
							Weight:       1,
							KeyObjective: true,
						},
					},
					TotalScore: metricsapi.TotalScore{
						PassPercentage:    80,
						WarningPercentage: 50,
					},
				},
			},
			want: types.AnalysisResult{
				TotalScore:   10.0,
				MaximumScore: 11.0,
				Pass:         false,
				Warning:      false,
				ObjectiveResults: []types.ObjectiveResult{
					{
						Result: types.TargetResult{},
						Value:  5.0,
						Query:  "query",
						Score:  10.0,
						Error:  nil,
					},
					{
						Result: types.TargetResult{},
						Value:  5.0,
						Query:  "query",
						Score:  0.0,
						Error:  nil,
					},
				},
			},
			mockedEvaluator: &fake.IObjectiveEvaluatorMock{
				EvaluateFunc: func(values map[string]metricsapi.ProviderResult, objective *metricsapi.Objective) types.ObjectiveResult {
					if objective.KeyObjective {
						return types.ObjectiveResult{
							Result: types.TargetResult{},
							Value:  5.0,
							Query:  "query",
							Score:  0.0,
							Error:  nil,
						}
					}
					return types.ObjectiveResult{
						Result: types.TargetResult{},
						Value:  5.0,
						Query:  "query",
						Score:  10.0,
						Error:  nil,
					}
				},
			},
		},
	}

	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			ae := NewAnalysisEvaluator(tt.mockedEvaluator)
			require.Equal(t, tt.want, ae.Evaluate(tt.values, &tt.a))
		})
	}
}