76 lines
2.7 KiB
Go
76 lines
2.7 KiB
Go
package kernel
|
|
|
|
import (
|
|
"os"
|
|
"path"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
warewulfconf "github.com/warewulf/warewulf/internal/pkg/config"
|
|
"github.com/warewulf/warewulf/internal/pkg/util"
|
|
"github.com/warewulf/warewulf/internal/pkg/wwlog"
|
|
)
|
|
|
|
var kernelBuildTests = []struct {
|
|
kernelVersion string
|
|
kernelName string
|
|
kernelFileName string
|
|
succeed bool
|
|
}{
|
|
{"4.3.2.1", "kernel1", "vmlinuz-1.2.3.4.gz", false},
|
|
{"1.2.3.4", "kernel1", "vmlinuz-1.2.3.4.gz", true},
|
|
{"1.2+3+4", "kernel1", "vmlinuz-1.2+3+4.gz", true},
|
|
}
|
|
|
|
func Test_BuildKernel(t *testing.T) {
|
|
wwlog.SetLogLevel(wwlog.DEBUG)
|
|
for _, tt := range kernelBuildTests {
|
|
srvDir, err := os.MkdirTemp(os.TempDir(), "ww-test-srv-*")
|
|
assert.NoError(t, err)
|
|
conf := warewulfconf.Get()
|
|
conf.Paths.WWProvisiondir = srvDir
|
|
kernelDir, err := os.MkdirTemp(os.TempDir(), "ww-test-kernel-*")
|
|
assert.NoError(t, err)
|
|
{
|
|
err = os.MkdirAll(path.Join(kernelDir, "boot"), 0755)
|
|
assert.NoError(t, err)
|
|
err = os.MkdirAll(path.Join(kernelDir, "lib/modules/old-kernel"), 0755)
|
|
assert.NoError(t, err)
|
|
_, err = os.Create(path.Join(kernelDir, "lib/modules/old-kernel/old-module"))
|
|
assert.NoError(t, err)
|
|
err = os.MkdirAll(path.Join(kernelDir, "lib/firmware"), 0755)
|
|
assert.NoError(t, err)
|
|
_, err = os.Create(path.Join(kernelDir, "lib/firmware/test-firmware"))
|
|
assert.NoError(t, err)
|
|
_, err = os.Create(path.Join(kernelDir, "boot", tt.kernelFileName))
|
|
assert.NoError(t, err)
|
|
err = os.MkdirAll(path.Join(kernelDir, "lib/modules", tt.kernelVersion, "/nested"), 0755)
|
|
assert.NoError(t, err)
|
|
_, err = os.Create(path.Join(kernelDir, "lib/modules", tt.kernelVersion, "test-module"))
|
|
assert.NoError(t, err)
|
|
err = os.Symlink(path.Join(kernelDir, "lib/modules/old-kernel/old-module"), path.Join(kernelDir, "lib/modules", tt.kernelVersion, "symlink-module"))
|
|
assert.NoError(t, err)
|
|
}
|
|
t.Run(tt.kernelName, func(t *testing.T) {
|
|
err = Build(tt.kernelVersion, tt.kernelName, kernelDir)
|
|
if tt.succeed {
|
|
assert.NoError(t, err)
|
|
assert.FileExists(t, path.Join(srvDir, "kernel", tt.kernelName, "vmlinuz"))
|
|
assert.FileExists(t, path.Join(srvDir, "kernel", tt.kernelName, "kmods.img.gz"))
|
|
assert.FileExists(t, path.Join(srvDir, "kernel", tt.kernelName, "kmods.img"))
|
|
files, err := util.CpioFiles(path.Join(srvDir, "kernel", tt.kernelName, "kmods.img"))
|
|
assert.NoError(t, err)
|
|
assert.ElementsMatch(t, files, []string{
|
|
"lib/firmware/test-firmware",
|
|
"lib/modules/" + tt.kernelVersion + "/symlink-module",
|
|
"lib/modules/" + tt.kernelVersion + "/test-module",
|
|
"lib/modules/" + tt.kernelVersion + "/nested"})
|
|
} else {
|
|
assert.Error(t, err)
|
|
}
|
|
})
|
|
os.RemoveAll(srvDir)
|
|
os.RemoveAll(kernelDir)
|
|
}
|
|
}
|