mirror of
https://github.com/VictoriaMetrics/VictoriaMetrics.git
synced 2024-12-18 14:40:26 +01:00
9edeecabc8
This makes easier to read and debug these tests. This also reduces test lines count by 15% from 3K to 2.5K . See https://itnext.io/f-tests-as-a-replacement-for-table-driven-tests-in-go-8814a8b19e9e . While at it, consistently use t.Fatal* instead of t.Error*, since t.Error* usually leads to more complicated and fragile tests, while it doesn't bring any practical benefits over t.Fatal*.
75 lines
1.9 KiB
Go
75 lines
1.9 KiB
Go
package digitalocean
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/VictoriaMetrics/VictoriaMetrics/lib/promscrape/discoveryutils"
|
|
"github.com/VictoriaMetrics/VictoriaMetrics/lib/promutils"
|
|
)
|
|
|
|
func TestAddDropletLabels(t *testing.T) {
|
|
f := func(droplets []droplet, labelssExpected []*promutils.Labels) {
|
|
t.Helper()
|
|
|
|
labelss := addDropletLabels(droplets, 9100)
|
|
discoveryutils.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 := []*promutils.Labels{
|
|
promutils.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)
|
|
}
|