github.com/1aal/kubeblocks@v0.0.0-20231107070852-e1c03e598921/pkg/configuration/validate/config_validate_test.go

/*
Copyright (C) 2022-2023 ApeCloud Co., Ltd

This file is part of KubeBlocks project

This program is free software: you can redistribute it and/or modify
it under the terms of the GNU Affero General Public License as published by
the Free Software Foundation, either version 3 of the License, or
(at your option) any later version.

This program is distributed in the hope that it will be useful
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
GNU Affero General Public License for more details.

You should have received a copy of the GNU Affero General Public License
along with this program.  If not, see <http://www.gnu.org/licenses/>.
*/

package validate

import (
	"errors"
	"testing"

	"github.com/stretchr/testify/require"

	appsv1alpha1 "github.com/1aal/kubeblocks/apis/apps/v1alpha1"
	"github.com/1aal/kubeblocks/test/testdata"
)

var fromTestData = func(fileName string) string {
	content, err := testdata.GetTestDataFileContent(fileName)
	if err != nil {
		panic(err)
	}
	return string(content)
}

var newFakeConfConstraint = func(cueFile string, cfgFormatter appsv1alpha1.CfgFileFormat) *appsv1alpha1.ConfigConstraintSpec {
	return &appsv1alpha1.ConfigConstraintSpec{
		ConfigurationSchema: &appsv1alpha1.CustomParametersValidation{
			CUE: fromTestData(cueFile),
		},
		FormatterConfig: &appsv1alpha1.FormatterConfig{
			Format: cfgFormatter,
		},
	}
}

func TestSchemaValidatorWithCue(t *testing.T) {
	type args struct {
		cueFile    string
		configFile string
		format     appsv1alpha1.CfgFileFormat
		options    []ValidatorOptions
	}
	tests := []struct {
		name string
		args args
		err  error
	}{{
		name: "mongod_test",
		args: args{
			cueFile:    "cue_testdata/mongod.cue",
			configFile: "cue_testdata/mongod.conf",
			format:     appsv1alpha1.YAML,
		},
		err: nil,
	}, {
		name: "test_wesql",
		args: args{
			cueFile:    "cue_testdata/wesql.cue",
			configFile: "cue_testdata/wesql.cnf",
			format:     appsv1alpha1.Ini,
		},
		err: nil,
	}, {
		name: "test_pg14",
		args: args{
			cueFile:    "cue_testdata/pg14.cue",
			configFile: "cue_testdata/pg14.conf",
			format:     appsv1alpha1.Properties,
		},
		err: nil,
	}, {
		name: "test_ck",
		args: args{
			cueFile:    "cue_testdata/clickhouse.cue",
			configFile: "cue_testdata/clickhouse.xml",
			format:     appsv1alpha1.XML,
		},
		err: nil,
	}, {
		name: "test_mysql",
		args: args{
			cueFile:    "cue_testdata/mysql.cue",
			configFile: "cue_testdata/mysql.cnf",
			format:     appsv1alpha1.Ini,
		},
		err: nil,
	}, {
		name: "test_failed",
		args: args{
			cueFile:    "cue_testdata/mysql.cue",
			configFile: "cue_testdata/mysql_err.cnf",
			format:     appsv1alpha1.Ini,
		},
		err: errors.New(`failed to render cue template configure: [mysqld.innodb_autoinc_lock_mode: 3 errors in empty disjunction:
mysqld.innodb_autoinc_lock_mode: conflicting values 0 and 100:
    31:35
mysqld.innodb_autoinc_lock_mode: conflicting values 1 and 100:
    31:39
mysqld.innodb_autoinc_lock_mode: conflicting values 2 and 100:
    31:43]`),
	}, {
		name: "configmap_key_filter",
		args: args{
			cueFile:    "cue_testdata/mysql.cue",
			configFile: "cue_testdata/mysql_err.cnf",
			format:     appsv1alpha1.Ini,
			options:    []ValidatorOptions{WithKeySelector([]string{"key2", "key3"})},
		},
	}}

	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			validator := NewConfigValidator(newFakeConfConstraint(tt.args.cueFile, tt.args.format), tt.args.options...)
			require.NotNil(t, validator)
			require.Equal(t, tt.err, validator.Validate(
				map[string]string{
					"key": fromTestData(tt.args.configFile),
				}))
		})
	}
}

func TestSchemaValidatorWithSelector(t *testing.T) {
	validator := NewConfigValidator(newFakeConfConstraint("cue_testdata/mysql.cue", appsv1alpha1.Ini))
	require.NotNil(t, validator)
	require.ErrorContains(t, validator.Validate(
		map[string]string{
			"normal_key":   fromTestData("cue_testdata/mysql.cnf"),
			"abnormal_key": fromTestData("cue_testdata/mysql_err.cnf"),
		}), "[mysqld.innodb_autoinc_lock_mode: 3 errors in empty disjunction")

	validator = NewConfigValidator(newFakeConfConstraint("cue_testdata/mysql.cue", appsv1alpha1.Ini), WithKeySelector([]string{}))
	require.NotNil(t, validator)
	require.ErrorContains(t, validator.Validate(
		map[string]string{
			"normal_key":   fromTestData("cue_testdata/mysql.cnf"),
			"abnormal_key": fromTestData("cue_testdata/mysql_err.cnf"),
		}), "[mysqld.innodb_autoinc_lock_mode: 3 errors in empty disjunction")

	validator = NewConfigValidator(newFakeConfConstraint("cue_testdata/mysql.cue", appsv1alpha1.Ini), WithKeySelector([]string{"normal_key"}))
	require.NotNil(t, validator)
	require.Nil(t, validator.Validate(
		map[string]string{
			"normal_key":   fromTestData("cue_testdata/mysql.cnf"),
			"abnormal_key": fromTestData("cue_testdata/mysql_err.cnf"),
		}))
}

func TestSchemaValidatorWithOpenSchema(t *testing.T) {
	type args struct {
		cueFile        string
		configFile     string
		format         appsv1alpha1.CfgFileFormat
		SchemaTypeName string
	}
	tests := []struct {
		name string
		args args
		err  error
	}{{
		name: "test_wesql",
		args: args{
			cueFile:    "cue_testdata/mysql.cue",
			configFile: "cue_testdata/mysql.cnf",
			format:     appsv1alpha1.Ini,
		},
		err: nil,
	}}

	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			tplConstraint := newFakeConfConstraint(tt.args.cueFile, tt.args.format)
			validator := &schemaValidator{
				typeName: tt.args.SchemaTypeName,
				cfgType:  tplConstraint.FormatterConfig.Format,
				schema:   tplConstraint.ConfigurationSchema.Schema,
			}
			require.Equal(t, tt.err, validator.Validate(
				map[string]string{
					"key": fromTestData(tt.args.configFile),
				}))
		})
	}
}