kelvin/warewulf
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity
Files
deaf9c99d7ecf5da5769e899c86aecba52d51ffe
warewulf/internal/app/wwctl/node/list/main_test.go
Jonathon Anderson 1a3d037703 Refactor GetNode merging with new MergeNode method 
- Adds many tests for node+profile merging
- Ensures that tracked fields match mergo results

Signed-off-by: Jonathon Anderson <janderson@ciq.com>
2024-12-25 09:07:28 -07:00

631 lines
11 KiB
Go
Raw Blame History

package list
import (
"bytes"
"strings"
"testing"
"github.com/stretchr/testify/assert"
"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 Profiles -- default
n01 Comment default profilecomment
`,
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 Profiles -- default
n01 Comment SUPERSEDED nodecomment
`,
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 VERSION 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 VERSION CONTAINER OVERLAYS (S/R)
--------- -------------- --------- --------------
n01 -- -- sop1/rop1,rop2,nop1,~rop1
`,
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,n01 rop1,rop2,nop1,~rop1
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,n01 rop1,rop2,nop1
`,
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: {}
`,
},
}
warewulfd.SetNoDaemon()
env := testenv.New(t)
defer env.RemoveAll(t)
for _, tt := range tests {
t.Run(tt.name, func(t *testing.T) {
env.WriteFile(t, "etc/warewulf/nodes.conf", tt.inDb)
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