VictoriaMetrics/lib/promscrape/discovery/docker/tasks_test.go

package docker

import (
	"reflect"
	"testing"

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

func Test_parseTasks(t *testing.T) {
	type args struct {
		data []byte
	}
	tests := []struct {
		name    string
		args    args
		want    []task
		wantErr bool
	}{
		{
			name: "parse ok",
			args: args{
				data: []byte(`[
  {
    "ID": "t4rdm7j2y9yctbrksiwvsgpu5",
    "Version": {
      "Index": 23
    },
    "Labels": {
	    "label1": "value1"
    },
    "Spec": {
      "ContainerSpec": {
        "Image": "redis:3.0.6@sha256:6a692a76c2081888b589e26e6ec835743119fe453d67ecf03df7de5b73d69842",
        "Init": false
      },
      "Resources": {
        "Limits": {},
        "Reservations": {}
      },
      "Placement": {
        "Platforms": [
          {
            "Architecture": "amd64",
            "OS": "linux"
          }
        ]
      },
      "ForceUpdate": 0
    },
    "ServiceID": "t91nf284wzle1ya09lqvyjgnq",
    "Slot": 1,
    "NodeID": "qauwmifceyvqs0sipvzu8oslu",
    "Status": {
      "State": "running",
      "ContainerStatus": {
        "ContainerID": "33034b69f6fa5f808098208752fd1fe4e0e1ca86311988cea6a73b998cdc62e8",
        "ExitCode": 0
      },
      "PortStatus": {}
    },
    "DesiredState": "running"
  }
]
`),
			},
			want: []task{
				{
					ID:        "t4rdm7j2y9yctbrksiwvsgpu5",
					ServiceID: "t91nf284wzle1ya09lqvyjgnq",
					NodeID:    "qauwmifceyvqs0sipvzu8oslu",
					Labels: map[string]string{
						"label1": "value1",
					},
					DesiredState: "running",
					Slot:         1,
					Status: struct {
						State           string
						ContainerStatus struct{ ContainerID string }
						PortStatus      struct{ Ports []portConfig }
					}{
						State: "running",
						ContainerStatus: struct{ ContainerID string }{
							ContainerID: "33034b69f6fa5f808098208752fd1fe4e0e1ca86311988cea6a73b998cdc62e8",
						},
						PortStatus: struct{ Ports []portConfig }{}},
				},
			},
		},
	}
	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			got, err := parseTasks(tt.args.data)
			if (err != nil) != tt.wantErr {
				t.Errorf("parseTasks() error = %v, wantErr %v", err, tt.wantErr)
				return
			}
			if !reflect.DeepEqual(got, tt.want) {
				t.Errorf("parseTasks() got = %v, want %v", got, tt.want)
			}
		})
	}
}

