use talos encoder

edgelesssys · Oct 16, 2023 · d0de973 · d0de973
1 parent b9bfa61
commit d0de973
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/cli/internal/state/BUILD.bazel b/cli/internal/state/BUILD.bazel
@@ -23,8 +23,8 @@ go_test(
     deps = [
         "//internal/constants",
         "//internal/file",
+        "@com_github_siderolabs_talos_pkg_machinery//config/encoder",
         "@com_github_spf13_afero//:afero",
         "@com_github_stretchr_testify//assert",
-        "@in_gopkg_yaml_v3//:yaml_v3",
     ],
 )
diff --git a/cli/internal/state/state_test.go b/cli/internal/state/state_test.go
@@ -11,9 +11,9 @@ import (
 
 	"github.com/edgelesssys/constellation/v2/internal/constants"
 	"github.com/edgelesssys/constellation/v2/internal/file"
+	"github.com/siderolabs/talos/pkg/machinery/config/encoder"
 	"github.com/spf13/afero"
 	"github.com/stretchr/testify/assert"
-	"gopkg.in/yaml.v3"
 )
 
 var defaultState = &State{
@@ -145,7 +145,7 @@ func TestReadFromFile(t *testing.T) {
 
 func mustMarshalYaml(t *testing.T, v any) []byte {
 	t.Helper()
-	b, err := yaml.Marshal(v)
+	b, err := encoder.NewEncoder(v).Encode()
 	if err != nil {
 		t.Fatalf("failed to marshal yaml: %v", err)
 	}