// 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/.

package k8s_test

import (
	"strings"
	"testing"

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

	"github.com/talos-systems/talos/pkg/resources/k8s"
)

func TestManifestSetYAML(t *testing.T) {
	manifest := k8s.NewManifest(k8s.ControlPlaneNamespaceName, "test")

	require.NoError(t, manifest.SetYAML([]byte(strings.TrimSpace(`
---
apiVersion: audit.k8s.io/v1beta1
kind: Policy
rules:
- level: Metadata
---
`))))

	assert.Len(t, manifest.Objects(), 1)
	assert.Equal(t, manifest.Objects()[0].GetKind(), "Policy")
}

func TestManifestSetYAMLEmptyComments(t *testing.T) {
	manifest := k8s.NewManifest(k8s.ControlPlaneNamespaceName, "test")

	require.NoError(t, manifest.SetYAML([]byte(strings.TrimSpace(`
---
apiVersion: audit.k8s.io/v1beta1
kind: Policy
rules:
- level: Metadata
---
# Left empty
---
`))))

	assert.Len(t, manifest.Objects(), 1)
	assert.Equal(t, manifest.Objects()[0].GetKind(), "Policy")
}