feat(preset): add preset_test.go

Signed-off-by: Vaughn Dice <[email protected]>

Author: vdice

internal/preset/preset_test.go

@@ -0,0 +1,104 @@
+package preset_test
+
+import (
+	"testing"
+
+	"github.com/spf13/afero"
+	"github.com/spinkube/runtime-class-manager/internal/preset"
+	tests "github.com/spinkube/runtime-class-manager/tests/node-installer"
+	"github.com/stretchr/testify/require"
+)
+
+func Test_WithSetup(t *testing.T) {
+	type args struct {
+		settings preset.Settings
+		hostFs   afero.Fs
+	}
+	tests := []struct {
+		name         string
+		args         args
+		wantErr      bool
+		wantFile     string
+		wantContents string
+	}{
+		{
+			"rke2_err",
+			args{
+				preset.RKE2,
+				tests.FixtureFs("../../testdata/node-installer/distros/unsupported"),
+			},
+			true,
+			"",
+			"",
+		},
+		{
+			"rke2_config_exists",
+			args{
+				preset.RKE2,
+				tests.FixtureFs("../../testdata/node-installer/containerd/rke2-existing-config-tmpl"),
+			},
+			false,
+			"/var/lib/rancher/rke2/agent/etc/containerd/config.toml.tmpl",
+			"version = 2\npreexisting-config = true",
+		},
+		{
+			"rke2_config_is_created",
+			args{
+				preset.RKE2,
+				tests.FixtureFs("../../testdata/node-installer/distros/rke2"),
+			},
+			false,
+			"/var/lib/rancher/rke2/agent/etc/containerd/config.toml.tmpl",
+			"version = 2",
+		},
+		{
+			"k3s",
+			args{
+				preset.K3s,
+				tests.FixtureFs("../../testdata/node-installer/distros/k3s"),
+			},
+			false,
+			"/var/lib/rancher/k3s/agent/etc/containerd/config.toml.tmpl",
+			"version = 2",
+		},
+		{
+			"k0s",
+			args{
+				preset.K0s,
+				tests.FixtureFs("../../testdata/node-installer/distros/k0s"),
+			},
+			false,
+			"/etc/k0s/containerd.d/config.toml",
+			"",
+		},
+		{
+			"microk8s",
+			args{
+				preset.MicroK8s,
+				tests.FixtureFs("../../testdata/node-installer/distros/microk8s"),
+			},
+			false,
+			"/var/snap/microk8s/current/args/containerd-template.toml",
+			"",
+		},
+	}
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+			err := tt.args.settings.Setup(
+				preset.Env{
+					ConfigPath: tt.args.settings.ConfigPath,
+					HostFs:     tt.args.hostFs,
+				},
+			)
+
+			if tt.wantErr {
+				require.Error(t, err)
+			} else {
+				require.NoError(t, err)
+				bytes, err := afero.ReadFile(tt.args.hostFs, tt.wantFile)
+				require.NoError(t, err)
+				require.Equal(t, tt.wantContents, string(bytes))
+			}
+		})
+	}
+}

testdata/node-installer/containerd/rke2-existing-config-tmpl/var/lib/rancher/rke2/agent/etc/containerd/config.toml.tmpl

@@ -0,0 +1,2 @@
+version = 2
+preexisting-config = true

testdata/node-installer/containerd/rke2-only-base-config-exists/var/lib/rancher/rke2/agent/etc/containerd/config.toml

@@ -0,0 +1 @@
+version = 2

testdata/node-installer/distros/k3s/var/lib/rancher/k3s/agent/etc/containerd/config.toml

@@ -0,0 +1 @@
+version = 2

testdata/node-installer/distros/rke2/var/lib/rancher/rke2/agent/etc/containerd/config.toml

@@ -0,0 +1 @@
+version = 2