From ab6bbc35877a9c28985de257281ebccde2d2d324 Mon Sep 17 00:00:00 2001
From: Jonas Ha <jonas-ha@outlook.com>
Date: Mon, 16 Dec 2024 20:50:24 -0500
Subject: [PATCH] updated tests to test output.

---
 .idea/dictionaries/jonasha.xml               |   5 +
 .idea/watcherTasks.xml                       |   1 +
 event_test.go                                | 201 +++++++++++++++----
 go.mod                                       |   1 +
 go.sum                                       |   2 +
 print_response.go                            |  12 +-
 parse_json_test.go => print_response_test.go |   1 -
 7 files changed, 180 insertions(+), 43 deletions(-)
 rename parse_json_test.go => print_response_test.go (99%)

diff --git a/.idea/dictionaries/jonasha.xml b/.idea/dictionaries/jonasha.xml
index 21c9a8e..58e06b9 100644
--- a/.idea/dictionaries/jonasha.xml
+++ b/.idea/dictionaries/jonasha.xml
@@ -1,7 +1,12 @@
 <component name="ProjectDictionaryState">
   <dictionary name="jonasha">
     <words>
+      <w>cyclop</w>
+      <w>forbidigo</w>
+      <w>funlen</w>
       <w>golangci</w>
+      <w>nolintlint</w>
+      <w>paralleltest</w>
     </words>
   </dictionary>
 </component>
\ No newline at end of file
diff --git a/.idea/watcherTasks.xml b/.idea/watcherTasks.xml
index 1b7206a..ad322a8 100644
--- a/.idea/watcherTasks.xml
+++ b/.idea/watcherTasks.xml
@@ -2,6 +2,7 @@
 <project version="4">
   <component name="ProjectTasksOptions">
     <enabled-global>
+      <option value="golines" />
       <option value="gofumpt" />
       <option value="gci" />
       <option value="tagalign" />
diff --git a/event_test.go b/event_test.go
index 8c43273..be5884b 100644
--- a/event_test.go
+++ b/event_test.go
@@ -1,13 +1,17 @@
 package main
 
 import (
+	"bytes"
 	"context"
 	"errors"
+	"fmt"
 	"log/slog"
-	"os"
+	"strings"
 	"testing"
 
 	"github.com/aws/aws-lambda-go/lambda/messages"
+	"github.com/lithammer/dedent"
+	"github.com/lmittmann/tint"
 	"github.com/stretchr/testify/assert"
 	"github.com/stretchr/testify/mock"
 	"github.com/stretchr/testify/require"
@@ -23,15 +27,14 @@ func (m *MockLambdaCaller) Invoke(data []byte) (messages.InvokeResponse, error)
 	return args.Get(0).(messages.InvokeResponse), args.Error(1) //nolint:wrapcheck,forcetypeassert
 }
 
