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

package logs

import (
	"encoding/json"
	"os"
	"testing"

	"github.com/stretchr/testify/assert"

	"github.com/aws/amazon-cloudwatch-agent/translator/config"
	"github.com/aws/amazon-cloudwatch-agent/translator/context"
	"github.com/aws/amazon-cloudwatch-agent/translator/translate/agent"
	"github.com/aws/amazon-cloudwatch-agent/translator/util/ecsutil"
)

func TestLogs(t *testing.T) {
	l := new(Logs)
	agent.Global_Config.Region = "us-east-1"

	var input interface{}
	err := json.Unmarshal([]byte(`{"logs":{"log_stream_name":"LOG_STREAM_NAME"}}`), &input)
	if err != nil {
		assert.Fail(t, err.Error())
	}

	_, actual := l.ApplyRule(input)
	expected := map[string]interface{}{
		"outputs": map[string]interface{}{
			"cloudwatchlogs": []interface{}{
				map[string]interface{}{
					"region":               "us-east-1",
					"log_stream_name":      "LOG_STREAM_NAME",
					"force_flush_interval": "5s",
				},
			},
		},
	}
	assert.Equal(t, expected, actual, "Expected to be equal")
}

func TestLogs_LogStreamName(t *testing.T) {
	l := new(Logs)
	agent.Global_Config.Region = "us-east-1"

	var input interface{}
	err := json.Unmarshal([]byte(`{"logs":{}}`), &input)
	if err != nil {
		assert.Fail(t, err.Error())
	}

	ctx := context.CurrentContext()
	ctx.SetMode(config.ModeOnPrem)

	hostname, _ := os.Hostname()
	_, actual := l.ApplyRule(input)
	expected := map[string]interface{}{
		"outputs": map[string]interface{}{
			"cloudwatchlogs": []interface{}{
				map[string]interface{}{
					"region":               "us-east-1",
					"log_stream_name":      hostname,
					"force_flush_interval": "5s",
				},
			},
		},
	}

	assert.Equal(t, expected, actual, "Expected to be equal")

	context.ResetContext()

	// default log stream name from task arn
	ctx = context.CurrentContext()
	ctx.SetRunInContainer(true)
	ecsUtilInstance := ecsutil.GetECSUtilSingleton()
	ecsUtilInstance.Region = "us-east-1"
	ecsUtilInstance.Cluster = "cluster-name"
	ecsUtilInstance.TaskARN = "arn:aws:ecs:us-east-2:012345678910:task/cluster-name/9781c248-0edd-4cdb-9a93-f63cb662a5d3"

	_, actual = l.ApplyRule(input)
	expected = map[string]interface{}{
		"outputs": map[string]interface{}{
			"cloudwatchlogs": []interface{}{
				map[string]interface{}{
					"region":               "us-east-1",
					"log_stream_name":      "arn_aws_ecs_us-east-2_012345678910_task/cluster-name/9781c248-0edd-4cdb-9a93-f63cb662a5d3",
					"force_flush_interval": "5s",
				},
			},
		},
	}
	assert.Equal(t, expected, actual, "Expected to be equal")

	context.ResetContext()
	ecsUtilInstance.Region = ""

	// default log stream name from pod id env variable
	ctx = context.CurrentContext()
	ctx.SetRunInContainer(true)
	os.Setenv(config.POD_NAME, "demo-app-5ffc89b95c-jgnf6")

	_, actual = l.ApplyRule(input)
	expected = map[string]interface{}{
		"outputs": map[string]interface{}{
			"cloudwatchlogs": []interface{}{
				map[string]interface{}{
					"region":               "us-east-1",
					"log_stream_name":      "demo-app-5ffc89b95c-jgnf6",
					"force_flush_interval": "5s",
				},
			},
		},
	}
	assert.Equal(t, expected, actual, "Expected to be equal")

	os.Clearenv()
	context.ResetContext()
}

func TestLogs_ForceFlushInterval(t *testing.T) {
	l := new(Logs)
	agent.Global_Config.Region = "us-east-1"

	var input interface{}
	err := json.Unmarshal([]byte(`{"logs":{"force_flush_interval":10}}`), &input)
	if err != nil {
		assert.Fail(t, err.Error())
	}

	ctx := context.CurrentContext()
	ctx.SetMode(config.ModeOnPrem)

	hostname, _ := os.Hostname()
	_, actual := l.ApplyRule(input)
	expected := map[string]interface{}{
		"outputs": map[string]interface{}{
			"cloudwatchlogs": []interface{}{
				map[string]interface{}{
					"region":               "us-east-1",
					"log_stream_name":      hostname,
					"force_flush_interval": "10s",
				},
			},
		},
	}

	assert.Equal(t, expected, actual, "Expected to be equal")

	ctx.SetMode(config.ModeEC2) //reset back to default mode
}

func TestLogs_EndpointOverride(t *testing.T) {
	l := new(Logs)
	agent.Global_Config.Region = "us-east-1"

	var input interface{}
	err := json.Unmarshal([]byte(`{"logs":{"endpoint_override":"https://logs-fips.us-east-1.amazonaws.com"}}`), &input)
	if err != nil {
		assert.Fail(t, err.Error())
	}

	ctx := context.CurrentContext()
	ctx.SetMode(config.ModeOnPrem)

	hostname, _ := os.Hostname()
	_, actual := l.ApplyRule(input)
	expected := map[string]interface{}{
		"outputs": map[string]interface{}{
			"cloudwatchlogs": []interface{}{
				map[string]interface{}{
					"region":               "us-east-1",
					"endpoint_override":    "https://logs-fips.us-east-1.amazonaws.com",
					"log_stream_name":      hostname,
					"force_flush_interval": "5s",
				},
			},
		},
	}

	assert.Equal(t, expected, actual, "Expected to be equal")

	ctx.SetMode(config.ModeEC2) //reset back to default mode
}