talos/pkg/machinery/config/configpatcher/apply_test.go

// This Source Code Form is subject to the terms of the Mozilla Public
// License, v. 2.0. If a copy of the MPL was not distributed with this
// file, You can obtain one at http://mozilla.org/MPL/2.0/.

//nolint:dupl
package configpatcher_test

import (
	_ "embed"
	"testing"

	"github.com/stretchr/testify/assert"
	"github.com/stretchr/testify/require"

	"github.com/siderolabs/talos/pkg/machinery/config/configloader"
	"github.com/siderolabs/talos/pkg/machinery/config/configpatcher"
)

//go:embed testdata/apply/config.yaml
var config []byte

//go:embed testdata/apply/expected.yaml
var expected []byte

//go:embed testdata/multidoc/config.yaml
var configMultidoc []byte

//go:embed testdata/multidoc/expected.yaml
var expectedMultidoc []byte

func TestApply(t *testing.T) {
	patches, err := configpatcher.LoadPatches([]string{
		"@testdata/apply/strategic1.yaml",
		"@testdata/apply/jsonpatch1.yaml",
		"@testdata/apply/jsonpatch2.yaml",
		"@testdata/apply/strategic2.yaml",
		"@testdata/apply/strategic3.yaml",
	})
	require.NoError(t, err)

	cfg, err := configloader.NewFromBytes(config)
	require.NoError(t, err)

	for _, tt := range []struct {
		name  string
		input configpatcher.Input
	}{
		{
			name:  "WithConfig",
			input: configpatcher.WithConfig(cfg),
		},
		{
			name:  "WithBytes",
			input: configpatcher.WithBytes(config),
		},
	} {
		t.Run(tt.name, func(t *testing.T) {
			out, err := configpatcher.Apply(tt.input, patches)
			require.NoError(t, err)

			bytes, err := out.Bytes()
			require.NoError(t, err)

			assert.Equal(t, string(expected), string(bytes))
		})
	}
}

func TestApplyMultiDocFail(t *testing.T) {
	patches, err := configpatcher.LoadPatches([]string{
		"@testdata/multidoc/jsonpatch.yaml",
		"@testdata/multidoc/strategic1.yaml",
	})
	require.NoError(t, err)

	cfg, err := configloader.NewFromBytes(configMultidoc)
	require.NoError(t, err)

	for _, tt := range []struct {
		name  string
		input configpatcher.Input
	}{
		{
			name:  "WithConfig",
			input: configpatcher.WithConfig(cfg),
		},
		{
			name:  "WithBytes",
			input: configpatcher.WithBytes(configMultidoc),
		},
	} {
		t.Run(tt.name, func(t *testing.T) {
			_, err := configpatcher.Apply(tt.input, patches)
			assert.EqualError(t, err, "JSON6902 patches are not supported for multi-document machine configuration")
		})
	}
}

func TestApplyMultiDoc(t *testing.T) {
	patches, err := configpatcher.LoadPatches([]string{
		"@testdata/multidoc/strategic1.yaml",
		"@testdata/multidoc/strategic2.yaml",
	})
	require.NoError(t, err)

	cfg, err := configloader.NewFromBytes(configMultidoc)
	require.NoError(t, err)

	for _, tt := range []struct {
		name  string
		input configpatcher.Input
	}{
		{
			name:  "WithConfig",
			input: configpatcher.WithConfig(cfg),
		},
		{
			name:  "WithBytes",
			input: configpatcher.WithBytes(configMultidoc),
		},
	} {
		t.Run(tt.name, func(t *testing.T) {
			out, err := configpatcher.Apply(tt.input, patches)
			require.NoError(t, err)

			bytes, err := out.Bytes()
			require.NoError(t, err)

			assert.Equal(t, string(expectedMultidoc), string(bytes))
		})
	}
}

//go:embed testdata/auditpolicy/config.yaml
var configAudit []byte

//go:embed testdata/auditpolicy/expected.yaml
var expectedAudit []byte

func TestApplyAuditPolicy(t *testing.T) {
	patches, err := configpatcher.LoadPatches([]string{
		"@testdata/auditpolicy/patch1.yaml",
	})
	require.NoError(t, err)

	cfg, err := configloader.NewFromBytes(configAudit)
	require.NoError(t, err)

	for _, tt := range []struct {
		name  string
		input configpatcher.Input
	}{
		{
			name:  "WithConfig",
			input: configpatcher.WithConfig(cfg),
		},
		{
			name:  "WithBytes",
			input: configpatcher.WithBytes(configAudit),
		},
	} {
		t.Run(tt.name, func(t *testing.T) {
			out, err := configpatcher.Apply(tt.input, patches)
			require.NoError(t, err)

			bytes, err := out.Bytes()
			require.NoError(t, err)

			assert.Equal(t, string(expectedAudit), string(bytes))
		})
	}
}