kelvin/warewulf
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity
Files
a7325b6ffaf3a1c0aacc79b343b98370ee719386
warewulf/internal/app/wwctl/node/list/main_test.go
Jonathon Anderson a7325b6ffa Remove WW_INTERNAL 
Signed-off-by: Jonathon Anderson <janderson@ciq.com>
2024-11-11 00:08:54 -07:00

639 lines
12 KiB
Go
Raw Blame History

package list
import (
"bytes"
"os"
"strings"
"testing"
warewulfconf "github.com/warewulf/warewulf/internal/pkg/config"
"github.com/stretchr/testify/assert"
"github.com/warewulf/warewulf/internal/pkg/node"
"github.com/warewulf/warewulf/internal/pkg/testenv"
"github.com/warewulf/warewulf/internal/pkg/warewulfd"
"github.com/warewulf/warewulf/internal/pkg/wwlog"
)
func Test_List(t *testing.T) {
tests := []struct {
name string
args []string
wantErr bool
stdout string
inDb string
}{
{
name: "single node list",
args: []string{},
wantErr: false,
stdout: `
NODE NAME PROFILES NETWORK
--------- -------- -------
n01 [default] --
`,
inDb: `nodeprofiles:
default: {}
nodes:
n01:
profiles:
- default
`,
},
{
name: "multiple nodes list",
args: []string{},
wantErr: false,
stdout: `
NODE NAME PROFILES NETWORK
--------- -------- -------
n01 [default] --
n02 [default] --
`,
inDb: `nodeprofiles:
default: {}
nodes:
n01:
profiles:
- default
n02:
profiles:
- default
`,
},
{
name: "node list returns multiple nodes",
args: []string{"n01,n02"},
wantErr: false,
stdout: `
NODE NAME PROFILES NETWORK
--------- -------- -------
n01 [default] --
n02 [default] --
`,
inDb: `nodeprofiles:
default: {}
nodes:
n01:
profiles:
- default
n02:
profiles:
- default
`,
},
{
name: "node list returns multiple nodes (case 2)",
args: []string{"n01,n03"},
wantErr: false,
stdout: `
NODE NAME PROFILES NETWORK
--------- -------- -------
n01 [default] --
n03 [default] --
`,
inDb: `nodeprofiles:
default: {}
nodes:
n01:
profiles:
- default
n02:
profiles:
- default
n03:
profiles:
- default
n04:
profiles:
- default
n05:
profiles:
- default
`,
},
{
name: "node list returns one node",
args: []string{"n01,"},
wantErr: false,
stdout: `
NODE NAME PROFILES NETWORK
--------- -------- -------
n01 [default] --
`,
inDb: `nodeprofiles:
default: {}
nodes:
n01:
profiles:
- default
n02:
profiles:
- default
`,
},
{
name: "node list profile with network",
args: []string{},
wantErr: false,
stdout: `
NODE NAME PROFILES NETWORK
--------- -------- -------
n01 [default] default
`,
inDb: `nodeprofiles:
default:
network devices:
default:
device: eth0
nodes:
n01:
profiles:
- default
`},
{
name: "node list profile with comment",
args: []string{"-a"},
wantErr: false,
stdout: `
NODE FIELD PROFILE VALUE
---- ----- ------- -----
n01 Comment default profilecomment
n01 Profiles -- default
`,
inDb: `nodeprofiles:
default:
comment: profilecomment
nodes:
n01:
profiles:
- default
`},
{
name: "node list profile with comment superseded",
args: []string{"-a"},
wantErr: false,
stdout: `
NODE FIELD PROFILE VALUE
---- ----- ------- -----
n01 Comment SUPERSEDED nodecomment
n01 Profiles -- default
`,
inDb: `nodeprofiles:
default:
comment: profilecomment
nodes:
n01:
comment: nodecomment
profiles:
- default
`},
{
name: "node list profile with ipmi user",
args: []string{"-i"},
wantErr: false,
stdout: `
NODE IPMI IPADDR IPMI PORT IPMI USERNAME IPMI INTERFACE
---- ----------- --------- ------------- --------------
n01 <nil> -- admin --
`,
inDb: `nodeprofiles:
default:
ipmi:
username: admin
nodes:
n01:
profiles:
- default
`},
{
name: "node list profile with ipmi user superseded",
args: []string{"-i"},
wantErr: false,
stdout: `
NODE IPMI IPADDR IPMI PORT IPMI USERNAME IPMI INTERFACE
---- ----------- --------- ------------- --------------
n01 <nil> -- user --
`,
inDb: `nodeprofiles:
default:
ipmi:
username: admin
nodes:
n01:
ipmi:
username: user
profiles:
- default
`},
{
name: "multiple profiles list",
args: []string{},
wantErr: false,
stdout: `
NODE NAME PROFILES NETWORK
--------- -------- -------
n01 [p1 p2] --
`,
inDb: `nodeprofiles:
p1: {}
p2: {}
nodes:
n01:
profiles:
- p1
- p2
`},
{
name: "multiple profiles list all",
args: []string{"-a"},
wantErr: false,
stdout: `
NODE FIELD PROFILE VALUE
---- ----- ------- -----
n01 Profiles -- p1,p2
`,
inDb: `nodeprofiles:
p1: {}
p2: {}
nodes:
n01:
profiles:
- p1
- p2
`},
{
name: "multiple overlays list long with negation",
args: []string{"-l"},
wantErr: false,
stdout: `
NODE NAME KERNEL OVERRIDE CONTAINER OVERLAYS (S/R)
--------- --------------- --------- --------------
n01 -- -- /rop1,rop2
`,
inDb: `nodeprofiles:
p1:
runtime overlay:
- rop1
- rop2
nodes:
n01:
profiles:
- p1
`},
{
name: "multiple overlays list long",
args: []string{"-l"},
wantErr: false,
stdout: `
NODE NAME KERNEL OVERRIDE CONTAINER OVERLAYS (S/R)
--------- --------------- --------- --------------
n01 -- -- sop1/nop1,~rop1,rop1,rop2
`,
inDb: `nodeprofiles:
p1:
system overlay:
- sop1
runtime overlay:
- rop1
- rop2
nodes:
n01:
profiles:
- p1
runtime overlay:
- nop1
- ~rop1
`},
{
name: "multiple overlays list all with negation",
args: []string{"-a"},
wantErr: false,
stdout: `
NODE FIELD PROFILE VALUE
---- ----- ------- -----
n01 Profiles -- p1
n01 RuntimeOverlay p1+ nop1,~rop1,rop1,rop2
n01 SystemOverlay p1 sop1
`,
inDb: `nodeprofiles:
p1:
system overlay:
- sop1
runtime overlay:
- rop1
- rop2
nodes:
n01:
profiles:
- p1
runtime overlay:
- nop1
- ~rop1
`},
{
name: "multiple overlays list all",
args: []string{"-a"},
wantErr: false,
stdout: `
NODE FIELD PROFILE VALUE
---- ----- ------- -----
n01 Profiles -- p1
n01 RuntimeOverlay p1+ nop1,rop1,rop2
`,
inDb: `nodeprofiles:
p1:
runtime overlay:
- rop1
- rop2
nodes:
n01:
profiles:
- p1
runtime overlay:
- nop1
`},
{
name: "network onboot",
args: []string{"-a"},
wantErr: false,
stdout: `
NODE FIELD PROFILE VALUE
---- ----- ------- -----
n1 NetDevs[default].OnBoot -- true
`,
inDb: `nodes:
n1:
network devices:
default:
onboot: true
`},
{
name: "empty network device",
args: []string{"-a"},
wantErr: false,
stdout: `
NODE FIELD PROFILE VALUE
---- ----- ------- -----
`,
inDb: `nodes:
wwnode1:
network devices:
default: {}
`},
}
conf_yml := ``
tempWarewulfConf, warewulfConfErr := os.CreateTemp("", "warewulf.conf-")
assert.NoError(t, warewulfConfErr)
defer os.Remove(tempWarewulfConf.Name())
_, warewulfConfErr = tempWarewulfConf.Write([]byte(conf_yml))
assert.NoError(t, warewulfConfErr)
assert.NoError(t, tempWarewulfConf.Sync())
assert.NoError(t, warewulfconf.New().Read(tempWarewulfConf.Name()))
tempNodeConf, nodesConfErr := os.CreateTemp("", "nodes.conf-")
assert.NoError(t, nodesConfErr)
defer os.Remove(tempNodeConf.Name())
node.ConfigFile = tempNodeConf.Name()
warewulfd.SetNoDaemon()
for _, tt := range tests {
var err error
_, err = tempNodeConf.Seek(0, 0)
assert.NoError(t, err)
assert.NoError(t, tempNodeConf.Truncate(0))
_, err = tempNodeConf.Write([]byte(tt.inDb))
assert.NoError(t, err)
assert.NoError(t, tempNodeConf.Sync())
t.Run(tt.name, func(t *testing.T) {
baseCmd := GetCommand()
baseCmd.SetArgs(tt.args)
buf := new(bytes.Buffer)
baseCmd.SetOut(buf)
baseCmd.SetErr(buf)
wwlog.SetLogWriter(buf)
wwlog.SetLogWriterErr(buf)
wwlog.SetLogWriterInfo(buf)
err := baseCmd.Execute()
assert.NoError(t, err)
assert.Equal(t, strings.TrimSpace(tt.stdout), strings.TrimSpace(buf.String()))
})
}
}
func TestListMultipleFormats(t *testing.T) {
tests := []struct {
name string
args []string
stdout string
inDb string
wantErr bool
}{
{
name: "single node list yaml output",
args: []string{"-y"},
stdout: `
- profiles:
- default
kernel: {}
ipmi: {}
`,
inDb: `
nodeprofiles:
default: {}
nodes:
n01:
profiles:
- default
`,
},
{
name: "single node list json output",
args: []string{"-j"},
stdout: `
[
{
"Discoverable": "",
"AssetKey": "",
"Profiles": [
"default"
],
"Comment": "",
"ClusterName": "",
"ContainerName": "",
"Ipxe": "",
"RuntimeOverlay": null,
"SystemOverlay": null,
"Kernel": {},
"Ipmi": {
"UserName": "",
"Password": "",
"Ipaddr": "",
"Gateway": "",
"Netmask": "",
"Port": "",
"Interface": "",
"EscapeChar": "",
"Write": "",
"Template": "",
"Tags": {}
},
"Init": "",
"Root": "",
"NetDevs": {},
"Tags": {},
"PrimaryNetDev": "",
"Disks": null,
"FileSystems": null
}
]
`,
inDb: `
nodeprofiles:
default: {}
nodes:
n01:
profiles:
- default
`,
},
{
name: "multiple nodes list json output",
args: []string{"-j"},
stdout: `
[
{
"Discoverable": "",
"AssetKey": "",
"Profiles": [
"default"
],
"Comment": "",
"ClusterName": "",
"ContainerName": "",
"Ipxe": "",
"RuntimeOverlay": null,
"SystemOverlay": null,
"Kernel": {},
"Ipmi": {
"UserName": "",
"Password": "",
"Ipaddr": "",
"Gateway": "",
"Netmask": "",
"Port": "",
"Interface": "",
"EscapeChar": "",
"Write": "",
"Template": "",
"Tags": {}
},
"Init": "",
"Root": "",
"NetDevs": {},
"Tags": {},
"PrimaryNetDev": "",
"Disks": null,
"FileSystems": null
},
{
"Discoverable": "",
"AssetKey": "",
"Profiles": [
"default"
],
"Comment": "",
"ClusterName": "",
"ContainerName": "",
"Ipxe": "",
"RuntimeOverlay": null,
"SystemOverlay": null,
"Kernel": {},
"Ipmi": {
"UserName": "",
"Password": "",
"Ipaddr": "",
"Gateway": "",
"Netmask": "",
"Port": "",
"Interface": "",
"EscapeChar": "",
"Write": "",
"Template": "",
"Tags": {}
},
"Init": "",
"Root": "",
"NetDevs": {},
"Tags": {},
"PrimaryNetDev": "",
"Disks": null,
"FileSystems": null
}
]
`,
inDb: `
nodeprofiles:
default: {}
nodes:
n01:
profiles:
- default
n02:
profiles:
- default
`,
},
{
name: "multiple nodes list yaml output",
args: []string{"-y"},
stdout: `
- profiles:
- default
kernel: {}
ipmi: {}
- profiles:
- default
kernel: {}
ipmi: {}
`,
inDb: `
nodeprofiles:
default: {}
nodes:
n01:
profiles:
- default
n02:
profiles:
- default
`,
},
}
warewulfd.SetNoDaemon()
for _, tt := range tests {
env := testenv.New(t)
env.WriteFile(t, "etc/warewulf/nodes.conf", tt.inDb)
var err error
t.Run(tt.name, func(t *testing.T) {
buf := new(bytes.Buffer)
wwlog.SetLogWriter(buf)
baseCmd := GetCommand()
baseCmd.SetArgs(tt.args)
baseCmd.SetOut(buf)
baseCmd.SetErr(buf)
err = baseCmd.Execute()
if tt.wantErr {
assert.Error(t, err)
} else {
assert.NoError(t, err)
}
assert.Equal(t, strings.TrimSpace(tt.stdout), strings.TrimSpace(buf.String()))
})
}
}
Reference in New Issue View Git Blame Copy Permalink