Log package test file

log_test.go

package log_test

import (
	"bytes"
	"encoding/json"
	"nomad-datadog-exporter/internal/log"
	"os"
	"testing"

	"github.com/google/go-cmp/cmp"
)

type format struct {
	AwsRegion    string `json:"aws_region,omitempty"`
	FunctionName string `json:"function_name,omitempty"`
	Level        string `json:"level,omitempty"`
	Msg          string `json:"msg,omitempty"`
}

func TestNewLoggerDefaults(t *testing.T) {
	t.Parallel()

	os.Setenv("AWS_REGION", "us-east-1")
	os.Setenv("AWS_LAMBDA_FUNCTION_NAME", "nomad-datadog-exporter")

	want := format{
		AwsRegion:    "us-east-1",
		FunctionName: "nomad-datadog-exporter",
		Level:        "info",
		Msg:          "Test!!!",
	}

	fakeStdErr := &bytes.Buffer{}

	logger, err := log.NewLogger(
		log.WithOutput(fakeStdErr),
	)
	if err != nil {
		t.Fatal(err)
	}
	logger.Info("Test!!!")

	var got format
	if err := json.Unmarshal(fakeStdErr.Bytes(), &got); err != nil {
		t.Fatal(err)
	}

	if !cmp.Equal(want, got) {
		cmp.Diff(want, got)
	}
}