func Test_addTasksLabels(t *testing.T) {
	type args struct {
		tasks          []task
		nodesLabels    []map[string]string
		servicesLabels []map[string]string
		networksLabels map[string]map[string]string
		services       []service
		port           int
	}
	tests := []struct {
		name string
		args args
		want [][]prompbmarshal.Label
	}{
		{
			name: "adds 1 task with nodes labels",
			args: args{
				port: 9100,
				tasks: []task{
					{
						ID:           "t4rdm7j2y9yctbrksiwvsgpu5",
						ServiceID:    "t91nf284wzle1ya09lqvyjgnq",
						NodeID:       "qauwmifceyvqs0sipvzu8oslu",
						Labels:       map[string]string{},
						DesiredState: "running",
						Slot:         1,
						Status: struct {
							State           string
							ContainerStatus struct{ ContainerID string }
							PortStatus      struct{ Ports []portConfig }
						}{
							State: "running",
							ContainerStatus: struct{ ContainerID string }{
								ContainerID: "33034b69f6fa5f808098208752fd1fe4e0e1ca86311988cea6a73b998cdc62e8",
							},
							PortStatus: struct{ Ports []portConfig }{
								Ports: []portConfig{
									{
										PublishMode:   "ingress",
										Name:          "redis",
										Protocol:      "tcp",
										PublishedPort: 6379,
									},
								},
							}},
					},
				},
				nodesLabels: []map[string]string{
					{
						"__address__":                                   "172.31.40.97:9100",
						"__meta_dockerswarm_node_address":               "172.31.40.97",
						"__meta_dockerswarm_node_availability":          "active",
						"__meta_dockerswarm_node_engine_version":        "19.03.11",
						"__meta_dockerswarm_node_hostname":              "ip-172-31-40-97",
						"__meta_dockerswarm_node_id":                    "qauwmifceyvqs0sipvzu8oslu",
						"__meta_dockerswarm_node_platform_architecture": "x86_64",
						"__meta_dockerswarm_node_platform_os":           "linux",
						"__meta_dockerswarm_node_role":                  "manager",
						"__meta_dockerswarm_node_status":                "ready",
					},
				},
			},
			want: [][]prompbmarshal.Label{
				discoveryutils.GetSortedLabels(map[string]string{
					"__address__":                                   "172.31.40.97:9100",
					"__meta_dockerswarm_node_address":               "172.31.40.97",
					"__meta_dockerswarm_node_availability":          "active",
					"__meta_dockerswarm_node_engine_version":        "19.03.11",
					"__meta_dockerswarm_node_hostname":              "ip-172-31-40-97",
					"__meta_dockerswarm_node_id":                    "qauwmifceyvqs0sipvzu8oslu",
					"__meta_dockerswarm_node_platform_architecture": "x86_64",
					"__meta_dockerswarm_node_platform_os":           "linux",
					"__meta_dockerswarm_node_role":                  "manager",
					"__meta_dockerswarm_node_status":                "ready",
					"__meta_dockerswarm_task_container_id":          "33034b69f6fa5f808098208752fd1fe4e0e1ca86311988cea6a73b998cdc62e8",
					"__meta_dockerswarm_task_desired_state":         "running",
					"__meta_dockerswarm_task_id":                    "t4rdm7j2y9yctbrksiwvsgpu5",
					"__meta_dockerswarm_task_port_publish_mode":     "ingress",
					"__meta_dockerswarm_task_slot":                  "1",
					"__meta_dockerswarm_task_state":                 "running",
				})},
		},
		{
			name: "adds 1 task with nodes, network and services labels",
			args: args{
				port: 9100,
				tasks: []task{
					{
						ID:           "t4rdm7j2y9yctbrksiwvsgpu5",
						ServiceID:    "tgsci5gd31aai3jyudv98pqxf",
						NodeID:       "qauwmifceyvqs0sipvzu8oslu",
						Labels:       map[string]string{},
						DesiredState: "running",
						Slot:         1,
						NetworksAttachments: []struct {
							Addresses []string
							Network   struct{ ID string }
						}{
							{
								Network: struct {
									ID string
								}{
									ID: "qs0hog6ldlei9ct11pr3c77v1",
								},
								Addresses: []string{"10.10.15.15/24"},
							},
						},
						Status: struct {
							State           string
							ContainerStatus struct{ ContainerID string }
							PortStatus      struct{ Ports []portConfig }
						}{
							State: "running",
							ContainerStatus: struct{ ContainerID string }{
								ContainerID: "33034b69f6fa5f808098208752fd1fe4e0e1ca86311988cea6a73b998cdc62e8",
							},
							PortStatus: struct{ Ports []portConfig }{}},
					},
				},
				networksLabels: map[string]map[string]string{
					"qs0hog6ldlei9ct11pr3c77v1": {
						"__meta_dockerswarm_network_id":         "qs0hog6ldlei9ct11pr3c77v1",
						"__meta_dockerswarm_network_ingress":    "true",
						"__meta_dockerswarm_network_internal":   "false",
						"__meta_dockerswarm_network_label_key1": "value1",
						"__meta_dockerswarm_network_name":       "ingress",
						"__meta_dockerswarm_network_scope":      "swarm",
					},
				},
				nodesLabels: []map[string]string{
					{
						"__address__":                                   "172.31.40.97:9100",
						"__meta_dockerswarm_node_address":               "172.31.40.97",
						"__meta_dockerswarm_node_availability":          "active",
						"__meta_dockerswarm_node_engine_version":        "19.03.11",
						"__meta_dockerswarm_node_hostname":              "ip-172-31-40-97",
						"__meta_dockerswarm_node_id":                    "qauwmifceyvqs0sipvzu8oslu",
						"__meta_dockerswarm_node_platform_architecture": "x86_64",
						"__meta_dockerswarm_node_platform_os":           "linux",
						"__meta_dockerswarm_node_role":                  "manager",
						"__meta_dockerswarm_node_status":                "ready",
					},
				},
				services: []service{
					{
						ID: "tgsci5gd31aai3jyudv98pqxf",
						Spec: struct {
							Labels       map[string]string
							Name         string
							TaskTemplate struct {
								ContainerSpec struct {
									Hostname string
									Image    string
								}
							}
							Mode struct {
								Global     interface{}
								Replicated interface{}
							}
						}{
							Labels: map[string]string{},
							Name:   "redis2",
							TaskTemplate: struct {
								ContainerSpec struct {
									Hostname string
									Image    string
								}
							}{
								ContainerSpec: struct {
									Hostname string
									Image    string
								}{
									Hostname: "node1",
									Image:    "redis:3.0.6@sha256:6a692a76c2081888b589e26e6ec835743119fe453d67ecf03df7de5b73d69842",
								},
							},
							Mode: struct {
								Global     interface{}
								Replicated interface{}
							}{
								Replicated: map[string]interface{}{},
							},
						},
						Endpoint: struct {
							Ports      []portConfig
							VirtualIPs []struct {
								NetworkID string
								Addr      string
							}
						}{
							Ports: []portConfig{
								{
									Protocol:    "tcp",
									Name:        "redis",
									PublishMode: "ingress",
								},
							}, VirtualIPs: []struct {
								NetworkID string
								Addr      string
							}{
								{
									NetworkID: "qs0hog6ldlei9ct11pr3c77v1",
									Addr:      "10.0.0.3/24",
								},
							},
						},
					},
				},
				servicesLabels: []map[string]string{},
			},
			want: [][]prompbmarshal.Label{
				discoveryutils.GetSortedLabels(map[string]string{
					"__address":                                     "10.10.15.15:0",
					"__address__":                                   "172.31.40.97:9100",
					"__meta_dockerswarm_network_id":                 "qs0hog6ldlei9ct11pr3c77v1",
					"__meta_dockerswarm_network_ingress":            "true",
					"__meta_dockerswarm_network_internal":           "false",
					"__meta_dockerswarm_network_label_key1":         "value1",
					"__meta_dockerswarm_network_name":               "ingress",
					"__meta_dockerswarm_network_scope":              "swarm",
					"__meta_dockerswarm_node_address":               "172.31.40.97",
					"__meta_dockerswarm_node_availability":          "active",
					"__meta_dockerswarm_node_engine_version":        "19.03.11",
					"__meta_dockerswarm_node_hostname":              "ip-172-31-40-97",
					"__meta_dockerswarm_node_id":                    "qauwmifceyvqs0sipvzu8oslu",
					"__meta_dockerswarm_node_platform_architecture": "x86_64",
					"__meta_dockerswarm_node_platform_os":           "linux",
					"__meta_dockerswarm_node_role":                  "manager",
					"__meta_dockerswarm_node_status":                "ready",
					"__meta_dockerswarm_task_container_id":          "33034b69f6fa5f808098208752fd1fe4e0e1ca86311988cea6a73b998cdc62e8",
					"__meta_dockerswarm_task_desired_state":         "running",
					"__meta_dockerswarm_task_id":                    "t4rdm7j2y9yctbrksiwvsgpu5",
					"__meta_dockerswarm_task_port_publish_mode":     "ingress",
					"__meta_dockerswarm_task_slot":                  "1",
					"__meta_dockerswarm_task_state":                 "running",
				}),
			},
		},
	}
	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			got := addTasksLabels(tt.args.tasks, tt.args.nodesLabels, tt.args.servicesLabels, tt.args.networksLabels, tt.args.services, tt.args.port)
			var sortedLabelss [][]prompbmarshal.Label
			for _, labels := range got {
				sortedLabelss = append(sortedLabelss, discoveryutils.GetSortedLabels(labels))
			}
			if !reflect.DeepEqual(sortedLabelss, tt.want) {
				t.Errorf("addTasksLabels() \ngot  %v, \nwant %v", sortedLabelss, tt.want)
			}
		})
	}
}