137 lines
4.3 KiB
Go
137 lines
4.3 KiB
Go
package proxmox
|
|
|
|
import (
|
|
"github.com/bmizerany/assert"
|
|
"github.com/influxdata/telegraf/testutil"
|
|
"github.com/stretchr/testify/require"
|
|
"net/url"
|
|
"strings"
|
|
"testing"
|
|
)
|
|
|
|
var nodeSearchDomainTestData = `{"data":{"search":"test.example.com","dns1":"1.0.0.1"}}`
|
|
var qemuTestData = `{"data":[{"name":"qemu1","status":"running","maxdisk":10737418240,"cpu":0.029336643550795,"vmid":"113","uptime":2159739,"disk":0,"maxmem":2147483648,"mem":1722451796}]}`
|
|
var qemuConfigTestData = `{"data":{"hostname":"qemu1","searchdomain":"test.example.com"}}`
|
|
var lxcTestData = `{"data":[{"vmid":"111","type":"lxc","uptime":2078164,"swap":9412608,"disk":"744189952","maxmem":536870912,"mem":98500608,"maxswap":536870912,"cpu":0.00371567669193613,"status":"running","maxdisk":"5217320960","name":"container1"}]}`
|
|
var lxcConfigTestData = `{"data":{"hostname":"container1","searchdomain":"test.example.com"}}`
|
|
|
|
func performTestRequest(px *Proxmox, apiUrl string, method string, data url.Values) ([]byte, error) {
|
|
var bytedata = []byte("")
|
|
|
|
if strings.HasSuffix(apiUrl, "dns") {
|
|
bytedata = []byte(nodeSearchDomainTestData)
|
|
} else if strings.HasSuffix(apiUrl, "qemu") {
|
|
bytedata = []byte(qemuTestData)
|
|
} else if strings.HasSuffix(apiUrl, "113/config") {
|
|
bytedata = []byte(qemuConfigTestData)
|
|
} else if strings.HasSuffix(apiUrl, "lxc") {
|
|
bytedata = []byte(lxcTestData)
|
|
} else if strings.HasSuffix(apiUrl, "111/config") {
|
|
bytedata = []byte(lxcConfigTestData)
|
|
}
|
|
|
|
return bytedata, nil
|
|
}
|
|
|
|
func setUp(t *testing.T) *Proxmox {
|
|
px := &Proxmox{
|
|
requestFunction: performTestRequest,
|
|
}
|
|
|
|
require.NoError(t, px.Init())
|
|
|
|
// Override hostname and logger for test
|
|
px.hostname = "testnode"
|
|
px.Log = testutil.Logger{}
|
|
return px
|
|
}
|
|
|
|
func TestGetNodeSearchDomain(t *testing.T) {
|
|
px := setUp(t)
|
|
|
|
err := getNodeSearchDomain(px)
|
|
|
|
require.NoError(t, err)
|
|
assert.Equal(t, px.nodeSearchDomain, "test.example.com")
|
|
}
|
|
|
|
func TestGatherLxcData(t *testing.T) {
|
|
px := setUp(t)
|
|
px.nodeSearchDomain = "test.example.com"
|
|
|
|
acc := &testutil.Accumulator{}
|
|
gatherLxcData(px, acc)
|
|
|
|
assert.Equal(t, acc.NFields(), 15)
|
|
testFields := map[string]interface{}{
|
|
"status": "running",
|
|
"uptime": int64(2078164),
|
|
"cpuload": float64(0.00371567669193613),
|
|
"mem_used": int64(98500608),
|
|
"mem_total": int64(536870912),
|
|
"mem_free": int64(438370304),
|
|
"mem_used_percentage": float64(18.34716796875),
|
|
"swap_used": int64(9412608),
|
|
"swap_total": int64(536870912),
|
|
"swap_free": int64(527458304),
|
|
"swap_used_percentage": float64(1.75323486328125),
|
|
"disk_used": int64(744189952),
|
|
"disk_total": int64(5217320960),
|
|
"disk_free": int64(4473131008),
|
|
"disk_used_percentage": float64(14.26383306117322),
|
|
}
|
|
testTags := map[string]string{
|
|
"node_fqdn": "testnode.test.example.com",
|
|
"vm_name": "container1",
|
|
"vm_fqdn": "container1.test.example.com",
|
|
"vm_type": "lxc",
|
|
}
|
|
acc.AssertContainsTaggedFields(t, "proxmox", testFields, testTags)
|
|
}
|
|
|
|
func TestGatherQemuData(t *testing.T) {
|
|
px := setUp(t)
|
|
px.nodeSearchDomain = "test.example.com"
|
|
|
|
acc := &testutil.Accumulator{}
|
|
gatherQemuData(px, acc)
|
|
|
|
assert.Equal(t, acc.NFields(), 15)
|
|
testFields := map[string]interface{}{
|
|
"status": "running",
|
|
"uptime": int64(2159739),
|
|
"cpuload": float64(0.029336643550795),
|
|
"mem_used": int64(1722451796),
|
|
"mem_total": int64(2147483648),
|
|
"mem_free": int64(425031852),
|
|
"mem_used_percentage": float64(80.20791206508875),
|
|
"swap_used": int64(0),
|
|
"swap_total": int64(0),
|
|
"swap_free": int64(0),
|
|
"swap_used_percentage": float64(0),
|
|
"disk_used": int64(0),
|
|
"disk_total": int64(10737418240),
|
|
"disk_free": int64(10737418240),
|
|
"disk_used_percentage": float64(0),
|
|
}
|
|
testTags := map[string]string{
|
|
"node_fqdn": "testnode.test.example.com",
|
|
"vm_name": "qemu1",
|
|
"vm_fqdn": "qemu1.test.example.com",
|
|
"vm_type": "qemu",
|
|
}
|
|
acc.AssertContainsTaggedFields(t, "proxmox", testFields, testTags)
|
|
}
|
|
|
|
func TestGather(t *testing.T) {
|
|
px := setUp(t)
|
|
px.nodeSearchDomain = "test.example.com"
|
|
|
|
acc := &testutil.Accumulator{}
|
|
err := px.Gather(acc)
|
|
require.NoError(t, err)
|
|
|
|
// Results from both tests above
|
|
assert.Equal(t, acc.NFields(), 30)
|
|
}
|