mirror of
https://github.com/VictoriaMetrics/VictoriaMetrics.git
synced 2024-12-22 16:36:27 +01:00
48 lines
1.5 KiB
Go
48 lines
1.5 KiB
Go
package azure
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/VictoriaMetrics/VictoriaMetrics/lib/promscrape/discoveryutils"
|
|
"github.com/VictoriaMetrics/VictoriaMetrics/lib/promutils"
|
|
)
|
|
|
|
func TestAppendMachineLabels(t *testing.T) {
|
|
f := func(name string, vms []virtualMachine, expectedLabels []*promutils.Labels) {
|
|
t.Run(name, func(t *testing.T) {
|
|
labelss := appendMachineLabels(vms, 80, &SDConfig{SubscriptionID: "some-id"})
|
|
discoveryutils.TestEqualLabelss(t, labelss, expectedLabels)
|
|
})
|
|
}
|
|
f("single vm", []virtualMachine{
|
|
{
|
|
Name: "vm-1",
|
|
ID: "id-2",
|
|
Type: "Azure",
|
|
Location: "eu-west-1",
|
|
Properties: virtualMachineProperties{
|
|
OsProfile: osProfile{ComputerName: "test-1"},
|
|
StorageProfile: storageProfile{OsDisk: osDisk{OsType: "Linux"}},
|
|
HardwareProfile: hardwareProfile{VMSize: "big"},
|
|
},
|
|
Tags: map[string]string{"key-1": "value-1"},
|
|
ipAddresses: []vmIPAddress{
|
|
{privateIP: "10.10.10.1"},
|
|
},
|
|
},
|
|
}, []*promutils.Labels{
|
|
promutils.NewLabelsFromMap(map[string]string{
|
|
"__address__": "10.10.10.1:80",
|
|
"__meta_azure_machine_id": "id-2",
|
|
"__meta_azure_subscription_id": "some-id",
|
|
"__meta_azure_machine_os_type": "Linux",
|
|
"__meta_azure_machine_name": "vm-1",
|
|
"__meta_azure_machine_computer_name": "test-1",
|
|
"__meta_azure_machine_location": "eu-west-1",
|
|
"__meta_azure_machine_private_ip": "10.10.10.1",
|
|
"__meta_azure_machine_size": "big",
|
|
"__meta_azure_machine_tag_key_1": "value-1",
|
|
}),
|
|
})
|
|
}
|