449 lines
12 KiB
Go
449 lines
12 KiB
Go
package vsphere
|
|
|
|
import (
|
|
"context"
|
|
"crypto/tls"
|
|
"fmt"
|
|
"regexp"
|
|
"sort"
|
|
"sync"
|
|
"sync/atomic"
|
|
"testing"
|
|
"time"
|
|
"unsafe"
|
|
|
|
"github.com/influxdata/telegraf/internal"
|
|
itls "github.com/influxdata/telegraf/internal/tls"
|
|
"github.com/influxdata/telegraf/testutil"
|
|
"github.com/influxdata/toml"
|
|
"github.com/stretchr/testify/require"
|
|
"github.com/vmware/govmomi/object"
|
|
"github.com/vmware/govmomi/simulator"
|
|
"github.com/vmware/govmomi/vim25/mo"
|
|
"github.com/vmware/govmomi/vim25/types"
|
|
)
|
|
|
|
var configHeader = `
|
|
[agent]
|
|
interval = "10s"
|
|
round_interval = true
|
|
metric_batch_size = 1000
|
|
metric_buffer_limit = 10000
|
|
collection_jitter = "0s"
|
|
flush_interval = "10s"
|
|
flush_jitter = "0s"
|
|
precision = ""
|
|
debug = false
|
|
quiet = false
|
|
logfile = ""
|
|
hostname = ""
|
|
omit_hostname = false
|
|
`
|
|
|
|
func defaultVSphere() *VSphere {
|
|
return &VSphere{
|
|
Log: testutil.Logger{},
|
|
ClusterMetricInclude: []string{
|
|
"cpu.usage.*",
|
|
"cpu.usagemhz.*",
|
|
"mem.usage.*",
|
|
"mem.active.*"},
|
|
ClusterMetricExclude: nil,
|
|
ClusterInclude: []string{"/**"},
|
|
HostMetricInclude: []string{
|
|
"cpu.coreUtilization.average",
|
|
"cpu.costop.summation",
|
|
"cpu.demand.average",
|
|
"cpu.idle.summation",
|
|
"cpu.latency.average",
|
|
"cpu.readiness.average",
|
|
"cpu.ready.summation",
|
|
"cpu.swapwait.summation",
|
|
"cpu.usage.average",
|
|
"cpu.usagemhz.average",
|
|
"cpu.used.summation",
|
|
"cpu.utilization.average",
|
|
"cpu.wait.summation",
|
|
"disk.deviceReadLatency.average",
|
|
"disk.deviceWriteLatency.average",
|
|
"disk.kernelReadLatency.average",
|
|
"disk.kernelWriteLatency.average",
|
|
"disk.numberReadAveraged.average",
|
|
"disk.numberWriteAveraged.average",
|
|
"disk.read.average",
|
|
"disk.totalReadLatency.average",
|
|
"disk.totalWriteLatency.average",
|
|
"disk.write.average",
|
|
"mem.active.average",
|
|
"mem.latency.average",
|
|
"mem.state.latest",
|
|
"mem.swapin.average",
|
|
"mem.swapinRate.average",
|
|
"mem.swapout.average",
|
|
"mem.swapoutRate.average",
|
|
"mem.totalCapacity.average",
|
|
"mem.usage.average",
|
|
"mem.vmmemctl.average",
|
|
"net.bytesRx.average",
|
|
"net.bytesTx.average",
|
|
"net.droppedRx.summation",
|
|
"net.droppedTx.summation",
|
|
"net.errorsRx.summation",
|
|
"net.errorsTx.summation",
|
|
"net.usage.average",
|
|
"power.power.average",
|
|
"storageAdapter.numberReadAveraged.average",
|
|
"storageAdapter.numberWriteAveraged.average",
|
|
"storageAdapter.read.average",
|
|
"storageAdapter.write.average",
|
|
"sys.uptime.latest"},
|
|
HostMetricExclude: nil,
|
|
HostInclude: []string{"/**"},
|
|
VMMetricInclude: []string{
|
|
"cpu.demand.average",
|
|
"cpu.idle.summation",
|
|
"cpu.latency.average",
|
|
"cpu.readiness.average",
|
|
"cpu.ready.summation",
|
|
"cpu.run.summation",
|
|
"cpu.usagemhz.average",
|
|
"cpu.used.summation",
|
|
"cpu.wait.summation",
|
|
"mem.active.average",
|
|
"mem.granted.average",
|
|
"mem.latency.average",
|
|
"mem.swapin.average",
|
|
"mem.swapinRate.average",
|
|
"mem.swapout.average",
|
|
"mem.swapoutRate.average",
|
|
"mem.usage.average",
|
|
"mem.vmmemctl.average",
|
|
"net.bytesRx.average",
|
|
"net.bytesTx.average",
|
|
"net.droppedRx.summation",
|
|
"net.droppedTx.summation",
|
|
"net.usage.average",
|
|
"power.power.average",
|
|
"virtualDisk.numberReadAveraged.average",
|
|
"virtualDisk.numberWriteAveraged.average",
|
|
"virtualDisk.read.average",
|
|
"virtualDisk.readOIO.latest",
|
|
"virtualDisk.throughput.usage.average",
|
|
"virtualDisk.totalReadLatency.average",
|
|
"virtualDisk.totalWriteLatency.average",
|
|
"virtualDisk.write.average",
|
|
"virtualDisk.writeOIO.latest",
|
|
"sys.uptime.latest"},
|
|
VMMetricExclude: nil,
|
|
VMInclude: []string{"/**"},
|
|
DatastoreMetricInclude: []string{
|
|
"disk.used.*",
|
|
"disk.provsioned.*"},
|
|
DatastoreMetricExclude: nil,
|
|
DatastoreInclude: []string{"/**"},
|
|
DatacenterMetricInclude: nil,
|
|
DatacenterMetricExclude: nil,
|
|
DatacenterInclude: []string{"/**"},
|
|
ClientConfig: itls.ClientConfig{InsecureSkipVerify: true},
|
|
|
|
MaxQueryObjects: 256,
|
|
MaxQueryMetrics: 256,
|
|
ObjectDiscoveryInterval: internal.Duration{Duration: time.Second * 300},
|
|
Timeout: internal.Duration{Duration: time.Second * 20},
|
|
ForceDiscoverOnInit: true,
|
|
DiscoverConcurrency: 1,
|
|
CollectConcurrency: 1,
|
|
}
|
|
}
|
|
|
|
func createSim(folders int) (*simulator.Model, *simulator.Server, error) {
|
|
model := simulator.VPX()
|
|
|
|
model.Folder = folders
|
|
model.Datacenter = 2
|
|
//model.App = 1
|
|
|
|
err := model.Create()
|
|
if err != nil {
|
|
return nil, nil, err
|
|
}
|
|
|
|
model.Service.TLS = new(tls.Config)
|
|
|
|
s := model.Service.NewServer()
|
|
return model, s, nil
|
|
}
|
|
|
|
func testAlignUniform(t *testing.T, n int) {
|
|
now := time.Now().Truncate(60 * time.Second)
|
|
info := make([]types.PerfSampleInfo, n)
|
|
values := make([]int64, n)
|
|
for i := 0; i < n; i++ {
|
|
info[i] = types.PerfSampleInfo{
|
|
Timestamp: now.Add(time.Duration(20*i) * time.Second),
|
|
Interval: 20,
|
|
}
|
|
values[i] = 1
|
|
}
|
|
newInfo, newValues := alignSamples(info, values, 60*time.Second)
|
|
require.Equal(t, n/3, len(newInfo), "Aligned infos have wrong size")
|
|
require.Equal(t, n/3, len(newValues), "Aligned values have wrong size")
|
|
for _, v := range newValues {
|
|
require.Equal(t, 1.0, v, "Aligned value should be 1")
|
|
}
|
|
}
|
|
|
|
func TestAlignMetrics(t *testing.T) {
|
|
testAlignUniform(t, 3)
|
|
testAlignUniform(t, 30)
|
|
testAlignUniform(t, 333)
|
|
|
|
// 20s to 60s of 1,2,3,1,2,3... (should average to 2)
|
|
n := 30
|
|
now := time.Now().Truncate(60 * time.Second)
|
|
info := make([]types.PerfSampleInfo, n)
|
|
values := make([]int64, n)
|
|
for i := 0; i < n; i++ {
|
|
info[i] = types.PerfSampleInfo{
|
|
Timestamp: now.Add(time.Duration(20*i) * time.Second),
|
|
Interval: 20,
|
|
}
|
|
values[i] = int64(i%3 + 1)
|
|
}
|
|
newInfo, newValues := alignSamples(info, values, 60*time.Second)
|
|
require.Equal(t, n/3, len(newInfo), "Aligned infos have wrong size")
|
|
require.Equal(t, n/3, len(newValues), "Aligned values have wrong size")
|
|
for _, v := range newValues {
|
|
require.Equal(t, 2.0, v, "Aligned value should be 2")
|
|
}
|
|
}
|
|
|
|
func TestParseConfig(t *testing.T) {
|
|
v := VSphere{}
|
|
c := v.SampleConfig()
|
|
p := regexp.MustCompile("\n#")
|
|
fmt.Printf("Source=%s", p.ReplaceAllLiteralString(c, "\n"))
|
|
c = configHeader + "\n[[inputs.vsphere]]\n" + p.ReplaceAllLiteralString(c, "\n")
|
|
fmt.Printf("Source=%s", c)
|
|
tab, err := toml.Parse([]byte(c))
|
|
require.NoError(t, err)
|
|
require.NotNil(t, tab)
|
|
}
|
|
|
|
func TestThrottledExecutor(t *testing.T) {
|
|
max := int64(0)
|
|
ngr := int64(0)
|
|
n := 10000
|
|
var mux sync.Mutex
|
|
results := make([]int, 0, n)
|
|
te := NewThrottledExecutor(5)
|
|
for i := 0; i < n; i++ {
|
|
func(i int) {
|
|
te.Run(context.Background(), func() {
|
|
atomic.AddInt64(&ngr, 1)
|
|
mux.Lock()
|
|
defer mux.Unlock()
|
|
results = append(results, i*2)
|
|
if ngr > max {
|
|
max = ngr
|
|
}
|
|
time.Sleep(100 * time.Microsecond)
|
|
atomic.AddInt64(&ngr, -1)
|
|
})
|
|
}(i)
|
|
}
|
|
te.Wait()
|
|
sort.Ints(results)
|
|
for i := 0; i < n; i++ {
|
|
require.Equal(t, results[i], i*2, "Some jobs didn't run")
|
|
}
|
|
require.Equal(t, int64(5), max, "Wrong number of goroutines spawned")
|
|
}
|
|
|
|
func TestMaxQuery(t *testing.T) {
|
|
// Don't run test on 32-bit machines due to bug in simulator.
|
|
// https://github.com/vmware/govmomi/issues/1330
|
|
var i int
|
|
if unsafe.Sizeof(i) < 8 {
|
|
return
|
|
}
|
|
m, s, err := createSim(0)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
defer m.Remove()
|
|
defer s.Close()
|
|
|
|
v := defaultVSphere()
|
|
v.MaxQueryMetrics = 256
|
|
ctx := context.Background()
|
|
c, err := NewClient(ctx, s.URL, v)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
require.Equal(t, 256, v.MaxQueryMetrics)
|
|
|
|
om := object.NewOptionManager(c.Client.Client, *c.Client.Client.ServiceContent.Setting)
|
|
err = om.Update(ctx, []types.BaseOptionValue{&types.OptionValue{
|
|
Key: "config.vpxd.stats.maxQueryMetrics",
|
|
Value: "42",
|
|
}})
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
v.MaxQueryMetrics = 256
|
|
ctx = context.Background()
|
|
c2, err := NewClient(ctx, s.URL, v)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
require.Equal(t, 42, v.MaxQueryMetrics)
|
|
c.close()
|
|
c2.close()
|
|
}
|
|
|
|
func testLookupVM(ctx context.Context, t *testing.T, f *Finder, path string, expected int, expectedName string) {
|
|
poweredOn := types.VirtualMachinePowerState("poweredOn")
|
|
var vm []mo.VirtualMachine
|
|
err := f.Find(ctx, "VirtualMachine", path, &vm)
|
|
require.NoError(t, err)
|
|
require.Equal(t, expected, len(vm))
|
|
if expectedName != "" {
|
|
require.Equal(t, expectedName, vm[0].Name)
|
|
}
|
|
for _, v := range vm {
|
|
require.Equal(t, poweredOn, v.Runtime.PowerState)
|
|
}
|
|
}
|
|
|
|
func TestFinder(t *testing.T) {
|
|
// Don't run test on 32-bit machines due to bug in simulator.
|
|
// https://github.com/vmware/govmomi/issues/1330
|
|
var i int
|
|
if unsafe.Sizeof(i) < 8 {
|
|
return
|
|
}
|
|
|
|
m, s, err := createSim(0)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
defer m.Remove()
|
|
defer s.Close()
|
|
|
|
v := defaultVSphere()
|
|
ctx := context.Background()
|
|
|
|
c, err := NewClient(ctx, s.URL, v)
|
|
|
|
f := Finder{c}
|
|
|
|
var dc []mo.Datacenter
|
|
err = f.Find(ctx, "Datacenter", "/DC0", &dc)
|
|
require.NoError(t, err)
|
|
require.Equal(t, 1, len(dc))
|
|
require.Equal(t, "DC0", dc[0].Name)
|
|
|
|
var host []mo.HostSystem
|
|
err = f.Find(ctx, "HostSystem", "/DC0/host/DC0_H0/DC0_H0", &host)
|
|
require.NoError(t, err)
|
|
require.Equal(t, 1, len(host))
|
|
require.Equal(t, "DC0_H0", host[0].Name)
|
|
|
|
host = []mo.HostSystem{}
|
|
err = f.Find(ctx, "HostSystem", "/DC0/host/DC0_C0/DC0_C0_H0", &host)
|
|
require.NoError(t, err)
|
|
require.Equal(t, 1, len(host))
|
|
require.Equal(t, "DC0_C0_H0", host[0].Name)
|
|
|
|
host = []mo.HostSystem{}
|
|
err = f.Find(ctx, "HostSystem", "/DC0/host/DC0_C0/*", &host)
|
|
require.NoError(t, err)
|
|
require.Equal(t, 3, len(host))
|
|
|
|
var vm []mo.VirtualMachine
|
|
testLookupVM(ctx, t, &f, "/DC0/vm/DC0_H0_VM0", 1, "")
|
|
testLookupVM(ctx, t, &f, "/DC0/vm/DC0_C0*", 2, "")
|
|
testLookupVM(ctx, t, &f, "/DC0/*/DC0_H0_VM0", 1, "DC0_H0_VM0")
|
|
testLookupVM(ctx, t, &f, "/DC0/*/DC0_H0_*", 2, "")
|
|
testLookupVM(ctx, t, &f, "/DC0/**/DC0_H0_VM*", 2, "")
|
|
testLookupVM(ctx, t, &f, "/DC0/**", 4, "")
|
|
testLookupVM(ctx, t, &f, "/DC1/**", 4, "")
|
|
testLookupVM(ctx, t, &f, "/**", 8, "")
|
|
testLookupVM(ctx, t, &f, "/**/vm/**", 8, "")
|
|
testLookupVM(ctx, t, &f, "/*/host/**/*DC*", 8, "")
|
|
testLookupVM(ctx, t, &f, "/*/host/**/*DC*VM*", 8, "")
|
|
testLookupVM(ctx, t, &f, "/*/host/**/*DC*/*/*DC*", 4, "")
|
|
|
|
vm = []mo.VirtualMachine{}
|
|
err = f.FindAll(ctx, "VirtualMachine", []string{"/DC0/vm/DC0_H0*", "/DC0/vm/DC0_C0*"}, &vm)
|
|
require.NoError(t, err)
|
|
require.Equal(t, 4, len(vm))
|
|
|
|
}
|
|
|
|
func TestFolders(t *testing.T) {
|
|
// Don't run test on 32-bit machines due to bug in simulator.
|
|
// https://github.com/vmware/govmomi/issues/1330
|
|
var i int
|
|
if unsafe.Sizeof(i) < 8 {
|
|
return
|
|
}
|
|
|
|
m, s, err := createSim(1)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
defer m.Remove()
|
|
defer s.Close()
|
|
|
|
v := defaultVSphere()
|
|
ctx := context.Background()
|
|
|
|
c, err := NewClient(ctx, s.URL, v)
|
|
|
|
f := Finder{c}
|
|
|
|
var folder []mo.Folder
|
|
err = f.Find(ctx, "Folder", "/F0", &folder)
|
|
require.NoError(t, err)
|
|
require.Equal(t, 1, len(folder))
|
|
require.Equal(t, "F0", folder[0].Name)
|
|
|
|
var dc []mo.Datacenter
|
|
err = f.Find(ctx, "Datacenter", "/F0/DC1", &dc)
|
|
require.NoError(t, err)
|
|
require.Equal(t, 1, len(dc))
|
|
require.Equal(t, "DC1", dc[0].Name)
|
|
|
|
testLookupVM(ctx, t, &f, "/F0/DC0/vm/**/F*", 0, "")
|
|
testLookupVM(ctx, t, &f, "/F0/DC1/vm/**/F*/*VM*", 4, "")
|
|
testLookupVM(ctx, t, &f, "/F0/DC1/vm/**/F*/**", 4, "")
|
|
}
|
|
|
|
func TestAll(t *testing.T) {
|
|
// Don't run test on 32-bit machines due to bug in simulator.
|
|
// https://github.com/vmware/govmomi/issues/1330
|
|
var i int
|
|
if unsafe.Sizeof(i) < 8 {
|
|
return
|
|
}
|
|
|
|
m, s, err := createSim(0)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
defer m.Remove()
|
|
defer s.Close()
|
|
|
|
var acc testutil.Accumulator
|
|
v := defaultVSphere()
|
|
v.Vcenters = []string{s.URL.String()}
|
|
v.Start(&acc)
|
|
defer v.Stop()
|
|
require.NoError(t, v.Gather(&acc))
|
|
require.Equal(t, 0, len(acc.Errors), fmt.Sprintf("Errors found: %s", acc.Errors))
|
|
require.True(t, len(acc.Metrics) > 0, "No metrics were collected")
|
|
}
|