// -
//   ========================LICENSE_START=================================
//   O-RAN-SC
//   %%
//   Copyright (C) 2021: Nordix Foundation
//   %%
//   Licensed under the Apache License, Version 2.0 (the "License");
//   you may not use this file except in compliance with the License.
//   You may obtain a copy of the License at
//
//        http://www.apache.org/licenses/LICENSE-2.0
//
//   Unless required by applicable law or agreed to in writing, software
//   distributed under the License is distributed on an "AS IS" BASIS,
//   WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
//   See the License for the specific language governing permissions and
//   limitations under the License.
//   ========================LICENSE_END===================================
//

package config

import (
	"bytes"
	"os"
	"testing"

	log "github.com/sirupsen/logrus"
	"github.com/stretchr/testify/require"
)

func TestNewEnvVarsSetConfigContainSetValues(t *testing.T) {
	assertions := require.New(t)
	os.Setenv("MR_HOST", "consumerHost")
	os.Setenv("MR_PORT", "8095")
	os.Setenv("SDNR_ADDR", "http://localhost:3904")
	os.Setenv("SDNR_USER", "admin")
	os.Setenv("SDNR_PASSWORD", "Kp8bJ4SXszM0WXlhak3eHlcse2gAw84vaoGGmJvUy2U")
	os.Setenv("Polltime", "30")
	os.Setenv("LOG_LEVEL", "Debug")
	os.Setenv("NODE_ID", "o-du-1122")
	t.Cleanup(func() {
		os.Clearenv()
	})
	wantConfig := Config{
		MRHost:      "consumerHost",
		MRPort:      "8095",
		SDNRAddress: "http://localhost:3904",
		SDNRUser:    "admin",
		SDNPassword: "Kp8bJ4SXszM0WXlhak3eHlcse2gAw84vaoGGmJvUy2U",
		Polltime:    30,
		LogLevel:    log.DebugLevel,
		NodeId:      "o-du-1122",
	}

	got := New()
	assertions.Equal(&wantConfig, got)
}

func TestNewFaultyIntValueSetConfigContainDefaultValueAndWarnInLog(t *testing.T) {
	assertions := require.New(t)
	var buf bytes.Buffer
	log.SetOutput(&buf)

	os.Setenv("Polltime", "wrong")
	t.Cleanup(func() {
		log.SetOutput(os.Stderr)
		os.Clearenv()
	})
	wantConfig := Config{
		MRHost:      "",
		MRPort:      "",
		SDNRAddress: "http://localhost:3904",
		SDNRUser:    "admin",
		SDNPassword: "Kp8bJ4SXszM0WXlhak3eHlcse2gAw84vaoGGmJvUy2U",
		Polltime:    30,
		LogLevel:    log.InfoLevel,
		NodeId:      "",
	}

	got := New()
	assertions.Equal(&wantConfig, got)

	logString := buf.String()
	assertions.Contains(logString, "Invalid int value: wrong for variable: Polltime. Default value: 30 will be used")
}

func TestNewEnvFaultyLogLevelConfigContainDefaultValues(t *testing.T) {
	assertions := require.New(t)
	var buf bytes.Buffer
	log.SetOutput(&buf)

	os.Setenv("LOG_LEVEL", "wrong")
	t.Cleanup(func() {
		log.SetOutput(os.Stderr)
		os.Clearenv()
	})
	wantConfig := Config{
		MRHost:      "",
		MRPort:      "",
		SDNRAddress: "http://localhost:3904",
		SDNRUser:    "admin",
		SDNPassword: "Kp8bJ4SXszM0WXlhak3eHlcse2gAw84vaoGGmJvUy2U",
		Polltime:    30,
		LogLevel:    log.InfoLevel,
		NodeId:      "",
	}
	got := New()
	assertions.Equal(&wantConfig, got)
	logString := buf.String()
	assertions.Contains(logString, "Invalid log level: wrong. Log level will be Info!")
}