From 7e77c7dfefbf75be53f4b9048c6f16360c736821 Mon Sep 17 00:00:00 2001
From: Jean-Baptiste Le Duigou <jb.leduigou@gmail.com>
Date: Mon, 18 Mar 2024 20:18:28 +0000
Subject: [PATCH] =?UTF-8?q?=E2=9C=85=20adding=20unit=20tests?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 go.mod          | 11 ++++++++++-
 go.sum          | 10 ++++++++++
 handler_test.go | 51 +++++++++++++++++++++++++++++++++++++++++++++++++
 3 files changed, 71 insertions(+), 1 deletion(-)
 create mode 100644 handler_test.go

diff --git a/go.mod b/go.mod
index 6525d09..ba0bd24 100644
--- a/go.mod
+++ b/go.mod
@@ -2,4 +2,13 @@ module github.com/jbleduigou/slog-aws-lambda
 
 go 1.22.1
 
-require github.com/aws/aws-lambda-go v1.46.0
+require (
+	github.com/aws/aws-lambda-go v1.46.0
+	github.com/stretchr/testify v1.9.0
+)
+
+require (
+	github.com/davecgh/go-spew v1.1.1 // indirect
+	github.com/pmezard/go-difflib v1.0.0 // indirect
+	gopkg.in/yaml.v3 v3.0.1 // indirect
+)
diff --git a/go.sum b/go.sum
index cc5d6d6..0025dd5 100644
--- a/go.sum
+++ b/go.sum
@@ -1,2 +1,12 @@
 github.com/aws/aws-lambda-go v1.46.0 h1:UWVnvh2h2gecOlFhHQfIPQcD8pL/f7pVCutmFl+oXU8=
 github.com/aws/aws-lambda-go v1.46.0/go.mod h1:dpMpZgvWx5vuQJfBt0zqBha60q7Dd7RfgJv23DymV8A=
+github.com/davecgh/go-spew v1.1.1 h1:vj9j/u1bqnvCEfJOwUhtlOARqs3+rkHYY13jYWTU97c=
+github.com/davecgh/go-spew v1.1.1/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
+github.com/pmezard/go-difflib v1.0.0 h1:4DBwDE0NGyQoBHbLQYPwSUPoCMWR5BEzIk/f1lZbAQM=
+github.com/pmezard/go-difflib v1.0.0/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZNVY4sRDYZ/4=
+github.com/stretchr/testify v1.9.0 h1:HtqpIVDClZ4nwg75+f6Lvsy/wHu+3BoSGCbBAcpTsTg=
+github.com/stretchr/testify v1.9.0/go.mod h1:r2ic/lqez/lEtzL7wO/rwa5dbSLXVDPFyf8C91i36aY=
+gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405 h1:yhCVgyC4o1eVCa2tZl7eS0r+SDo693bJlVdllGtEeKM=
+gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0=
+gopkg.in/yaml.v3 v3.0.1 h1:fxVm/GzAzEWqLHuvctI91KS9hhNmmWOoWu0XTYJS7CA=
+gopkg.in/yaml.v3 v3.0.1/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM=
diff --git a/handler_test.go b/handler_test.go
new file mode 100644
index 0000000..3179be8
--- /dev/null
+++ b/handler_test.go
@@ -0,0 +1,51 @@
+package slogawslambda
+
+import (
+	"log/slog"
+	"os"
+	"testing"
+
+	"github.com/stretchr/testify/assert"
+)
+
+func TestGetLogLevel(t *testing.T) {
+	tests := []struct {
+		name             string
+		envVariableValue string
+		want             slog.Leveler
+	}{
+		{
+			name:             "Should return Info given not set",
+			envVariableValue: "",
+			want:             slog.LevelInfo,
+		},
+		{
+			name:             "Should return Info given invalid value",
+			envVariableValue: "not-a-valid-value",
+			want:             slog.LevelInfo,
+		},
+		{
+			name:             "Should return Debug given lowercase debug",
+			envVariableValue: "debug",
+			want:             slog.LevelDebug,
+		},
+		{
+			name:             "Should return Warn given uppercase warn",
+			envVariableValue: "WARN",
+			want:             slog.LevelWarn,
+		},
+	}
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+			if tt.envVariableValue != "" {
+				os.Setenv("LOG_LEVEL", tt.envVariableValue)
+			}
+
+			got := getLogLevel()
+
+			os.Unsetenv("LOG_LEVEL")
+
+			assert.Equal(t, got, tt.want)
+		})
+	}
+}