talos/pkg/resources/k8s/manifest_test.go
Andrey Smirnov d2d5c72bb5 fix: skip empty manifest YAML sub-documents
If sub-doc contains a comment, it's not empty, but when
loaded into JSON it yields `nil` which should be skipped, as
`Unstructured` can't load from such object.

E.g.:

```
---
 # Some comment
---
```

Signed-off-by: Andrey Smirnov <smirnov.andrey@gmail.com>
2021-02-19 05:26:41 -08:00

50 lines
1.1 KiB
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/.
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")
}