mirror of
https://github.com/siderolabs/omni.git
synced 2026-01-21 19:01:04 +01:00
de6e2c66f7
Refactor for code simplicity. Signed-off-by: Pranav Patil <pranavppatil767@gmail.com> Signed-off-by: Utku Ozdemir <utku.ozdemir@siderolabs.com>
108 lines
3.0 KiB
Go
108 lines
3.0 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/.
|
|
|
|
//nolint:staticcheck // we are ok with accesssing the deprecated fields in these tests.
|
|
package compression_test
|
|
|
|
import (
|
|
"encoding/json"
|
|
"strings"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
"go.yaml.in/yaml/v4"
|
|
|
|
"github.com/siderolabs/omni/client/api/omni/specs"
|
|
"github.com/siderolabs/omni/client/pkg/constants"
|
|
"github.com/siderolabs/omni/client/pkg/omni/resources/omni"
|
|
)
|
|
|
|
func TestClusterMachineConfigPatchesYAML(t *testing.T) {
|
|
res := omni.NewClusterMachineConfigPatches("test")
|
|
|
|
// set some patches
|
|
|
|
aString := strings.Repeat("a", constants.CompressionThresholdBytes)
|
|
bString := strings.Repeat("b", constants.CompressionThresholdBytes)
|
|
|
|
err := res.TypedSpec().Value.SetUncompressedPatches([]string{aString, bString})
|
|
require.NoError(t, err)
|
|
|
|
// assert that the patches are compressed
|
|
|
|
require.Empty(t, res.TypedSpec().Value.Patches)
|
|
require.NotEmpty(t, res.TypedSpec().Value.CompressedPatches)
|
|
|
|
// marshal the spec to yaml
|
|
|
|
spec := res.TypedSpec().Value
|
|
|
|
specYAML, err := yaml.Marshal(spec)
|
|
require.NoError(t, err)
|
|
|
|
// assert that the patches are in the YAML in uncompressed form, and do not contain the compressed form
|
|
|
|
require.Contains(t, string(specYAML), aString)
|
|
require.Contains(t, string(specYAML), bString)
|
|
require.NotContains(t, string(specYAML), "compressed")
|
|
|
|
t.Logf("yaml:\n%s", string(specYAML))
|
|
|
|
// unmarshal the spec from the yaml
|
|
|
|
var newSpec specs.ClusterMachineConfigPatchesSpec
|
|
|
|
err = yaml.Unmarshal(specYAML, &newSpec)
|
|
require.NoError(t, err)
|
|
|
|
// assert that the patches got compressed again
|
|
|
|
require.Empty(t, newSpec.Patches)
|
|
require.NotEmpty(t, newSpec.CompressedPatches)
|
|
}
|
|
|
|
func TestClusterMachineConfigPatchesJSON(t *testing.T) {
|
|
res := omni.NewClusterMachineConfigPatches("test")
|
|
|
|
aString := strings.Repeat("a", constants.CompressionThresholdBytes)
|
|
bString := strings.Repeat("b", constants.CompressionThresholdBytes)
|
|
|
|
// set some patches
|
|
|
|
err := res.TypedSpec().Value.SetUncompressedPatches([]string{aString, bString})
|
|
require.NoError(t, err)
|
|
|
|
// assert that the patches are compressed
|
|
|
|
require.Empty(t, res.TypedSpec().Value.Patches)
|
|
require.NotEmpty(t, res.TypedSpec().Value.CompressedPatches)
|
|
|
|
// marshal the spec to json
|
|
|
|
spec := res.TypedSpec().Value
|
|
|
|
specJSON, err := json.Marshal(spec)
|
|
require.NoError(t, err)
|
|
|
|
// assert that the patches are in the JSON in uncompressed form, and do not contain the compressed form
|
|
|
|
require.Contains(t, string(specJSON), aString)
|
|
require.Contains(t, string(specJSON), bString)
|
|
require.NotContains(t, string(specJSON), "compressed")
|
|
|
|
t.Logf("json:\n%s", string(specJSON))
|
|
|
|
// unmarshal the spec from the json
|
|
|
|
var newSpec *specs.ClusterMachineConfigPatchesSpec
|
|
|
|
err = json.Unmarshal(specJSON, &newSpec)
|
|
require.NoError(t, err)
|
|
|
|
// assert that the patches got compressed again
|
|
|
|
require.Empty(t, newSpec.Patches)
|
|
require.NotEmpty(t, newSpec.CompressedPatches)
|
|
}
|