118 lines
4.0 KiB
Go
118 lines
4.0 KiB
Go
package warewulfd
|
|
|
|
import (
|
|
"io"
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"os"
|
|
"path"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
|
|
warewulfconf "github.com/hpcng/warewulf/internal/pkg/config"
|
|
"github.com/hpcng/warewulf/internal/pkg/node"
|
|
"github.com/hpcng/warewulf/internal/pkg/wwlog"
|
|
)
|
|
|
|
var provisionSendTests = []struct {
|
|
description string
|
|
url string
|
|
body string
|
|
status int
|
|
ip string
|
|
}{
|
|
{"system overlay", "/overlay-system/00:00:00:ff:ff:ff", "system overlay", 200, "10.10.10.10:9873"},
|
|
{"runtime overlay", "/overlay-runtime/00:00:00:ff:ff:ff", "runtime overlay", 200, "10.10.10.10:9873"},
|
|
{"fake overlay", "/overlay-system/00:00:00:ff:ff:ff?overlay=fake", "", 404, "10.10.10.10:9873:9873"},
|
|
{"specific overlay", "/overlay-system/00:00:00:ff:ff:ff?overlay=o1", "specific overlay", 200, "10.10.10.10:9873"},
|
|
{"find shim", "/efiboot/shim.efi", "", 200, "10.10.10.10:9873"},
|
|
{"find shim", "/efiboot/shim.efi", "", 404, "10.10.10.11:9873"},
|
|
{"find grub", "/efiboot/grub.efi", "", 200, "10.10.10.10:9873"},
|
|
{"find grub", "/efiboot/grub.efi", "", 404, "10.10.10.11:9873"},
|
|
}
|
|
|
|
func Test_ProvisionSend(t *testing.T) {
|
|
conf_file, err := os.CreateTemp(os.TempDir(), "ww-test-nodes.conf-*")
|
|
assert.NoError(t, err)
|
|
defer conf_file.Close()
|
|
{
|
|
_, err := conf_file.WriteString(`WW_INTERNAL: 43
|
|
nodeprofiles:
|
|
default:
|
|
container name: suse
|
|
nodes:
|
|
n1:
|
|
network devices:
|
|
default:
|
|
hwaddr: 00:00:00:ff:ff:ff
|
|
n2:
|
|
network devices:
|
|
default:
|
|
hwaddr: 00:00:00:00:ff:ff
|
|
container name: none`)
|
|
assert.NoError(t, err)
|
|
}
|
|
assert.NoError(t, conf_file.Sync())
|
|
node.ConfigFile = conf_file.Name()
|
|
|
|
// create a arp file as for grub we look up the ip address through the arp cache
|
|
arp_file, err := os.CreateTemp(os.TempDir(), "ww-arp")
|
|
assert.NoError(t, err)
|
|
defer arp_file.Close()
|
|
{
|
|
_, err := arp_file.WriteString(`IP address HW type Flags HW address Mask Device
|
|
10.10.10.10 0x1 0x2 00:00:00:ff:ff:ff * dummy
|
|
10.10.10.11 0x1 0x2 00:00:00:00:ff:ff * dummy`)
|
|
assert.NoError(t, err)
|
|
}
|
|
assert.NoError(t, arp_file.Sync())
|
|
SetArpFile(arp_file.Name())
|
|
|
|
conf := warewulfconf.Get()
|
|
containerDir, imageDirErr := os.MkdirTemp(os.TempDir(), "ww-test-container-*")
|
|
assert.NoError(t, imageDirErr)
|
|
defer os.RemoveAll(containerDir)
|
|
conf.Paths.WWChrootdir = containerDir
|
|
assert.NoError(t, os.MkdirAll(path.Join(containerDir, "suse/rootfs/usr/lib64/efi"), 0700))
|
|
{
|
|
_, err := os.Create(path.Join(containerDir, "suse/rootfs/usr/lib64/efi", "shim.efi"))
|
|
assert.NoError(t, err)
|
|
}
|
|
assert.NoError(t, os.MkdirAll(path.Join(containerDir, "suse/rootfs/usr/share/efi/x86_64/"), 0700))
|
|
{
|
|
_, err := os.Create(path.Join(containerDir, "suse/rootfs/usr/share/efi/x86_64/", "grub.efi"))
|
|
assert.NoError(t, err)
|
|
}
|
|
|
|
dbErr := LoadNodeDB()
|
|
assert.NoError(t, dbErr)
|
|
|
|
provisionDir, provisionDirErr := os.MkdirTemp(os.TempDir(), "ww-test-provision-*")
|
|
assert.NoError(t, provisionDirErr)
|
|
defer os.RemoveAll(provisionDir)
|
|
conf.Paths.WWProvisiondir = provisionDir
|
|
conf.Warewulf.Secure = false
|
|
wwlog.SetLogLevel(wwlog.DEBUG)
|
|
assert.NoError(t, os.MkdirAll(path.Join(provisionDir, "overlays", "n1"), 0700))
|
|
assert.NoError(t, os.WriteFile(path.Join(provisionDir, "overlays", "n1", "__SYSTEM__.img"), []byte("system overlay"), 0600))
|
|
assert.NoError(t, os.WriteFile(path.Join(provisionDir, "overlays", "n1", "__RUNTIME__.img"), []byte("runtime overlay"), 0600))
|
|
assert.NoError(t, os.WriteFile(path.Join(provisionDir, "overlays", "n1", "o1.img"), []byte("specific overlay"), 0600))
|
|
|
|
for _, tt := range provisionSendTests {
|
|
t.Run(tt.description, func(t *testing.T) {
|
|
req := httptest.NewRequest(http.MethodGet, tt.url, nil)
|
|
req.RemoteAddr = tt.ip
|
|
w := httptest.NewRecorder()
|
|
ProvisionSend(w, req)
|
|
res := w.Result()
|
|
defer res.Body.Close()
|
|
|
|
data, readErr := io.ReadAll(res.Body)
|
|
assert.NoError(t, readErr)
|
|
assert.Equal(t, tt.body, string(data))
|
|
assert.Equal(t, tt.status, res.StatusCode)
|
|
})
|
|
}
|
|
}
|