a322061fe9
The initramfs stage supports serving an initramfs image from within the container image, typically generated by dracut. Signed-off-by: xu yang <xyang@ciq.com> Signed-off-by: Jonathon Anderson <janderson@ciq.com>
89 lines
2.1 KiB
Go
89 lines
2.1 KiB
Go
package container
|
|
|
|
import (
|
|
"fmt"
|
|
"os"
|
|
"path/filepath"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
warewulfconf "github.com/warewulf/warewulf/internal/pkg/config"
|
|
)
|
|
|
|
func TestInitramfsBootPath(t *testing.T) {
|
|
conf := warewulfconf.Get()
|
|
temp, err := os.MkdirTemp(os.TempDir(), "ww-conf-*")
|
|
assert.NoError(t, err)
|
|
defer os.RemoveAll(temp)
|
|
conf.Paths.WWChrootdir = temp
|
|
|
|
assert.NoError(t, os.MkdirAll(filepath.Join(RootFsDir("image"), "boot"), 0700))
|
|
|
|
tests := []struct {
|
|
name string
|
|
initramfs []string
|
|
ver string
|
|
err error
|
|
retName string
|
|
}{
|
|
{
|
|
name: "ok case 1",
|
|
initramfs: []string{"initramfs-1.1.1.aarch64.img"},
|
|
ver: "1.1.1.aarch64",
|
|
err: nil,
|
|
},
|
|
{
|
|
name: "ok case 2",
|
|
initramfs: []string{"initrd-1.1.1.aarch64"},
|
|
ver: "1.1.1.aarch64",
|
|
err: nil,
|
|
},
|
|
{
|
|
name: "ok case 3",
|
|
initramfs: []string{"initramfs-1.1.1.aarch64"},
|
|
ver: "1.1.1.aarch64",
|
|
err: nil,
|
|
},
|
|
{
|
|
name: "ok case 4",
|
|
initramfs: []string{"initrd-1.1.1.aarch64.img"},
|
|
ver: "1.1.1.aarch64",
|
|
err: nil,
|
|
},
|
|
{
|
|
name: "error case, wrong init name",
|
|
initramfs: []string{"initrr-1.1.1.aarch64.img"},
|
|
ver: "1.1.1.aarch64",
|
|
err: fmt.Errorf("Failed to find a target kernel version initramfs"),
|
|
},
|
|
{
|
|
name: "error case, wrong ver",
|
|
initramfs: []string{"initrr-1.1.1.aarch64.img"},
|
|
ver: "1.1.2.aarch64",
|
|
err: fmt.Errorf("Failed to find a target kernel version initramfs"),
|
|
},
|
|
}
|
|
|
|
for _, tt := range tests {
|
|
t.Logf("running test: %s", tt.name)
|
|
for _, init := range tt.initramfs {
|
|
assert.NoError(t, os.WriteFile(filepath.Join(RootFsDir("image"), "boot", init), []byte(""), 0600))
|
|
}
|
|
initPath, err := InitramfsBootPath("image", tt.ver)
|
|
assert.Equal(t, tt.err, err)
|
|
if err == nil {
|
|
assert.NotEmpty(t, initPath)
|
|
} else {
|
|
assert.Empty(t, initPath)
|
|
}
|
|
|
|
if tt.retName != "" {
|
|
assert.Equal(t, filepath.Base(initPath), tt.retName)
|
|
}
|
|
// remove the file
|
|
for _, init := range tt.initramfs {
|
|
assert.NoError(t, os.Remove(filepath.Join(RootFsDir("image"), "boot", init)))
|
|
}
|
|
}
|
|
}
|