github.com/1aal/kubeblocks@v0.0.0-20231107070852-e1c03e598921/controllers/apps/configuration/revision_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 configuration

import (
	"fmt"
	"testing"

	"github.com/stretchr/testify/assert"

	appsv1alpha1 "github.com/1aal/kubeblocks/apis/apps/v1alpha1"
	"github.com/1aal/kubeblocks/pkg/configuration/core"
	"github.com/1aal/kubeblocks/pkg/constant"
	"github.com/1aal/kubeblocks/pkg/controller/builder"
	intctrlutil "github.com/1aal/kubeblocks/pkg/controllerutil"
)

func TestGcConfigRevision(t *testing.T) {
	cm := builder.NewConfigMapBuilder("default", "test").
		AddAnnotations(core.GenerateRevisionPhaseKey("1"), "Finished").
		AddAnnotations(core.GenerateRevisionPhaseKey("2"), "Init").
		AddAnnotations(core.GenerateRevisionPhaseKey("3"), "Finished").
		AddAnnotations(core.GenerateRevisionPhaseKey("4"), "Finished").
		GetObject()
	revisions := GcRevision(cm.GetAnnotations())
	assert.Equal(t, 0, len(revisions))

	cm = builder.NewConfigMapBuilder("default", "test").
		AddAnnotations(core.GenerateRevisionPhaseKey("1"), "Finished").
		AddAnnotations(core.GenerateRevisionPhaseKey("2"), "Init").
		AddAnnotations(core.GenerateRevisionPhaseKey("3"), "Finished").
		AddAnnotations(core.GenerateRevisionPhaseKey("4"), "Finished").
		AddAnnotations(core.GenerateRevisionPhaseKey("5"), "Finished").
		AddAnnotations(core.GenerateRevisionPhaseKey("6"), "Finished").
		AddAnnotations(core.GenerateRevisionPhaseKey("7"), "Finished").
		AddAnnotations(core.GenerateRevisionPhaseKey("8"), "Finished").
		AddAnnotations(core.GenerateRevisionPhaseKey("9"), "Finished").
		AddAnnotations(core.GenerateRevisionPhaseKey("10"), "Finished").
		AddAnnotations(core.GenerateRevisionPhaseKey("11"), "Finished").
		AddAnnotations(core.GenerateRevisionPhaseKey("12"), `{"Phase":"Finished","Revision":"12","Policy":"","ExecResult":"","SucceedCount":0,"ExpectedCount":0,"Retry":false,"Failed":false,"Message":"the configuration file has not been modified, skip reconfigure"}`).
		GetObject()

	assert.Equal(t, 12, len(RetrieveRevision(cm.GetAnnotations())))

	revisions = GcRevision(cm.GetAnnotations())
	assert.Equal(t, 2, len(revisions))
	assert.Equal(t, string(appsv1alpha1.CInitPhase), string(revisions[1].Phase))
	assert.Equal(t, string(appsv1alpha1.CFinishedPhase), string(revisions[0].Phase))

	GcConfigRevision(cm)
	assert.Equal(t, 10, len(RetrieveRevision(cm.GetAnnotations())))
}

func TestParseRevision(t *testing.T) {
	type args struct {
		revision string
		phase    string
	}
	tests := []struct {
		name    string
		args    args
		want    ConfigurationRevision
		wantErr bool
	}{{
		name: "test",
		args: args{
			revision: "12absdl",
			phase:    "Init",
		},
		want:    ConfigurationRevision{},
		wantErr: true,
	}, {
		name: "test",
		args: args{
			revision: "120000",
			phase:    "Pending",
		},
		want: ConfigurationRevision{
			StrRevision: "120000",
			Revision:    120000,
			Phase:       appsv1alpha1.CPendingPhase,
			Result: intctrlutil.Result{
				Phase:    appsv1alpha1.CPendingPhase,
				Revision: "120000",
			},
		},
		wantErr: false,
	}, {
		name: "test",
		args: args{
			revision: "",
			phase:    "Init",
		},
		want:    ConfigurationRevision{},
		wantErr: true,
	}}
	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			got, err := parseRevision(tt.args.revision, tt.args.phase)
			if (err != nil) != tt.wantErr {
				assert.Error(t, err, fmt.Sprintf("parseRevision(%v, %v)", tt.args.revision, tt.args.phase))
			} else {
				assert.Equalf(t, tt.want, got, "parseRevision(%v, %v)", tt.args.revision, tt.args.phase)
			}
		})
	}
}

func TestGetCurrentRevision(t *testing.T) {
	type args struct {
		annotations map[string]string
	}
	tests := []struct {
		name string
		args args
		want string
	}{{
		name: "test",
		args: args{
			annotations: map[string]string{},
		},
		want: "",
	}, {
		name: "test",
		args: args{
			annotations: map[string]string{"abcd": "finished"},
		},
		want: "",
	}, {
		name: "test",
		args: args{
			annotations: map[string]string{constant.ConfigurationRevision: "mytest"},
		},
		want: "mytest",
	}}
	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			assert.Equalf(t, tt.want, GetCurrentRevision(tt.args.annotations), "GetCurrentRevision(%v)", tt.args.annotations)
		})
	}
}

func TestGetLastRevision(t *testing.T) {
	type args struct {
		annotations map[string]string
		revision    int64
	}
	tests := []struct {
		name  string
		args  args
		want  ConfigurationRevision
		want1 bool
	}{{
		name: "test",
		args: args{
			annotations: map[string]string{
				core.GenerateRevisionPhaseKey("1"): "Finished",
				core.GenerateRevisionPhaseKey("2"): "Running",
			},
			revision: 2,
		},
		want: ConfigurationRevision{
			Revision:    2,
			StrRevision: "2",
			Phase:       appsv1alpha1.CRunningPhase,
			Result: intctrlutil.Result{
				Phase:    appsv1alpha1.CRunningPhase,
				Revision: "2",
			},
		},
		want1: true,
	}, {
		name: "test",
		args: args{
			annotations: map[string]string{
				core.GenerateRevisionPhaseKey("1"): "Finished",
				core.GenerateRevisionPhaseKey("2"): "Running",
			},
			revision: 3,
		},
		want1: false,
	}}
	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			got, got1 := GetLastRevision(tt.args.annotations, tt.args.revision)
			assert.Equalf(t, tt.want, got, "GetLastRevision(%v, %v)", tt.args.annotations, tt.args.revision)
			assert.Equalf(t, tt.want1, got1, "GetLastRevision(%v, %v)", tt.args.annotations, tt.args.revision)
		})
	}
}