warewulf/internal/pkg/warewulfd/provision_test.go

package warewulfd

import (
	"io"
	"net/http"
	"net/http/httptest"
	"os"
	"path"
	"testing"

	"github.com/stretchr/testify/assert"

	warewulfconf "github.com/warewulf/warewulf/internal/pkg/config"
	"github.com/warewulf/warewulf/internal/pkg/node"
	"github.com/warewulf/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"},
	{"find initramfs", "/provision/00:00:00:ff:ff:ff?stage=initramfs", "", 200, "10.10.10.10: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: 45
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)
	}
	assert.NoError(t, os.MkdirAll(path.Join(containerDir, "suse/rootfs/boot"), 0700))
	{
		_, err := os.Create(path.Join(containerDir, "suse/rootfs/boot", "initramfs-.img"))
		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)
		})
	}
}