provider-k3s/pkg/provider/provider_test.go

package provider

import (
	"bytes"
	"testing"

	"github.com/kairos-io/kairos-sdk/clusterplugin"
)

func Test_parseOptions(t *testing.T) {
	tests := []struct {
		name                 string
		cluster              clusterplugin.Cluster
		expectedOptions      []byte
		expectedProxyOptions []byte
		expectedUserOptions  []byte
	}{
		{
			name:                 "Empty input",
			cluster:              clusterplugin.Cluster{},
			expectedOptions:      []byte(`{}`),
			expectedProxyOptions: []byte(`null`),
			expectedUserOptions:  []byte(`null`),
		},
		{
			name: "Init: Standard",
			cluster: clusterplugin.Cluster{
				ClusterToken:     "token",
				ControlPlaneHost: "localhost",
				Role:             "init",
			},
			expectedOptions:      []byte(`{"cluster-init":true,"token":"token","tls-san":["localhost"]}`),
			expectedProxyOptions: []byte(`null`),
			expectedUserOptions:  []byte(`null`),
		},
		{
			name: "Init: 2-Node",
			cluster: clusterplugin.Cluster{
				ClusterToken:     "token",
				ControlPlaneHost: "localhost",
				Role:             "init",
				ProviderOptions: map[string]string{
					"cluster-init":       "no",
					"datastore-endpoint": "localhost:2379",
				},
			},
			expectedOptions:      []byte(`{"cluster-init":false,"token":"token","tls-san":["localhost"],"datastore-endpoint":"localhost:2379"}`),
			expectedProxyOptions: []byte(`null`),
			expectedUserOptions:  []byte(`null`),
		},
		{
			name: "Control Plane",
			cluster: clusterplugin.Cluster{
				ClusterToken:     "token",
				ControlPlaneHost: "localhost",
				Role:             "controlplane",
			},
			expectedOptions:      []byte(`{"token":"token","server":"https://localhost:6443","tls-san":["localhost"]}`),
			expectedProxyOptions: []byte(`null`),
			expectedUserOptions:  []byte(`null`),
		},
		{
			name: "Worker",
			cluster: clusterplugin.Cluster{
				ClusterToken:     "token",
				ControlPlaneHost: "localhost",
				Role:             "worker",
			},
			expectedOptions:      []byte(`{"token":"token","server":"https://localhost:6443"}`),
			expectedProxyOptions: []byte(`null`),
			expectedUserOptions:  []byte(`{}`),
		},
	}
	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			options, proxyOptions, userOptions := parseOptions(tt.cluster)
			if !bytes.Equal(options, tt.expectedOptions) {
				t.Errorf("parseOptions() options = %v, want %v", string(options), string(tt.expectedOptions))
			}
			if !bytes.Equal(proxyOptions, tt.expectedProxyOptions) {
				t.Errorf("parseOptions() proxyOptions = %v, want %v", string(proxyOptions), string(tt.expectedProxyOptions))
			}
			if !bytes.Equal(userOptions, tt.expectedUserOptions) {
				t.Errorf("parseOptions() userOptions = %v, want %v", string(userOptions), string(tt.expectedUserOptions))
			}
		})
	}
}