755411d3cb
Co-authored-by: Jonathon Anderson <janderson@ciq.com> Signed-off-by: Christian Goll <cgoll@suse.com> Signed-off-by: Jonathon Anderson <janderson@ciq.com>
111 lines
2.8 KiB
Go
111 lines
2.8 KiB
Go
package warewulfd
|
|
|
|
import (
|
|
"io"
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
|
|
"github.com/warewulf/warewulf/internal/pkg/testenv"
|
|
)
|
|
|
|
var overlaySendTests = map[string]struct {
|
|
url string
|
|
body string
|
|
status int
|
|
}{
|
|
"get file": {
|
|
url: "/overlay-file/pub/non-template",
|
|
body: "Non-template: {{.Id}}",
|
|
status: 200,
|
|
},
|
|
"getting a missing file returns 404": {
|
|
url: "/overlay-file/pub/does-not-exist",
|
|
body: "",
|
|
status: 404,
|
|
},
|
|
"get raw template": {
|
|
url: "/overlay-file/pub/template.ww",
|
|
body: "Template: {{.Id}}",
|
|
status: 200,
|
|
},
|
|
"get rendered template": {
|
|
url: "/overlay-file/pub/template.ww?render=n1",
|
|
body: "Template: n1",
|
|
status: 200,
|
|
},
|
|
"get rendered template without explicit suffix": {
|
|
url: "/overlay-file/pub/template?render=n1",
|
|
body: "Template: n1",
|
|
status: 200,
|
|
},
|
|
"explicit suffix required when no node specified": {
|
|
url: "/overlay-file/pub/template",
|
|
body: "",
|
|
status: 404,
|
|
},
|
|
"getting a template with a missing node returns 404": {
|
|
url: "/overlay-file/pub/test.template.ww?render=n2",
|
|
body: "",
|
|
status: 404,
|
|
},
|
|
"don't render non-template files": {
|
|
url: "/overlay-file/pub/non-template?render=n1",
|
|
body: "Non-template: {{.Id}}",
|
|
status: 200,
|
|
},
|
|
"getting a missing template returns 404": {
|
|
url: "/overlay-file/pub/does-not-exist.ww?render=n1",
|
|
body: "",
|
|
status: 404,
|
|
},
|
|
"get a file from a subdir": {
|
|
url: "/overlay-file/pub/subdir/non-template",
|
|
body: "Non-template (subdir): {{.Id}}",
|
|
status: 200,
|
|
},
|
|
"render a template from a subdir": {
|
|
url: "/overlay-file/pub/subdir/template.ww?render=n1",
|
|
body: "Template (subdir): n1",
|
|
status: 200,
|
|
},
|
|
}
|
|
|
|
func Test_OverlaySend(t *testing.T) {
|
|
env := testenv.New(t)
|
|
env.WriteFile("etc/warewulf/warewulf.conf", `
|
|
warewulf:
|
|
secure: false
|
|
`)
|
|
env.WriteFile("etc/warewulf/nodes.conf", `
|
|
nodeprofiles:
|
|
default: {}
|
|
nodes:
|
|
n1: {}
|
|
`)
|
|
_ = env.Configure()
|
|
env.WriteFile("var/lib/warewulf/overlays/pub/rootfs/non-template", "Non-template: {{.Id}}")
|
|
env.WriteFile("var/lib/warewulf/overlays/pub/rootfs/template.ww", "Template: {{.Id}}")
|
|
env.WriteFile("var/lib/warewulf/overlays/pub/rootfs/subdir/non-template", "Non-template (subdir): {{.Id}}")
|
|
env.WriteFile("var/lib/warewulf/overlays/pub/rootfs/subdir/template.ww", "Template (subdir): {{.Id}}")
|
|
|
|
for description, tt := range overlaySendTests {
|
|
t.Run(description, func(t *testing.T) {
|
|
req := httptest.NewRequest(http.MethodGet, tt.url, nil)
|
|
w := httptest.NewRecorder()
|
|
OverlaySend(w, req)
|
|
res := w.Result()
|
|
defer res.Body.Close()
|
|
|
|
data, readErr := io.ReadAll(res.Body)
|
|
assert.NoError(t, readErr)
|
|
if tt.body != "" {
|
|
assert.Equal(t, tt.body, string(data))
|
|
}
|
|
assert.Equal(t, tt.status, res.StatusCode)
|
|
})
|
|
}
|
|
}
|