739 lines
12 KiB
Go
739 lines
12 KiB
Go
package set
|
|
|
|
import (
|
|
"bytes"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/warewulf/warewulf/internal/pkg/testenv"
|
|
"github.com/warewulf/warewulf/internal/pkg/warewulfd"
|
|
)
|
|
|
|
type test_description struct {
|
|
name string
|
|
args []string
|
|
wantErr bool
|
|
stdout string
|
|
inDB string
|
|
outDb string
|
|
}
|
|
|
|
func run_test(t *testing.T, test test_description) {
|
|
env := testenv.New(t)
|
|
defer env.RemoveAll()
|
|
env.WriteFile("etc/warewulf/nodes.conf", test.inDB)
|
|
warewulfd.SetNoDaemon()
|
|
name := test.name
|
|
if name == "" {
|
|
name = t.Name()
|
|
}
|
|
t.Run(name, func(t *testing.T) {
|
|
baseCmd := GetCommand()
|
|
test.args = append(test.args, "--yes")
|
|
baseCmd.SetArgs(test.args)
|
|
buf := new(bytes.Buffer)
|
|
baseCmd.SetOut(buf)
|
|
baseCmd.SetErr(buf)
|
|
err := baseCmd.Execute()
|
|
if test.wantErr {
|
|
assert.Error(t, err)
|
|
} else {
|
|
assert.NoError(t, err)
|
|
assert.Equal(t, buf.String(), test.stdout)
|
|
content := env.ReadFile("etc/warewulf/nodes.conf")
|
|
assert.YAMLEq(t, test.outDb, content)
|
|
}
|
|
})
|
|
}
|
|
|
|
func Test_Single_Node_Change_Profile(t *testing.T) {
|
|
test := test_description{
|
|
args: []string{"--profile=foo", "n01"},
|
|
wantErr: false,
|
|
stdout: "",
|
|
inDB: `nodeprofiles:
|
|
default:
|
|
comment: testit
|
|
nodes:
|
|
n01:
|
|
profiles:
|
|
- default`,
|
|
outDb: `nodeprofiles:
|
|
default:
|
|
comment: testit
|
|
nodes:
|
|
n01:
|
|
profiles:
|
|
- foo
|
|
`,
|
|
}
|
|
run_test(t, test)
|
|
}
|
|
|
|
func Test_Node_Unset(t *testing.T) {
|
|
test := test_description{
|
|
args: []string{"--comment=UNDEF", "n01"},
|
|
wantErr: false,
|
|
stdout: "",
|
|
inDB: `nodeprofiles: {}
|
|
nodes:
|
|
n01:
|
|
comment: foo
|
|
profiles:
|
|
- default`,
|
|
outDb: `nodeprofiles: {}
|
|
nodes:
|
|
n01:
|
|
profiles:
|
|
- default
|
|
`,
|
|
}
|
|
run_test(t, test)
|
|
}
|
|
|
|
func Test_Set_Ipmi_Write_Explicit(t *testing.T) {
|
|
test := test_description{
|
|
args: []string{"--ipmiwrite", "true", "n01"},
|
|
wantErr: false,
|
|
stdout: "",
|
|
inDB: `nodeprofiles: {}
|
|
nodes:
|
|
n01: {}
|
|
`,
|
|
outDb: `nodeprofiles: {}
|
|
nodes:
|
|
n01:
|
|
ipmi:
|
|
write: "true"
|
|
`,
|
|
}
|
|
run_test(t, test)
|
|
}
|
|
|
|
func Test_Set_Ipmi_Write_Implicit(t *testing.T) {
|
|
test := test_description{
|
|
args: []string{"--ipmiwrite", "n01"},
|
|
wantErr: false,
|
|
stdout: "",
|
|
inDB: `nodeprofiles: {}
|
|
nodes:
|
|
n01: {}
|
|
`,
|
|
outDb: `nodeprofiles: {}
|
|
nodes:
|
|
n01:
|
|
ipmi:
|
|
write: "true"
|
|
`,
|
|
}
|
|
run_test(t, test)
|
|
}
|
|
|
|
func Test_Unset_Ipmi_Write(t *testing.T) {
|
|
test := test_description{
|
|
args: []string{"--ipmiwrite=UNDEF", "n01"},
|
|
wantErr: false,
|
|
stdout: "",
|
|
inDB: `nodeprofiles: {}
|
|
nodes:
|
|
n01:
|
|
ipmi:
|
|
write: "true"
|
|
`,
|
|
outDb: `nodeprofiles: {}
|
|
nodes:
|
|
n01: {}
|
|
`,
|
|
}
|
|
run_test(t, test)
|
|
}
|
|
|
|
func Test_Unset_Ipmi_Write_False(t *testing.T) {
|
|
test := test_description{
|
|
args: []string{"--ipmiwrite=UNDEF", "n01"},
|
|
wantErr: false,
|
|
stdout: "",
|
|
inDB: `nodeprofiles: {}
|
|
nodes:
|
|
n01:
|
|
ipmi:
|
|
write: "false"
|
|
`,
|
|
outDb: `nodeprofiles: {}
|
|
nodes:
|
|
n01: {}
|
|
`,
|
|
}
|
|
run_test(t, test)
|
|
}
|
|
|
|
func Test_Ipmi_Hidden_False(t *testing.T) {
|
|
test := test_description{
|
|
args: []string{"--ipmiwrite=false", "n01"},
|
|
wantErr: false,
|
|
stdout: "",
|
|
inDB: `nodeprofiles:
|
|
default:
|
|
ipmi:
|
|
write: "true"
|
|
nodes:
|
|
n01:
|
|
profiles:
|
|
- default
|
|
`,
|
|
outDb: `nodeprofiles:
|
|
default:
|
|
ipmi:
|
|
write: "true"
|
|
nodes:
|
|
n01:
|
|
profiles:
|
|
- default
|
|
ipmi:
|
|
write: "false"
|
|
`,
|
|
}
|
|
run_test(t, test)
|
|
}
|
|
|
|
func Test_Add_NetTags(t *testing.T) {
|
|
test := test_description{
|
|
args: []string{"--nettagadd=dns=1.1.1.1", "n01"},
|
|
wantErr: false,
|
|
stdout: "",
|
|
inDB: `nodeprofiles: {}
|
|
nodes:
|
|
n01: {}
|
|
`,
|
|
outDb: `nodeprofiles: {}
|
|
nodes:
|
|
n01:
|
|
network devices:
|
|
default:
|
|
tags:
|
|
dns: 1.1.1.1
|
|
`,
|
|
}
|
|
run_test(t, test)
|
|
}
|
|
|
|
func Test_Del_NetTags(t *testing.T) {
|
|
test := test_description{
|
|
args: []string{"--netname=default", "--nettagdel=dns1,dns2", "n01"},
|
|
wantErr: false,
|
|
stdout: "",
|
|
inDB: `nodeprofiles: {}
|
|
nodes:
|
|
n01:
|
|
network devices:
|
|
default:
|
|
tags:
|
|
dns1: 1.1.1.1
|
|
dns2: 2.2.2.2
|
|
`,
|
|
outDb: `nodeprofiles: {}
|
|
nodes:
|
|
n01: {}
|
|
`,
|
|
}
|
|
run_test(t, test)
|
|
}
|
|
|
|
func Test_Multiple_Set_Tests(t *testing.T) {
|
|
tests := []test_description{
|
|
{
|
|
name: "single node change profile",
|
|
args: []string{"--profile=foo", "n01"},
|
|
wantErr: false,
|
|
stdout: "",
|
|
inDB: `nodeprofiles:
|
|
default:
|
|
comment: testit
|
|
nodes:
|
|
n01:
|
|
profiles:
|
|
- default`,
|
|
outDb: `nodeprofiles:
|
|
default:
|
|
comment: testit
|
|
nodes:
|
|
n01:
|
|
profiles:
|
|
- foo
|
|
`,
|
|
},
|
|
{
|
|
name: "multiple nodes change profile",
|
|
args: []string{"--profile=foo", "n0[1-2]"},
|
|
wantErr: false,
|
|
stdout: "",
|
|
inDB: `nodeprofiles:
|
|
default:
|
|
comment: testit
|
|
nodes:
|
|
n01:
|
|
profiles:
|
|
- default
|
|
n02:
|
|
profiles:
|
|
- default`,
|
|
outDb: `nodeprofiles:
|
|
default:
|
|
comment: testit
|
|
nodes:
|
|
n01:
|
|
profiles:
|
|
- foo
|
|
n02:
|
|
profiles:
|
|
- foo
|
|
`,
|
|
},
|
|
{
|
|
name: "single node set ipmitag",
|
|
args: []string{"--ipmitagadd", "foo=baar", "n01"},
|
|
wantErr: false,
|
|
stdout: "",
|
|
inDB: `nodeprofiles:
|
|
default:
|
|
comment: testit
|
|
nodes:
|
|
n01:
|
|
profiles:
|
|
- default`,
|
|
outDb: `nodeprofiles:
|
|
default:
|
|
comment: testit
|
|
nodes:
|
|
n01:
|
|
profiles:
|
|
- default
|
|
ipmi:
|
|
tags:
|
|
foo: baar
|
|
`,
|
|
},
|
|
{
|
|
name: "single node delete tag",
|
|
args: []string{"--tagdel", "tag1", "n01"},
|
|
wantErr: false,
|
|
stdout: "",
|
|
inDB: `nodeprofiles:
|
|
default:
|
|
comment: testit
|
|
nodes:
|
|
n01:
|
|
profiles:
|
|
- default
|
|
tags:
|
|
tag1: value1
|
|
tag2: value2`,
|
|
outDb: `nodeprofiles:
|
|
default:
|
|
comment: testit
|
|
nodes:
|
|
n01:
|
|
profiles:
|
|
- default
|
|
tags:
|
|
tag2: value2
|
|
`,
|
|
},
|
|
{
|
|
name: "single node add tag",
|
|
args: []string{"--tagadd", "tag1=foobaar", "n01"},
|
|
wantErr: false,
|
|
stdout: "",
|
|
inDB: `nodeprofiles:
|
|
default: {}
|
|
nodes:
|
|
n01: {}
|
|
`,
|
|
outDb: `nodeprofiles:
|
|
default: {}
|
|
nodes:
|
|
n01:
|
|
tags:
|
|
tag1: foobaar
|
|
`,
|
|
},
|
|
{
|
|
name: "single node add tag with netdev",
|
|
args: []string{"--tagadd", "tag1=foobaar", "n01"},
|
|
wantErr: false,
|
|
stdout: "",
|
|
inDB: `nodeprofiles:
|
|
default: {}
|
|
nodes:
|
|
n01:
|
|
network devices:
|
|
default:
|
|
ipaddr: 172.16.130.101
|
|
|
|
`,
|
|
outDb: `nodeprofiles:
|
|
default: {}
|
|
nodes:
|
|
n01:
|
|
tags:
|
|
tag1: foobaar
|
|
network devices:
|
|
default:
|
|
ipaddr: 172.16.130.101
|
|
`,
|
|
},
|
|
{
|
|
name: "single node set onboot",
|
|
args: []string{"--netname", "default", "--onboot=true", "n01"},
|
|
wantErr: false,
|
|
stdout: "",
|
|
inDB: `nodeprofiles:
|
|
default: {}
|
|
nodes:
|
|
n01:
|
|
network devices:
|
|
default:
|
|
ipaddr: 172.16.130.101
|
|
|
|
`,
|
|
outDb: `nodeprofiles:
|
|
default: {}
|
|
nodes:
|
|
n01:
|
|
network devices:
|
|
default:
|
|
ipaddr: 172.16.130.101
|
|
onboot: "true"
|
|
`,
|
|
},
|
|
|
|
{
|
|
name: "single node set fs,part and disk",
|
|
args: []string{"--fsname=var", "--fspath=/var", "--fsformat=btrfs", "--partname=var", "--partnumber=1", "--diskname=/dev/vda", "n01"},
|
|
wantErr: false,
|
|
stdout: "",
|
|
inDB: `nodeprofiles:
|
|
default:
|
|
comment: testit
|
|
nodes:
|
|
n01:
|
|
profiles:
|
|
- default
|
|
`,
|
|
outDb: `nodeprofiles:
|
|
default:
|
|
comment: testit
|
|
nodes:
|
|
n01:
|
|
profiles:
|
|
- default
|
|
disks:
|
|
/dev/vda:
|
|
partitions:
|
|
var:
|
|
number: "1"
|
|
filesystems:
|
|
/dev/disk/by-partlabel/var:
|
|
format: btrfs
|
|
path: /var
|
|
`,
|
|
},
|
|
{
|
|
name: "single delete not existing fs",
|
|
args: []string{"--fsdel=foo", "n01"},
|
|
wantErr: true,
|
|
stdout: "",
|
|
inDB: `nodeprofiles:
|
|
default:
|
|
comment: testit
|
|
nodes:
|
|
n01:
|
|
profiles:
|
|
- default
|
|
disks:
|
|
/dev/vda:
|
|
partitions:
|
|
var:
|
|
number: "1"
|
|
path: /var
|
|
filesystems:
|
|
/dev/disk/by-partlabel/var:
|
|
format: btrfs
|
|
path: /var
|
|
`,
|
|
outDb: `nodeprofiles:
|
|
default:
|
|
comment: testit
|
|
nodes:
|
|
n01:
|
|
profiles:
|
|
- default
|
|
disks:
|
|
/dev/vda:
|
|
partitions:
|
|
var:
|
|
number: "1"
|
|
filesystems:
|
|
/dev/disk/by-partlabel/var:
|
|
format: btrfs
|
|
path: /var
|
|
`,
|
|
},
|
|
{
|
|
name: "single node delete existing fs",
|
|
args: []string{"--fsdel=/dev/disk/by-partlabel/var", "n01"},
|
|
wantErr: false,
|
|
stdout: "",
|
|
inDB: `nodeprofiles:
|
|
default:
|
|
comment: testit
|
|
nodes:
|
|
n01:
|
|
profiles:
|
|
- default
|
|
disks:
|
|
/dev/vda:
|
|
partitions:
|
|
var:
|
|
number: "1"
|
|
path: /var
|
|
filesystems:
|
|
/dev/disk/by-partlabel/var:
|
|
format: btrfs
|
|
path: /var
|
|
`,
|
|
outDb: `nodeprofiles:
|
|
default:
|
|
comment: testit
|
|
nodes:
|
|
n01:
|
|
profiles:
|
|
- default
|
|
disks:
|
|
/dev/vda:
|
|
partitions:
|
|
var:
|
|
number: "1"
|
|
`,
|
|
},
|
|
{
|
|
name: "single node delete existing partition",
|
|
args: []string{"--partdel=var", "n01"},
|
|
wantErr: false,
|
|
stdout: "",
|
|
inDB: `nodeprofiles:
|
|
default:
|
|
comment: testit
|
|
nodes:
|
|
n01:
|
|
profiles:
|
|
- default
|
|
disks:
|
|
/dev/vda:
|
|
partitions:
|
|
var:
|
|
number: "1"
|
|
path: /var
|
|
filesystems:
|
|
/dev/disk/by-partlabel/var:
|
|
format: btrfs
|
|
path: /var
|
|
`,
|
|
outDb: `nodeprofiles:
|
|
default:
|
|
comment: testit
|
|
nodes:
|
|
n01:
|
|
profiles:
|
|
- default
|
|
disks:
|
|
/dev/vda: {}
|
|
filesystems:
|
|
/dev/disk/by-partlabel/var:
|
|
format: btrfs
|
|
path: /var
|
|
`,
|
|
},
|
|
{
|
|
name: "single node delete existing disk",
|
|
args: []string{"--diskdel=/dev/vda", "n01"},
|
|
wantErr: false,
|
|
stdout: "",
|
|
inDB: `nodeprofiles:
|
|
default:
|
|
comment: testit
|
|
nodes:
|
|
n01:
|
|
profiles:
|
|
- default
|
|
disks:
|
|
/dev/vda:
|
|
partitions:
|
|
var: {}
|
|
path: /var
|
|
filesystems:
|
|
/dev/disk/by-partlabel/var:
|
|
format: btrfs
|
|
path: /var
|
|
`,
|
|
outDb: `nodeprofiles:
|
|
default:
|
|
comment: testit
|
|
nodes:
|
|
n01:
|
|
profiles:
|
|
- default
|
|
filesystems:
|
|
/dev/disk/by-partlabel/var:
|
|
format: btrfs
|
|
path: /var
|
|
`,
|
|
},
|
|
{
|
|
name: "single node set mtu",
|
|
args: []string{"--mtu", "1234", "--netname=mynet", "n01"},
|
|
wantErr: false,
|
|
stdout: "",
|
|
inDB: `nodeprofiles:
|
|
default:
|
|
comment: testit
|
|
nodes:
|
|
n01:
|
|
profiles:
|
|
- default`,
|
|
outDb: `nodeprofiles:
|
|
default:
|
|
comment: testit
|
|
nodes:
|
|
n01:
|
|
profiles:
|
|
- default
|
|
network devices:
|
|
mynet:
|
|
mtu: "1234"
|
|
`,
|
|
},
|
|
{
|
|
name: "single node set ipmitag",
|
|
args: []string{"--tagadd", "nodetag1=nodevalue1", "n01"},
|
|
wantErr: false,
|
|
stdout: "",
|
|
inDB: `nodeprofiles:
|
|
p1:
|
|
comment: testit 1
|
|
tags:
|
|
p1tag1: p1val1
|
|
p2:
|
|
comment: testit 1
|
|
tags:
|
|
p2tag2: p1val2
|
|
nodes:
|
|
n01:
|
|
profiles:
|
|
- p1
|
|
- p2`,
|
|
outDb: `nodeprofiles:
|
|
p1:
|
|
comment: testit 1
|
|
tags:
|
|
p1tag1: p1val1
|
|
p2:
|
|
comment: testit 1
|
|
tags:
|
|
p2tag2: p1val2
|
|
nodes:
|
|
n01:
|
|
profiles:
|
|
- p1
|
|
- p2
|
|
tags:
|
|
nodetag1: nodevalue1
|
|
`,
|
|
},
|
|
{
|
|
name: "single node set comma in comment",
|
|
args: []string{"n01", "--comment", "This is a , comment"},
|
|
wantErr: false,
|
|
stdout: "",
|
|
inDB: `nodes:
|
|
n01:
|
|
comment: old comment
|
|
`,
|
|
outDb: `nodeprofiles: {}
|
|
nodes:
|
|
n01:
|
|
comment: This is a , comment
|
|
`,
|
|
},
|
|
}
|
|
|
|
warewulfd.SetNoDaemon()
|
|
for _, tt := range tests {
|
|
run_test(t, tt)
|
|
}
|
|
}
|
|
|
|
func Test_Node_Add(t *testing.T) {
|
|
tests := []test_description{
|
|
{
|
|
args: []string{"--tagadd=email=node", "n01"},
|
|
wantErr: false,
|
|
stdout: "",
|
|
inDB: `nodeprofiles:
|
|
default:
|
|
comment: testit
|
|
tags:
|
|
email: profile
|
|
nodes:
|
|
n01:
|
|
profiles:
|
|
- default`,
|
|
outDb: `nodeprofiles:
|
|
default:
|
|
comment: testit
|
|
tags:
|
|
email: profile
|
|
nodes:
|
|
n01:
|
|
profiles:
|
|
- default
|
|
tags:
|
|
email: node
|
|
`,
|
|
},
|
|
{
|
|
args: []string{"--tagadd=newtag=newval", "n01"},
|
|
wantErr: false,
|
|
stdout: "",
|
|
inDB: `nodeprofiles:
|
|
default:
|
|
comment: testit
|
|
tags:
|
|
email: profile
|
|
nodes:
|
|
n01:
|
|
profiles:
|
|
- default
|
|
tags:
|
|
email: node`,
|
|
outDb: `nodeprofiles:
|
|
default:
|
|
comment: testit
|
|
tags:
|
|
email: profile
|
|
nodes:
|
|
n01:
|
|
profiles:
|
|
- default
|
|
tags:
|
|
email: node
|
|
newtag: newval
|
|
`,
|
|
},
|
|
}
|
|
|
|
for _, tt := range tests {
|
|
run_test(t, tt)
|
|
}
|
|
}
|