107 lines
3.8 KiB
Go
107 lines
3.8 KiB
Go
package kernels
|
|
|
|
import (
|
|
"bytes"
|
|
"path/filepath"
|
|
"strings"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/warewulf/warewulf/internal/pkg/testenv"
|
|
"github.com/warewulf/warewulf/internal/pkg/wwlog"
|
|
)
|
|
|
|
func Test_List(t *testing.T) {
|
|
tests := map[string]struct {
|
|
files map[string][]string
|
|
args []string
|
|
stdout string
|
|
}{
|
|
"default": {
|
|
files: map[string][]string{},
|
|
args: []string{},
|
|
stdout: `
|
|
Image Kernel Version Default Nodes
|
|
----- ------ ------- ------- -----
|
|
`,
|
|
},
|
|
"list": {
|
|
files: map[string][]string{
|
|
"image1": {
|
|
"/boot/vmlinuz-5.14.0-427.18.1.el9_4.x86_64",
|
|
"/boot/vmlinuz-5.14.0-427.24.1.el9_4.x86_64",
|
|
"/boot/vmlinuz-4.14.0-427.18.1.el8_4.x86_64",
|
|
},
|
|
"image2": {
|
|
"/boot/vmlinuz-0-rescue-eb46964329b146e39518c625feab3ea0",
|
|
"/boot/vmlinuz-5.14.0-362.24.1.el9_3.aarch64",
|
|
"/boot/vmlinuz-5.14.0-427.31.1.el9_4.aarch64+debug",
|
|
"/boot/vmlinuz-5.14.0-284.30.1.el9_2.aarch64",
|
|
"/boot/vmlinuz-5.14.0-427.31.1.el9_4.aarch64",
|
|
},
|
|
},
|
|
args: []string{},
|
|
stdout: `
|
|
Image Kernel Version Default Nodes
|
|
----- ------ ------- ------- -----
|
|
image1 /boot/vmlinuz-4.14.0-427.18.1.el8_4.x86_64 4.14.0-427.18.1 false 0
|
|
image1 /boot/vmlinuz-5.14.0-427.18.1.el9_4.x86_64 5.14.0-427.18.1 false 0
|
|
image1 /boot/vmlinuz-5.14.0-427.24.1.el9_4.x86_64 5.14.0-427.24.1 true 0
|
|
image2 /boot/vmlinuz-0-rescue-eb46964329b146e39518c625feab3ea0 -- false 0
|
|
image2 /boot/vmlinuz-5.14.0-284.30.1.el9_2.aarch64 5.14.0-284.30.1 false 0
|
|
image2 /boot/vmlinuz-5.14.0-362.24.1.el9_3.aarch64 5.14.0-362.24.1 false 0
|
|
image2 /boot/vmlinuz-5.14.0-427.31.1.el9_4.aarch64 5.14.0-427.31.1 true 0
|
|
image2 /boot/vmlinuz-5.14.0-427.31.1.el9_4.aarch64+debug 5.14.0-427.31.1 false 0
|
|
`,
|
|
},
|
|
"single image": {
|
|
files: map[string][]string{
|
|
"image1": {
|
|
"/boot/vmlinuz-5.14.0-427.18.1.el9_4.x86_64",
|
|
"/boot/vmlinuz-5.14.0-427.24.1.el9_4.x86_64",
|
|
"/boot/vmlinuz-4.14.0-427.18.1.el8_4.x86_64",
|
|
},
|
|
"image2": {
|
|
"/boot/vmlinuz-0-rescue-eb46964329b146e39518c625feab3ea0",
|
|
"/boot/vmlinuz-5.14.0-362.24.1.el9_3.aarch64",
|
|
"/boot/vmlinuz-5.14.0-427.31.1.el9_4.aarch64+debug",
|
|
"/boot/vmlinuz-5.14.0-284.30.1.el9_2.aarch64",
|
|
"/boot/vmlinuz-5.14.0-427.31.1.el9_4.aarch64",
|
|
},
|
|
},
|
|
args: []string{"image2"},
|
|
stdout: `
|
|
Image Kernel Version Default Nodes
|
|
----- ------ ------- ------- -----
|
|
image2 /boot/vmlinuz-0-rescue-eb46964329b146e39518c625feab3ea0 -- false 0
|
|
image2 /boot/vmlinuz-5.14.0-284.30.1.el9_2.aarch64 5.14.0-284.30.1 false 0
|
|
image2 /boot/vmlinuz-5.14.0-362.24.1.el9_3.aarch64 5.14.0-362.24.1 false 0
|
|
image2 /boot/vmlinuz-5.14.0-427.31.1.el9_4.aarch64 5.14.0-427.31.1 true 0
|
|
image2 /boot/vmlinuz-5.14.0-427.31.1.el9_4.aarch64+debug 5.14.0-427.31.1 false 0
|
|
`,
|
|
},
|
|
}
|
|
|
|
for name, tt := range tests {
|
|
t.Run(name, func(t *testing.T) {
|
|
env := testenv.New(t)
|
|
defer env.RemoveAll()
|
|
for image, files := range tt.files {
|
|
rootfs := filepath.Join(filepath.Join("/var/lib/warewulf/chroots", image), "rootfs")
|
|
for _, file := range files {
|
|
env.CreateFile(filepath.Join(rootfs, file))
|
|
}
|
|
}
|
|
buf := new(bytes.Buffer)
|
|
baseCmd := GetCommand()
|
|
baseCmd.SetArgs(tt.args)
|
|
baseCmd.SetOut(buf)
|
|
baseCmd.SetErr(buf)
|
|
wwlog.SetLogWriter(buf)
|
|
err := baseCmd.Execute()
|
|
assert.NoError(t, err)
|
|
assert.Equal(t, strings.TrimSpace(tt.stdout), strings.TrimSpace(buf.String()))
|
|
})
|
|
}
|
|
}
|