talos

1
// This Source Code Form is subject to the terms of the Mozilla Public
2
// License, v. 2.0. If a copy of the MPL was not distributed with this
3
// file, You can obtain one at http://mozilla.org/MPL/2.0/.
4

5
package quirks_test
6

7
import (
8
	"testing"
9

10
	"github.com/stretchr/testify/assert"
11

12
	"github.com/siderolabs/talos/pkg/machinery/imager/quirks"
13
)
14

15
func TestSupportsResetOption(t *testing.T) {
16
	for _, test := range []struct {
17
		version string
18

19
		expected bool
20
	}{
21
		{
22
			version:  "1.5.0",
23
			expected: true,
24
		},
25
		{
26
			expected: true,
27
		},
28
		{
29
			version:  "1.3.7",
30
			expected: false,
31
		},
32
	} {
33
		t.Run(test.version, func(t *testing.T) {
34
			assert.Equal(t, test.expected, quirks.New(test.version).SupportsResetGRUBOption())
35
		})
36
	}
37
}
38

39
func TestSupportsCompressedEncodedMETA(t *testing.T) {
40
	for _, test := range []struct {
41
		version string
42

43
		expected bool
44
	}{
45
		{
46
			version:  "1.6.3",
47
			expected: true,
48
		},
49
		{
50
			version:  "1.7.0",
51
			expected: true,
52
		},
53
		{
54
			expected: true,
55
		},
56
		{
57
			version:  "1.6.2",
58
			expected: false,
59
		},
60
	} {
61
		t.Run(test.version, func(t *testing.T) {
62
			assert.Equal(t, test.expected, quirks.New(test.version).SupportsCompressedEncodedMETA())
63
		})
64
	}
65
}
66

67
func TestSupportsOverlay(t *testing.T) {
68
	for _, test := range []struct {
69
		version string
70

71
		expected bool
72
	}{
73
		{
74
			version:  "1.6.3",
75
			expected: false,
76
		},
77
		{
78
			version:  "1.7.0",
79
			expected: true,
80
		},
81
		{
82
			expected: true,
83
		},
84
		{
85
			version:  "1.6.2",
86
			expected: false,
87
		},
88
		{
89
			version:  "1.7.0-alpha.0",
90
			expected: true,
91
		},
92
		{
93
			version:  "v1.7.0-alpha.0-75-gff08e2821",
94
			expected: true,
95
		},
96
	} {
97
		t.Run(test.version, func(t *testing.T) {
98
			assert.Equal(t, test.expected, quirks.New(test.version).SupportsOverlay())
99
		})
100
	}
101
}
102

103
func TestSupportsZstd(t *testing.T) {
104
	for _, test := range []struct {
105
		version string
106

107
		expected bool
108
	}{
109
		{
110
			version:  "1.7.3",
111
			expected: false,
112
		},
113
		{
114
			expected: true,
115
		},
116
		{
117
			version:  "1.6.2",
118
			expected: false,
119
		},
120
		{
121
			version:  "1.8.0-alpha.0",
122
			expected: true,
123
		},
124
		{
125
			version:  "v1.8.3",
126
			expected: true,
127
		},
128
	} {
129
		t.Run(test.version, func(t *testing.T) {
130
			assert.Equal(t, test.expected, quirks.New(test.version).UseZSTDCompression())
131
		})
132
	}
133
}
134