talos/pkg/machinery/resources/network/hostname_spec_test.go
Andrey Smirnov d02d944ec7
chore: provide umarshal from YAML methods for network resource specs
This will be required to preserve platform network configuration in a
cache across boots. Network configuration will be marshaled to YAML and
unmarshaled back.

Changes are pretty simple and most of the code is auto-generated:
replace `stringer` codegen with `enumer` which produces also methods
which convert from string back to enum values.

Added tests to verify YAML marshaling/unmarshaling.

Signed-off-by: Andrey Smirnov <andrey.smirnov@talos-systems.com>
2022-01-10 17:00:53 +03:00

35 lines
879 B
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 network_test
import (
"testing"
"github.com/stretchr/testify/assert"
"github.com/stretchr/testify/require"
"gopkg.in/yaml.v3"
"github.com/talos-systems/talos/pkg/machinery/resources/network"
)
func TestHostnameSpecMarshalYAML(t *testing.T) {
spec := network.HostnameSpecSpec{
Hostname: "foo",
Domainname: "example.com",
ConfigLayer: network.ConfigPlatform,
}
marshaled, err := yaml.Marshal(spec)
require.NoError(t, err)
assert.Equal(t, "hostname: foo\ndomainname: example.com\nlayer: platform\n", string(marshaled))
var spec2 network.HostnameSpecSpec
require.NoError(t, yaml.Unmarshal(marshaled, &spec2))
assert.Equal(t, spec, spec2)
}