mirror of
https://github.com/hanzoai/metrics.git
synced 2026-10-30 20:03:39 +00:00
75 lines
1.9 KiB
Go
75 lines
1.9 KiB
Go
package digitalocean
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/VictoriaMetrics/VictoriaMetrics/lib/promscrape/discoveryutil"
|
|
"github.com/VictoriaMetrics/VictoriaMetrics/lib/promutil"
|
|
)
|
|
|
|
func TestAddDropletLabels(t *testing.T) {
|
|
f := func(droplets []droplet, labelssExpected []*promutil.Labels) {
|
|
t.Helper()
|
|
|
|
labelss := addDropletLabels(droplets, 9100)
|
|
discoveryutil.TestEqualLabelss(t, labelss, labelssExpected)
|
|
}
|
|
|
|
// base labels add test
|
|
droplets := []droplet{
|
|
{
|
|
ID: 15,
|
|
Tags: []string{"private", "test"},
|
|
Status: "active",
|
|
Name: "ubuntu-1",
|
|
Region: dropletRegion{
|
|
Slug: "do",
|
|
},
|
|
Features: []string{"feature-1", "feature-2"},
|
|
SizeSlug: "base-1",
|
|
VpcUUID: "vpc-1",
|
|
Image: dropletImage{
|
|
Name: "ubuntu",
|
|
Slug: "18",
|
|
},
|
|
Networks: networks{
|
|
V4: []network{
|
|
{
|
|
Type: "public",
|
|
IPAddress: "100.100.100.100",
|
|
},
|
|
{
|
|
Type: "private",
|
|
IPAddress: "10.10.10.10",
|
|
},
|
|
},
|
|
V6: []network{
|
|
{
|
|
Type: "public",
|
|
IPAddress: "::1",
|
|
},
|
|
},
|
|
},
|
|
},
|
|
}
|
|
labelssExpected := []*promutil.Labels{
|
|
promutil.NewLabelsFromMap(map[string]string{
|
|
"__address__": "100.100.100.100:9100",
|
|
"__meta_digitalocean_droplet_id": "15",
|
|
"__meta_digitalocean_droplet_name": "ubuntu-1",
|
|
"__meta_digitalocean_features": ",feature-1,feature-2,",
|
|
"__meta_digitalocean_image": "18",
|
|
"__meta_digitalocean_image_name": "ubuntu",
|
|
"__meta_digitalocean_private_ipv4": "10.10.10.10",
|
|
"__meta_digitalocean_public_ipv4": "100.100.100.100",
|
|
"__meta_digitalocean_public_ipv6": "::1",
|
|
"__meta_digitalocean_region": "do",
|
|
"__meta_digitalocean_size": "base-1",
|
|
"__meta_digitalocean_status": "active",
|
|
"__meta_digitalocean_tags": ",private,test,",
|
|
"__meta_digitalocean_vpc": "vpc-1",
|
|
}),
|
|
}
|
|
f(droplets, labelssExpected)
|
|
}
|