130 lines
3.4 KiB
Go
130 lines
3.4 KiB
Go
package node
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"gopkg.in/yaml.v3"
|
|
)
|
|
|
|
func NewUtilTestNode() (NodesYaml, error) {
|
|
var data = `
|
|
nodeprofiles:
|
|
default:
|
|
comment: This profile is automatically included for each node
|
|
nodes:
|
|
test_node:
|
|
primary network: net0
|
|
comment: Node Comment
|
|
profiles:
|
|
- default
|
|
network devices:
|
|
net0:
|
|
hwaddr: 00:00:00:00:12:34
|
|
ipaddr: 1.2.3.4
|
|
device: eth0
|
|
net1:
|
|
hwaddr: ab:cd:ef:00:12:34
|
|
ipaddr: 1.2.3.4
|
|
device: eth1
|
|
net2:
|
|
hwaddr: aB:Cd:eF:12:34:56
|
|
ipaddr: 1.2.3.4
|
|
device: eth2
|
|
test_node_IPv6:
|
|
primary network: net1
|
|
profiles:
|
|
- default
|
|
network devices:
|
|
net1:
|
|
default: false
|
|
ipaddr: fd1a:2b3c:4d5e:06f0:1234:5678:90ab:cdef
|
|
`
|
|
var ret NodesYaml
|
|
err := yaml.Unmarshal([]byte(data), &ret)
|
|
if err != nil {
|
|
return ret, err
|
|
}
|
|
return ret, nil
|
|
}
|
|
|
|
func Test_nodeYaml_FindByHwaddr(t *testing.T) {
|
|
c, err := NewUtilTestNode()
|
|
assert.NoError(t, err)
|
|
//type fields struct {
|
|
// NodeProfiles map[string]*NodeConf
|
|
// Nodes map[string]*NodeConf
|
|
//}
|
|
type args struct {
|
|
hwa string
|
|
}
|
|
tests := []struct {
|
|
name string
|
|
//fields fields
|
|
config NodesYaml
|
|
args args
|
|
want string
|
|
wantErr bool
|
|
}{
|
|
{"emptyString", c, args{hwa: ""}, "", true},
|
|
{"noIpString", c, args{hwa: "this is not a MAC"}, "", true},
|
|
{"intString", c, args{hwa: "4294967296"}, "", true},
|
|
{"invalidMAC", c, args{hwa: "xx:00:00:00:12:34"}, "", true},
|
|
{"validMACNotFound", c, args{hwa: "aa:FF:ee:65:43:21"}, "", true},
|
|
{"validMAC", c, args{hwa: "ab:cd:ef:00:12:34"}, "test_node", false},
|
|
{"validMAC2", c, args{hwa: "aB:Cd:eF:00:12:34"}, "test_node", false},
|
|
{"validMAC3", c, args{hwa: "Ab:cD:Ef:12:34:56"}, "test_node", false},
|
|
}
|
|
for _, tt := range tests {
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
config := tt.config
|
|
got, err := config.FindByHwaddr(tt.args.hwa)
|
|
if (err != nil) != tt.wantErr {
|
|
t.Errorf("FindByHwaddr() error = %v, wantErr %v", err, tt.wantErr)
|
|
return
|
|
}
|
|
if got.id != tt.want {
|
|
t.Errorf("FindByHwaddr() got = %v, want %v", got, tt.want)
|
|
}
|
|
})
|
|
}
|
|
}
|
|
|
|
func Test_nodeYaml_FindByIpaddr(t *testing.T) {
|
|
c, err := NewUtilTestNode()
|
|
assert.NoError(t, err)
|
|
type args struct {
|
|
ipaddr string
|
|
}
|
|
tests := []struct {
|
|
name string
|
|
config NodesYaml
|
|
args args
|
|
want string
|
|
wantErr bool
|
|
}{
|
|
{"emptyString", c, args{ipaddr: ""}, "", true},
|
|
{"noIpString", c, args{ipaddr: "this is not an IP"}, "", true},
|
|
{"intString", c, args{ipaddr: "4294967296"}, "", true},
|
|
{"invalidIPv4", c, args{ipaddr: "1.2.3.256"}, "", true},
|
|
{"invalidIPv6", c, args{ipaddr: "xd1a:2b3c:4d5e:06f0:1234:5678:90ab:cdef"}, "", true},
|
|
{"validIPv4NotFound", c, args{ipaddr: "1.1.1.1"}, "", true},
|
|
{"validIPv6NotFound", c, args{ipaddr: "fd1a:2b3c:4d5e:06f0:1234:5678:90ab:fedc"}, "", true},
|
|
{"validIPv4", c, args{ipaddr: "1.2.3.4"}, "test_node", false},
|
|
{"validIPv6", c, args{ipaddr: "fd1a:2b3c:4d5e:06f0:1234:5678:90ab:cdef"}, "test_node_IPv6", false},
|
|
}
|
|
for _, tt := range tests {
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
config := tt.config
|
|
got, err := config.FindByIpaddr(tt.args.ipaddr)
|
|
if (err != nil) != tt.wantErr {
|
|
t.Errorf("FindByIpaddr() error = %v, wantErr %v", err, tt.wantErr)
|
|
return
|
|
}
|
|
if got.id != tt.want {
|
|
t.Errorf("FindByHwaddr() got = %v, want %v", got, tt.want)
|
|
}
|
|
})
|
|
}
|
|
}
|