mirror of
https://github.com/yunionio/cloudpods.git
synced 2026-05-08 06:31:00 +08:00
481 lines
10 KiB
Go
481 lines
10 KiB
Go
// Copyright 2019 Yunion
|
|
//
|
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
|
// you may not use this file except in compliance with the License.
|
|
// You may obtain a copy of the License at
|
|
//
|
|
// http://www.apache.org/licenses/LICENSE-2.0
|
|
//
|
|
// Unless required by applicable law or agreed to in writing, software
|
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
// See the License for the specific language governing permissions and
|
|
// limitations under the License.
|
|
|
|
package sysutils
|
|
|
|
import (
|
|
"fmt"
|
|
"net"
|
|
"reflect"
|
|
"testing"
|
|
|
|
"yunion.io/x/jsonutils"
|
|
|
|
"yunion.io/x/onecloud/pkg/apis/compute"
|
|
"yunion.io/x/onecloud/pkg/cloudcommon/types"
|
|
)
|
|
|
|
func Test_valueOfKeyword(t *testing.T) {
|
|
type args struct {
|
|
line string
|
|
key string
|
|
}
|
|
t1 := "Manufacturer: LENOVO"
|
|
w1 := "LENOVO"
|
|
tests := []struct {
|
|
name string
|
|
args args
|
|
want *string
|
|
}{
|
|
{
|
|
name: "EmptyInput",
|
|
args: args{
|
|
line: "",
|
|
key: "Family",
|
|
},
|
|
want: nil,
|
|
},
|
|
{
|
|
name: "NormalInput",
|
|
args: args{
|
|
line: t1,
|
|
key: "manufacturer:",
|
|
},
|
|
want: &w1,
|
|
},
|
|
}
|
|
for _, tt := range tests {
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
if got := valueOfKeyword(tt.args.line, tt.args.key); got != nil && *got != *(tt.want) {
|
|
t.Errorf("valueOfKeyword() = %q, want %q", *got, *(tt.want))
|
|
} else if got == nil && tt.want != nil {
|
|
t.Errorf("valueOfKeyword() = %v, want %v", got, tt.want)
|
|
}
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestParseDMISysinfo(t *testing.T) {
|
|
type args struct {
|
|
lines []string
|
|
}
|
|
tests := []struct {
|
|
name string
|
|
args args
|
|
want *types.SSystemInfo
|
|
wantErr bool
|
|
}{
|
|
{
|
|
name: "EmptyInput",
|
|
args: args{lines: nil},
|
|
want: nil,
|
|
wantErr: true,
|
|
},
|
|
{
|
|
name: "NormalInput",
|
|
args: args{lines: []string{
|
|
"Handle 0x000C, DMI type 1, 27 bytes",
|
|
"System Information",
|
|
" Manufacturer: LENOVO",
|
|
" Product Name: 20J6CTO1WW",
|
|
" Version: ThinkPad T470p",
|
|
" Serial Number: PF112JKK",
|
|
" UUID: bca177cc-2bce-11b2-a85c-e98996f19d2f",
|
|
" SKU Number: LENOVO_MT_20J6_BU_Think_FM_ThinkPad T470p",
|
|
}},
|
|
want: &types.SSystemInfo{
|
|
Manufacture: "LENOVO",
|
|
Model: "20J6CTO1WW",
|
|
Version: "ThinkPad T470p",
|
|
SN: "PF112JKK",
|
|
OemName: "lenovo",
|
|
},
|
|
wantErr: false,
|
|
},
|
|
{
|
|
name: "NoneVersionInput",
|
|
args: args{lines: []string{
|
|
" Product Name: 20J6CTO1WW",
|
|
" Version: None",
|
|
" Serial Number: PF112JKK",
|
|
}},
|
|
want: &types.SSystemInfo{
|
|
Model: "20J6CTO1WW",
|
|
Version: "",
|
|
SN: "PF112JKK",
|
|
},
|
|
wantErr: false,
|
|
},
|
|
}
|
|
for _, tt := range tests {
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
got, err := ParseDMISysinfo(tt.args.lines)
|
|
if (err != nil) != tt.wantErr {
|
|
t.Errorf("ParseDMISysinfo() error = %v, wantErr %v", err, tt.wantErr)
|
|
return
|
|
}
|
|
if !reflect.DeepEqual(got, tt.want) {
|
|
t.Errorf("ParseDMISysinfo() = %v, want %v", got, tt.want)
|
|
}
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestParseCPUInfo(t *testing.T) {
|
|
type args struct {
|
|
lines []string
|
|
}
|
|
tests := []struct {
|
|
name string
|
|
args args
|
|
want *types.SCPUInfo
|
|
wantErr bool
|
|
}{
|
|
{
|
|
name: "NormalInput",
|
|
args: args{lines: []string{
|
|
"model name : Intel(R) Xeon(R) CPU E5-2680 v2 @ 2.80GHz",
|
|
"cpu MHz : 2793.238",
|
|
"processor : 0",
|
|
"processor : 1",
|
|
"cache size : 16384 KB",
|
|
}},
|
|
want: &types.SCPUInfo{
|
|
Model: "Intel(R) Xeon(R) CPU E5-2680 v2 @ 2.80GHz",
|
|
Count: 2,
|
|
Freq: 2793,
|
|
Cache: 16384,
|
|
},
|
|
wantErr: false,
|
|
},
|
|
{
|
|
name: "arm",
|
|
args: args{lines: []string{
|
|
"processor\t: 62",
|
|
"BogoMIPS\t: 200.00",
|
|
"Features\t: fp asimd evtstrm aes pmull sha1 sha2 crc32 atomics fphp asimdhp cpuid asimdrdm jscvt fcma dcpop",
|
|
"CPU implementer\t: 0x48",
|
|
"CPU architecture: 8",
|
|
"CPU variant\t: 0x1",
|
|
"CPU part\t: 0xd01",
|
|
"CPU revision\t: 0",
|
|
}},
|
|
want: &types.SCPUInfo{
|
|
Model: "0xd01",
|
|
Microcode: "0",
|
|
Count: 1,
|
|
},
|
|
wantErr: false,
|
|
},
|
|
}
|
|
for _, tt := range tests {
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
got, err := ParseCPUInfo(tt.args.lines)
|
|
if (err != nil) != tt.wantErr {
|
|
t.Errorf("ParseCPUInfo() error = %v, wantErr %v", err, tt.wantErr)
|
|
return
|
|
}
|
|
if !reflect.DeepEqual(got, tt.want) {
|
|
t.Errorf("ParseCPUInfo() = %v, want %v", got, tt.want)
|
|
}
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestParseDMICPUInfo(t *testing.T) {
|
|
type args struct {
|
|
lines []string
|
|
}
|
|
tests := []struct {
|
|
name string
|
|
args args
|
|
want *types.SDMICPUInfo
|
|
}{
|
|
{
|
|
name: "NormalInput",
|
|
args: args{
|
|
lines: []string{"Processor Information"},
|
|
},
|
|
want: &types.SDMICPUInfo{Nodes: 1},
|
|
},
|
|
}
|
|
for _, tt := range tests {
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
if got := ParseDMICPUInfo(tt.args.lines); !reflect.DeepEqual(got, tt.want) {
|
|
t.Errorf("ParseDMICPUInfo() = %v, want %v", got, tt.want)
|
|
}
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestParseDMIMemInfo(t *testing.T) {
|
|
type args struct {
|
|
lines []string
|
|
}
|
|
tests := []struct {
|
|
name string
|
|
args args
|
|
want *types.SDMIMemInfo
|
|
}{
|
|
{
|
|
name: "NormalInputMB",
|
|
args: args{
|
|
lines: []string{
|
|
" Size: 16384 MB",
|
|
" Size: No Module Installed"},
|
|
},
|
|
want: &types.SDMIMemInfo{Total: 16384},
|
|
},
|
|
{
|
|
name: "NormalInputGB",
|
|
args: args{
|
|
lines: []string{
|
|
" Size: 16 GB",
|
|
" Size: No Module Installed"},
|
|
},
|
|
want: &types.SDMIMemInfo{Total: 16 * 1024},
|
|
},
|
|
}
|
|
for _, tt := range tests {
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
if got := ParseDMIMemInfo(tt.args.lines); !reflect.DeepEqual(got, tt.want) {
|
|
t.Errorf("ParseDMIMemInfo() = %v, want %v", got, tt.want)
|
|
}
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestParseNicInfo(t *testing.T) {
|
|
type args struct {
|
|
lines []string
|
|
}
|
|
up := true
|
|
down := false
|
|
mac1Str := "00:22:25:0b:ab:49"
|
|
mac2Str := "00:22:25:0b:ab:50"
|
|
mac1, _ := net.ParseMAC(mac1Str)
|
|
mac2, _ := net.ParseMAC(mac2Str)
|
|
tests := []struct {
|
|
name string
|
|
args args
|
|
want []*types.SNicDevInfo
|
|
}{
|
|
{
|
|
name: "NormalInput",
|
|
args: args{
|
|
lines: []string{
|
|
fmt.Sprintf("eth0 %s 0 1 1500", mac1Str),
|
|
fmt.Sprintf("eth1 %s 0 0 1500", mac2Str),
|
|
},
|
|
},
|
|
want: []*types.SNicDevInfo{
|
|
{Dev: "eth0", Mac: mac1, Speed: 0, Up: &up, Mtu: 1500},
|
|
{Dev: "eth1", Mac: mac2, Speed: 0, Up: &down, Mtu: 1500},
|
|
},
|
|
},
|
|
}
|
|
for _, tt := range tests {
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
got := ParseNicInfo(tt.args.lines)
|
|
gotJson := jsonutils.Marshal(got)
|
|
wantJson := jsonutils.Marshal(tt.want)
|
|
if !gotJson.Equals(wantJson) {
|
|
t.Errorf("got:\n%s\nwant:\n%s", gotJson, wantJson)
|
|
}
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestGetSecureTTYs(t *testing.T) {
|
|
tests := []struct {
|
|
name string
|
|
lines []string
|
|
want []string
|
|
}{
|
|
{
|
|
name: "empty tty",
|
|
lines: []string{"", "#comment"},
|
|
want: []string{},
|
|
},
|
|
{
|
|
name: "ttys",
|
|
lines: []string{
|
|
"tty1",
|
|
"ttyS0",
|
|
"console",
|
|
"#tty0",
|
|
},
|
|
want: []string{
|
|
"tty1",
|
|
"ttyS0",
|
|
"console",
|
|
},
|
|
},
|
|
}
|
|
for _, tt := range tests {
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
if got := GetSecureTTYs(tt.lines); !reflect.DeepEqual(got, tt.want) {
|
|
t.Errorf("GetSecureTTYs() = %v, want %v", got, tt.want)
|
|
}
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestParseDiskInfo(t *testing.T) {
|
|
type args struct {
|
|
lines []string
|
|
driver string
|
|
}
|
|
tests := []struct {
|
|
name string
|
|
args args
|
|
want []*types.SDiskInfo
|
|
}{
|
|
{
|
|
name: "raid",
|
|
args: args{
|
|
lines: []string{
|
|
"sda 15625879552 512 1 megaraid_sas 0x010400 DELL PERC H730P Mini 4.26",
|
|
},
|
|
driver: "raid",
|
|
},
|
|
want: []*types.SDiskInfo{
|
|
{
|
|
Dev: "sda",
|
|
Sector: 15625879552,
|
|
Block: 512,
|
|
Size: 15625879552 * 512 / 1024 / 1024,
|
|
Rotate: true,
|
|
Kernel: "megaraid_sas",
|
|
PCIClass: "0x010400",
|
|
ModuleInfo: "DELL PERC H730P Mini 4.26",
|
|
Driver: "raid",
|
|
},
|
|
},
|
|
},
|
|
}
|
|
for _, tt := range tests {
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
got := ParseDiskInfo(tt.args.lines, tt.args.driver)
|
|
gotStr := jsonutils.Marshal(got).PrettyString()
|
|
wantStr := jsonutils.Marshal(tt.want).PrettyString()
|
|
if !reflect.DeepEqual(gotStr, wantStr) {
|
|
t.Errorf("ParseDiskInfo() = %v, want %v", gotStr, wantStr)
|
|
}
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestParseSGMap(t *testing.T) {
|
|
tests := []struct {
|
|
name string
|
|
lines []string
|
|
want []compute.SGMapItem
|
|
}{
|
|
{
|
|
name: "empty",
|
|
lines: []string{""},
|
|
want: []compute.SGMapItem{},
|
|
},
|
|
{
|
|
name: "normal",
|
|
lines: []string{
|
|
"/dev/sg0 0 0 3 0 0 /dev/sda",
|
|
"/dev/sg1 2 0 4 0 5 /dev/sr0",
|
|
},
|
|
want: []compute.SGMapItem{
|
|
{
|
|
SGDeviceName: "/dev/sg0",
|
|
HostNumber: 0,
|
|
Bus: 0,
|
|
SCSIId: 3,
|
|
Lun: 0,
|
|
Type: 0,
|
|
LinuxDeviceName: "/dev/sda",
|
|
},
|
|
{
|
|
SGDeviceName: "/dev/sg1",
|
|
HostNumber: 2,
|
|
Bus: 0,
|
|
SCSIId: 4,
|
|
Lun: 0,
|
|
Type: 5,
|
|
LinuxDeviceName: "/dev/sr0",
|
|
},
|
|
},
|
|
},
|
|
}
|
|
for _, tt := range tests {
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
if got := ParseSGMap(tt.lines); !reflect.DeepEqual(got, tt.want) {
|
|
t.Errorf("ParseSGMap() = %v, want %v", got, tt.want)
|
|
}
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestParseIPMIUser(t *testing.T) {
|
|
tests := []struct {
|
|
name string
|
|
args []string
|
|
want []compute.IPMIUser
|
|
}{
|
|
{
|
|
name: "empty line",
|
|
args: []string{"ID Name Callin Link Auth IPMI Msg Channel Priv Limit"},
|
|
want: []compute.IPMIUser{},
|
|
},
|
|
{
|
|
name: "users",
|
|
args: []string{
|
|
"ID Name Callin Link Auth IPMI Msg Channel Priv Limit",
|
|
"1 root true false true ADMINISTRATOR",
|
|
"2 admin true false true ADMINISTRATOR",
|
|
"3 true false true USER",
|
|
"4 (Empty User) true false false NO ACCESS",
|
|
"5 (Empty User) true false false NO ACCESS",
|
|
},
|
|
want: []compute.IPMIUser{
|
|
{
|
|
Id: 1,
|
|
Name: "root",
|
|
Priv: "ADMINISTRATOR",
|
|
},
|
|
{
|
|
Id: 2,
|
|
Name: "admin",
|
|
Priv: "ADMINISTRATOR",
|
|
},
|
|
{
|
|
Id: 4,
|
|
Name: "",
|
|
Priv: "",
|
|
},
|
|
{
|
|
Id: 5,
|
|
Name: "",
|
|
Priv: "",
|
|
},
|
|
},
|
|
},
|
|
}
|
|
for _, tt := range tests {
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
if got := ParseIPMIUser(tt.args); !reflect.DeepEqual(got, tt.want) {
|
|
t.Errorf("ParseIPMIUser() = %v, want %v", got, tt.want)
|
|
}
|
|
})
|
|
}
|
|
}
|