-func TestRunLambdaEvent(t *testing.T) {
-	t.Parallel()
-
+func TestRunLambdaEvent(t *testing.T) { //nolint:funlen
 	tests := map[string]struct {
-		event       string
-		parseJSON   bool
-		invokeResp  messages.InvokeResponse
-		invokeErr   error
-		expectedErr string
+		event          string
+		parseJSON      bool
+		invokeResp     messages.InvokeResponse
+		invokeErr      error
+		expectedOutput string
+		expectedErr    error
 	}{
 		"successful invocation without JSON parsing": {
 			event:     `{"key": "value"}`,
@@ -39,8 +42,20 @@ func TestRunLambdaEvent(t *testing.T) {
 			invokeResp: messages.InvokeResponse{
 				Payload: []byte(`{"response": "success"}`),
 			},
-			invokeErr:   nil,
-			expectedErr: "",
+			invokeErr: nil,
+			expectedOutput: `
+				---------------------------------------------------------------------------
+				0 INF Starting local Lambda invocation with Event
+				0 DBG Invoking lambda event
+				0 DBG Handling lambda event response
+				0 INF Lambda returned JSON payload:
+				{
+				    "response": "success"
+				}
+				0 INF Lambda invocation complete, Exiting...
+				---------------------------------------------------------------------------
+			`,
+			expectedErr: nil,
 		},
 		"successful invocation with JSON parsing": {
 			event:     `{"key": "value"}`,
@@ -48,48 +63,158 @@ func TestRunLambdaEvent(t *testing.T) {
 			invokeResp: messages.InvokeResponse{
 				Payload: []byte(`{"response": "{\"innerKey\": \"innerValue\"}"}`),
 			},
-			invokeErr:   nil,
-			expectedErr: "",
+			invokeErr: nil,
+			expectedOutput: `
+				---------------------------------------------------------------------------
+				0 INF Starting local Lambda invocation with Event
+				0 DBG Invoking lambda event
+				0 DBG Handling lambda event response
+				0 INF Lambda returned JSON payload:
+				{
+				    "response": {
+				        "innerKey": "innerValue"
+				    }
+				}
+				0 INF Lambda invocation complete, Exiting...
+				---------------------------------------------------------------------------
+			`,
+			expectedErr: nil,
 		},
 		"invoke error": {
-			event:       `{"key": "value"}`,
-			parseJSON:   false,
-			invokeResp:  messages.InvokeResponse{},
-			invokeErr:   errors.New("invoke error"),
-			expectedErr: "invoke failed: invoke error",
+			event:      `{"key": "value"}`,
+			parseJSON:  false,
+			invokeResp: messages.InvokeResponse{},
+			invokeErr:  errors.New("invoke error"),
+			expectedOutput: `
+				---------------------------------------------------------------------------
+				0 INF Starting local Lambda invocation with Event
+				0 DBG Invoking lambda event
+			`,
+			expectedErr: fmt.Errorf("[in lambdalocal.RunLambdaEvent] invoke failed: %w", errors.New("invoke error")),
 		},
-		"unmarshal error": {
+		"lambda returned error - valid JSON body": {
 			event:     `{"key": "value"}`,
 			parseJSON: false,
 			invokeResp: messages.InvokeResponse{
-				Payload: []byte(`invalid json`),
+				Payload: []byte(`{"error": "test error"}`),
+				Error: &messages.InvokeResponse_Error{
+					Message: "test error",
+					Type:    "",
+					StackTrace: []*messages.InvokeResponse_Error_StackFrame{
+						{
+							Path:  "path1",
+							Line:  1,
+							Label: "part_1",
+						},
+						{
+							Path:  "path2",
+							Line:  2,
+							Label: "part_2",
+						},
+					},
+					ShouldExit: false,
+				},
 			},
-			invokeErr:   nil,
-			expectedErr: "unmarshal response failed: invalid character 'i' looking for beginning of value",
+			invokeErr: nil,
+			expectedOutput: `
+				---------------------------------------------------------------------------
+				0 INF Starting local Lambda invocation with Event
+				0 DBG Invoking lambda event
+				0 DBG Handling lambda event response
+				0 ERR Lambda returned error:
+				Returned error: test error
+				
+				Stack trace:
+				path1:1 - part_1
+				path2:2 - part_2
+				
+				0 INF Lambda returned JSON payload:
+				{
+				    "error": "test error"
+				}
+				0 INF Lambda invocation complete, Exiting...
+				---------------------------------------------------------------------------
+			`,
+			expectedErr: nil,
+		},
+		"lambda returned error - invalid JSON body": {
+			event:     `{"key": "value"}`,
+			parseJSON: false,
+			invokeResp: messages.InvokeResponse{
+				Payload: []byte(`invalid JSON`),
+				Error: &messages.InvokeResponse_Error{
+					Message: "test error",
+					Type:    "",
+					StackTrace: []*messages.InvokeResponse_Error_StackFrame{
+						{
+							Path:  "path1",
+							Line:  1,
+							Label: "part_1",
+						},
+						{
+							Path:  "path2",
+							Line:  2,
+							Label: "part_2",
+						},
+					},
+					ShouldExit: false,
+				},
+			},
+			invokeErr: nil,
+			expectedOutput: `
+				---------------------------------------------------------------------------
+				0 INF Starting local Lambda invocation with Event
+				0 DBG Invoking lambda event
+				0 DBG Handling lambda event response
+				0 ERR Lambda returned error:
+				Returned error: test error
+				
+				Stack trace:
+				path1:1 - part_1
+				path2:2 - part_2
+				
+				0 INF Lambda returned non-JSON payload:
+				invalid JSON
+				0 INF Lambda invocation complete, Exiting...
+				---------------------------------------------------------------------------
+			`,
+			expectedErr: nil,
 		},
 	}
 
 	for name, tc := range tests {
-		t.Run(
-			name, func(t *testing.T) {
-				t.Parallel()
+		t.Run(name, func(t *testing.T) {
+			mockLambdaRPC := new(MockLambdaCaller)
 
-				mockLambdaRPC := new(MockLambdaCaller)
-				logger := slog.Default()
+			var buf bytes.Buffer
 
-				mockLambdaRPC.On("Invoke", []byte(tc.event)).Return(tc.invokeResp, tc.invokeErr)
+			logger := slog.New(
+				tint.NewHandler(
+					&buf, &tint.Options{
+						Level:      slog.LevelDebug,
+						TimeFormat: "0",
+						NoColor:    true,
+					},
+				),
+			)
 
-				err := RunLambdaEvent(context.Background(), os.Stdout, mockLambdaRPC, tc.event, tc.parseJSON, logger)
+			mockLambdaRPC.On("Invoke", []byte(tc.event)).Return(tc.invokeResp, tc.invokeErr)
 
-				if tc.expectedErr == "" {
-					require.NoError(t, err)
-				} else {
-					require.Error(t, err)
-					assert.Contains(t, err.Error(), tc.expectedErr)
-				}
+			err := RunLambdaEvent(context.Background(), &buf, mockLambdaRPC, tc.event, tc.parseJSON, logger)
 
-				mockLambdaRPC.AssertExpectations(t)
-			},
-		)
+			if tc.expectedErr == nil {
+				require.NoError(t, err)
+			} else {
+				assert.Equal(t, tc.expectedErr, err)
+			}
+
+			output := strings.TrimSpace(buf.String())
+
+			// fmt.Println(output)
+
+			assert.Equal(t, strings.TrimSpace(dedent.Dedent(tc.expectedOutput)), output)
+
+			mockLambdaRPC.AssertExpectations(t)
+		})
 	}
 }
diff --git a/go.mod b/go.mod
index b859fa1..5613ee5 100644
--- a/go.mod
+++ b/go.mod
@@ -14,6 +14,7 @@ require (
 
 require (
 	github.com/davecgh/go-spew v1.1.1 // indirect
+	github.com/lithammer/dedent v1.1.0 // indirect
 	github.com/pmezard/go-difflib v1.0.0 // indirect
 	github.com/stretchr/objx v0.5.0 // indirect
 	github.com/xrash/smetrics v0.0.0-20201216005158-039620a65673 // indirect
diff --git a/go.sum b/go.sum
index a78e9fc..70d0e8d 100644
--- a/go.sum
+++ b/go.sum
@@ -5,6 +5,8 @@ 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/google/uuid v1.6.0 h1:NIvaJDMOsjHA8n1jAhLSgzrAzy1Hgr+hNrb57e+94F0=
 github.com/google/uuid v1.6.0/go.mod h1:TIyPZe4MgqvfeYDBFedMoGGpEw/LqOeaOT+nhxU+yHo=
+github.com/lithammer/dedent v1.1.0 h1:VNzHMVCBNG1j0fh3OrsFRkVUwStdDArbgBWoPAffktY=
+github.com/lithammer/dedent v1.1.0/go.mod h1:jrXYCQtgg0nJiN+StA2KgR7w6CiQNv9Fd/Z9BP0jIOc=
 github.com/lmittmann/tint v1.0.5 h1:NQclAutOfYsqs2F1Lenue6OoWCajs5wJcP3DfWVpePw=
 github.com/lmittmann/tint v1.0.5/go.mod h1:HIS3gSy7qNwGCj+5oRjAutErFBl4BzdQP6cJZ0NfMwE=
 github.com/pmezard/go-difflib v1.0.0 h1:4DBwDE0NGyQoBHbLQYPwSUPoCMWR5BEzIk/f1lZbAQM=
diff --git a/print_response.go b/print_response.go
index a10d396..0a47bd2 100644
--- a/print_response.go
+++ b/print_response.go
@@ -4,7 +4,7 @@ import (
 	"encoding/json"
 	"fmt"
 	"log/slog"
-	"os"
+	"strings"
 
 	"github.com/aws/aws-lambda-go/lambda/messages"
 )
@@ -17,12 +17,16 @@ func printResponse(
 	logger.Debug("Handling lambda event response")
 
 	if invokeResponse.Error != nil {
-		logger.Error("Lambda returned error")
-		_, _ = fmt.Fprintln(os.Stdout, invokeResponse.Error.Message)
+		var builder strings.Builder
+
+		builder.WriteString(fmt.Sprintf("Returned error: %s\n", invokeResponse.Error.Message))
+		builder.WriteString("\nStack trace:\n")
 
 		for _, detail := range invokeResponse.Error.StackTrace {
-			_, _ = fmt.Fprintf(os.Stdout, "\t%s:%d - %s\n", detail.Path, detail.Line, detail.Label)
+			builder.WriteString(fmt.Sprintf("%s:%d - %s\n", detail.Path, detail.Line, detail.Label))
 		}
+
+		logger.Error("Lambda returned error:\n" + builder.String())
 	}
 
 	if invokeResponse.Payload == nil {
diff --git a/parse_json_test.go b/print_response_test.go
similarity index 99%
rename from parse_json_test.go
rename to print_response_test.go
index 9c7a65c..778ab7f 100644
--- a/parse_json_test.go
+++ b/print_response_test.go
@@ -1,4 +1,3 @@
-//nolint:cyclop
 package main
 
 import (