// Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
// SPDX-License-Identifier: MIT

package config

import (
	"testing"

	"github.com/stretchr/testify/assert"

	"github.com/aws/amazon-cloudwatch-agent/tool/runtime"
	"github.com/aws/amazon-cloudwatch-agent/tool/util"
)

func TestMetrics_ToMap(t *testing.T) {
	expectedKey := "metrics"
	expectedValue := map[string]interface{}{
		"aggregation_dimensions": [][]string{{"InstanceId"}},
		"append_dimensions":      map[string]interface{}{"ImageId": "${aws:ImageId}", "InstanceId": "${aws:InstanceId}", "InstanceType": "${aws:InstanceType}", "AutoScalingGroupName": "${aws:AutoScalingGroupName}"},
		"metrics_collected": map[string]interface{}{
			"diskio":   map[string]interface{}{"resources": []string{"*"}, "measurement": []string{"io_time", "write_bytes", "read_bytes", "writes", "reads"}},
			"mem":      map[string]interface{}{"measurement": []string{"mem_used_percent"}},
			"net":      map[string]interface{}{"resources": []string{"*"}, "measurement": []string{"bytes_sent", "bytes_recv", "packets_sent", "packets_recv"}},
			"netstat":  map[string]interface{}{"measurement": []string{"tcp_established", "tcp_time_wait"}},
			"swap":     map[string]interface{}{"measurement": []string{"swap_used_percent"}},
			"cpu":      map[string]interface{}{"resources": []string{"*"}, "totalcpu": true, "measurement": []string{"cpu_usage_idle", "cpu_usage_iowait", "cpu_usage_steal", "cpu_usage_guest", "cpu_usage_user", "cpu_usage_system"}},
			"disk":     map[string]interface{}{"resources": []string{"*"}, "measurement": []string{"used_percent", "inodes_free"}},
			"statsd":   map[string]interface{}{"service_address": ":8125", "metrics_collection_interval": 10, "metrics_aggregation_interval": 60},
			"collectd": map[string]interface{}{"metrics_aggregation_interval": 60},
		},
	}
	conf := new(Metrics)
	ctx := &runtime.Context{
		OsParameter:             util.OsTypeLinux,
		WantEC2TagDimensions:    true,
		WantAggregateDimensions: true,
		IsOnPrem:                true,
		WantPerInstanceMetrics:  true,
	}
	conf.CollectAllMetrics(ctx)
	key, value := conf.ToMap(ctx)
	assert.Equal(t, expectedKey, key)
	assert.Equal(t, expectedValue, value)

	conf = new(Metrics)
	ctx = &runtime.Context{
		OsParameter:             util.OsTypeDarwin,
		WantEC2TagDimensions:    true,
		WantAggregateDimensions: true,
		IsOnPrem:                true,
		WantPerInstanceMetrics:  true,
	}
	conf.CollectAllMetrics(ctx)
	key, value = conf.ToMap(ctx)
	assert.Equal(t, expectedKey, key)
	assert.Equal(t, expectedValue, value)

	expectedValue = map[string]interface{}{
		"aggregation_dimensions": [][]string{{"InstanceId"}},
		"append_dimensions":      map[string]interface{}{"InstanceId": "${aws:InstanceId}", "InstanceType": "${aws:InstanceType}", "AutoScalingGroupName": "${aws:AutoScalingGroupName}", "ImageId": "${aws:ImageId}"},
		"metrics_collected": map[string]interface{}{
			"LogicalDisk":       map[string]interface{}{"resources": []string{"*"}, "measurement": []string{"% Free Space"}},
			"Memory":            map[string]interface{}{"measurement": []string{"% Committed Bytes In Use"}},
			"Network Interface": map[string]interface{}{"resources": []string{"*"}, "measurement": []string{"Bytes Sent/sec", "Bytes Received/sec", "Packets Sent/sec", "Packets Received/sec"}},
			"Processor":         map[string]interface{}{"resources": []string{"*"}, "measurement": []string{"% Processor Time", "% User Time", "% Idle Time", "% Interrupt Time"}},
			"PhysicalDisk":      map[string]interface{}{"resources": []string{"*"}, "measurement": []string{"% Disk Time", "Disk Write Bytes/sec", "Disk Read Bytes/sec", "Disk Writes/sec", "Disk Reads/sec"}},
			"TCPv4":             map[string]interface{}{"measurement": []string{"Connections Established"}},
			"TCPv6":             map[string]interface{}{"measurement": []string{"Connections Established"}},
			"Paging File":       map[string]interface{}{"resources": []string{"*"}, "measurement": []string{"% Usage"}},
			"statsd":            map[string]interface{}{"service_address": ":8125", "metrics_collection_interval": 10, "metrics_aggregation_interval": 60},
		},
	}
	ctx.OsParameter = util.OsTypeWindows
	conf.CollectAllMetrics(ctx)
	key, value = conf.ToMap(ctx)
	assert.Equal(t, expectedKey, key)
	assert.Equal(t, expectedValue, value)
}