talos/pkg/machinery/resources/network/resolver_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
942 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"
"inet.af/netaddr"
"github.com/talos-systems/talos/pkg/machinery/resources/network"
)
func TestResolverSpecMarshalYAML(t *testing.T) {
spec := network.ResolverSpecSpec{
DNSServers: []netaddr.IP{netaddr.MustParseIP("1.1.1.1"), netaddr.MustParseIP("8.8.8.8")},
ConfigLayer: network.ConfigPlatform,
}
marshaled, err := yaml.Marshal(spec)
require.NoError(t, err)
assert.Equal(t, "dnsServers:\n - 1.1.1.1\n - 8.8.8.8\nlayer: platform\n", string(marshaled))
var spec2 network.ResolverSpecSpec
require.NoError(t, yaml.Unmarshal(marshaled, &spec2))
assert.Equal(t, spec, spec2)
}