VictoriaMetrics/lib/promscrape/discovery/digitalocean/digitalocean_test.go

package digitalocean

import (
	"reflect"
	"testing"

	"github.com/VictoriaMetrics/VictoriaMetrics/lib/prompbmarshal"
	"github.com/VictoriaMetrics/VictoriaMetrics/lib/promscrape/discoveryutils"
)

func Test_addDropletLabels(t *testing.T) {
	type args struct {
		droplets    []droplet
		defaultPort int
	}
	tests := []struct {
		name string
		args args
		want [][]prompbmarshal.Label
	}{
		{
			name: "base labels add test",
			args: args{
				droplets: []droplet{
					{
						ID:     15,
						Tags:   []string{"private", "test"},
						Status: "active",
						Name:   "ubuntu-1",
						Region: struct {
							Slug string `json:"slug"`
						}(struct{ Slug string }{Slug: "do"}),
						Features: []string{"feature-1", "feature-2"},
						SizeSlug: "base-1",
						VpcUUID:  "vpc-1",
						Image: struct {
							Name string `json:"name"`
							Slug string `json:"slug"`
						}(struct {
							Name string
							Slug string
						}{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",
								},
							},
						},
					},
				},
				defaultPort: 9100,
			},
			want: [][]prompbmarshal.Label{
				discoveryutils.GetSortedLabels(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",
				}),
			},
		},
	}
	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			got := addDropletLabels(tt.args.droplets, tt.args.defaultPort)
			var sortedLabelss [][]prompbmarshal.Label
			for _, labels := range got {
				sortedLabelss = append(sortedLabelss, discoveryutils.GetSortedLabels(labels))
			}
			if !reflect.DeepEqual(sortedLabelss, tt.want) {
				t.Errorf("addTasksLabels() \ngot  \n%v\n, \nwant \n%v\n", sortedLabelss, tt.want)
			}

		})
	}
}