// Copyright 2023 PingCAP, Inc.
//
// 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 fixcontrol_test

import (
	"testing"

	"github.com/pingcap/tidb/pkg/planner/util/fixcontrol"
	"github.com/pingcap/tidb/pkg/testkit"
	"github.com/pingcap/tidb/pkg/testkit/testdata"
	"github.com/stretchr/testify/require"
	"golang.org/x/exp/maps"
)

type resultForSingleFix struct {
	ValueInMap string
	GetStr     string
	GetBool    bool
	GetInt     int64
	GetFloat   float64
}

func getTestResultForSingleFix(fixControlMap map[uint64]string, key uint64) *resultForSingleFix {
	result := &resultForSingleFix{}
	result.ValueInMap = fixControlMap[key]
	result.GetBool = fixcontrol.GetBoolWithDefault(fixControlMap, key, false)
	result.GetStr = fixcontrol.GetStrWithDefault(fixControlMap, key, "default")
	result.GetInt = fixcontrol.GetIntWithDefault(fixControlMap, key, 12345)
	result.GetFloat = fixcontrol.GetFloatWithDefault(fixControlMap, key, 1234.5)
	return result
}

func TestFixControl(t *testing.T) {
	store := testkit.CreateMockStore(t)
	tk := testkit.NewTestKit(t, store)
	tk.MustExec("use test;")
	s := tk.Session()
	var input []string
	var output []struct {
		SQL        string
		FixControl map[uint64]*resultForSingleFix
		Error      string
		Warnings   [][]any
		Variable   []string
	}

	integrationSuiteData := testDataMap["fix_control_suite"]
	integrationSuiteData.LoadTestCases(t, &input, &output)
	for i, tt := range input {
		tk.MustExec("set @@tidb_opt_fix_control = \"\"")
		err := tk.ExecToErr(tt)
		var errStr string
		if err != nil {
			errStr = err.Error()
		}
		warning := tk.MustQuery("show warnings").Sort().Rows()
		rows := testdata.ConvertRowsToStrings(tk.MustQuery("select @@tidb_opt_fix_control").Sort().Rows())
		testdata.OnRecord(func() {
			output[i].SQL = tt
			keys := maps.Keys(s.GetSessionVars().OptimizerFixControl)
			output[i].FixControl = make(map[uint64]*resultForSingleFix, len(keys))
			for _, key := range keys {
				output[i].FixControl[key] = getTestResultForSingleFix(s.GetSessionVars().OptimizerFixControl, key)
			}
			output[i].Error = errStr
			output[i].Warnings = warning
			output[i].Variable = rows
		})
		keys := maps.Keys(s.GetSessionVars().OptimizerFixControl)
		for _, key := range keys {
			require.Equal(t, output[i].FixControl[key], getTestResultForSingleFix(s.GetSessionVars().OptimizerFixControl, key))
		}
		require.Equal(t, output[i].Error, errStr)
		require.Equal(t, output[i].Warnings, warning)
		require.Equal(t, output[i].Variable, rows)
	}
}