Files
warewulf/internal/app/wwctl/image/rename/main_test.go
Jonathon Anderson 45a690ca4e Rename "container" to "image"
- Updated `wwctl upgrade` to handle updates
- Maintained `.Container` and `.ContainerName` in tstruct
- Added `ContainerName()` methods to node and profile objects
- Added `--container`, `-C` aliases to wwctl commands (`<node|profile> <add|set>`)
- Added `wwctl container` alias
- Added support for `container_exit.sh` if `image_exit.sh` is not found

Signed-off-by: Jonathon Anderson <janderson@ciq.com>
2025-01-19 05:54:19 -07:00

56 lines
1.4 KiB
Go

package rename
import (
"bytes"
"os"
"path"
"testing"
"github.com/stretchr/testify/assert"
imageList "github.com/warewulf/warewulf/internal/app/wwctl/image/list"
"github.com/warewulf/warewulf/internal/pkg/testenv"
"github.com/warewulf/warewulf/internal/pkg/warewulfd"
"github.com/warewulf/warewulf/internal/pkg/wwlog"
)
func Test_Rename(t *testing.T) {
env := testenv.New(t)
env.WriteFile(path.Join(testenv.WWChrootdir, "test-image/rootfs/file"), `test`)
defer env.RemoveAll()
warewulfd.SetNoDaemon()
// first we will verify that there is an existing image
t.Run("image list", func(t *testing.T) {
verifyImageListOutput(t, "test-image")
})
// then rename it
t.Run("image rename", func(t *testing.T) {
baseCmd := GetCommand()
baseCmd.SetOut(os.Stdout)
baseCmd.SetErr(os.Stdout)
baseCmd.SetArgs([]string{"test-image", "test-image-rename"})
err := baseCmd.Execute()
assert.NoError(t, err)
})
// retrieve again
t.Run("Image list", func(t *testing.T) {
verifyImageListOutput(t, "test-image-rename")
})
}
func verifyImageListOutput(t *testing.T, content string) {
baseCmd := imageList.GetCommand()
buf := new(bytes.Buffer)
baseCmd.SetOut(buf)
baseCmd.SetErr(buf)
wwlog.SetLogWriterErr(buf)
wwlog.SetLogWriterInfo(buf)
err := baseCmd.Execute()
assert.NoError(t, err)
assert.NotEmpty(t, buf.String(), "output should not be empty")
assert.Contains(t, buf.String(), content)
}