From b7f48819727dd54ca6547fd93d2b668cf075de6d Mon Sep 17 00:00:00 2001
From: Panagiotis Siatras <azazeal@users.noreply.github.com>
Date: Thu, 17 Nov 2022 16:00:01 +0200
Subject: [PATCH] merged log tests

---
 api/log/log_test.go | 112 ++++++++++++++++----------------------------
 1 file changed, 41 insertions(+), 71 deletions(-)

diff --git a/api/log/log_test.go b/api/log/log_test.go
index d466bfce..28a6badf 100644
--- a/api/log/log_test.go
+++ b/api/log/log_test.go
@@ -1,102 +1,72 @@
 package log
 
 import (
-	"errors"
-	"net/http"
 	"net/http/httptest"
+	"strconv"
 	"testing"
 	"unsafe"
 
 	pkgerrors "github.com/pkg/errors"
+	"github.com/stretchr/testify/assert"
 
 	"github.com/smallstep/certificates/logging"
 )
 
 func TestError(t *testing.T) {
-
-	t.Setenv("STEPDEBUG", "1") // force to end of `Error` function instead of early return
-	theError := errors.New("the error")
-
-	type args struct {
-		rw  http.ResponseWriter
-		err error
-	}
-	tests := []struct {
-		name       string
-		args       args
-		withFields bool
-		want       string
+	cases := []struct {
+		error
+		stepDebug        bool
+		expectStackTrace bool
 	}{
-		{"normalLogger", args{httptest.NewRecorder(), theError}, false, "the error"},
-		{"responseLogger", args{logging.NewResponseLogger(httptest.NewRecorder()), theError}, true, "the error"},
+		0: {nil, false, false},
+		1: {nil, true, false},
+		2: {assert.AnError, false, false},
+		3: {assert.AnError, true, false},
+		4: {new(stackTracedError), false, false},
+		5: {new(stackTracedError), true, true},
 	}
 
-	for _, tt := range tests {
-		t.Run(tt.name, func(t *testing.T) {
-			Error(tt.args.rw, tt.args.err)
-			if tt.withFields {
-				if rl, ok := tt.args.rw.(logging.ResponseLogger); ok {
-					fields := rl.Fields()
-					if fields["error"].(error).Error() != tt.want {
-						t.Errorf(`ResponseLogger["error"] = %s, wants %s`, fields["error"], tt.want)
-					}
-				} else {
-					t.Error("ResponseWriter does not implement logging.ResponseLogger")
-				}
+	for caseIndex := range cases {
+		kase := cases[caseIndex]
+
+		t.Run(strconv.Itoa(caseIndex), func(t *testing.T) {
+			if kase.stepDebug {
+				t.Setenv("STEPDEBUG", "1")
+			} else {
+				t.Setenv("STEPDEBUG", "0")
+			}
+
+			rw := logging.NewResponseLogger(httptest.NewRecorder())
+			Error(rw, kase.error)
+
+			fields := rw.Fields()
+
+			// expect the error field to be set and to be the same error that was fed to Error
+			if kase.error == nil {
+				assert.Nil(t, fields["error"])
+			} else {
+				assert.Same(t, kase.error, fields["error"])
+			}
+
+			if _, hasStackTrace := fields["stack-trace"]; kase.expectStackTrace && !hasStackTrace {
+				t.Error("stack-trace was not set")
+			} else if !kase.expectStackTrace && hasStackTrace {
+				t.Error("stack-trace was set")
 			}
 		})
 	}
 }
 
-type mockStackTracedError struct{}
+type stackTracedError struct{}
 
-func (t *mockStackTracedError) Error() string {
+func (stackTracedError) Error() string {
 	return "a stacktraced error"
 }
 
-func (t *mockStackTracedError) StackTrace() pkgerrors.StackTrace {
+func (stackTracedError) StackTrace() pkgerrors.StackTrace {
 	f := struct{}{}
 	return pkgerrors.StackTrace{ // fake stacktrace
 		pkgerrors.Frame(unsafe.Pointer(&f)),
 		pkgerrors.Frame(unsafe.Pointer(&f)),
 	}
 }
-
-func TestError_StackTracedError(t *testing.T) {
-
-	t.Setenv("STEPDEBUG", "1")
-	aStackTracedError := mockStackTracedError{}
-
-	type args struct {
-		rw  http.ResponseWriter
-		err error
-	}
-	tests := []struct {
-		name       string
-		args       args
-		withFields bool
-		want       string
-	}{
-		{"responseLoggerWithStackTracedError", args{logging.NewResponseLogger(httptest.NewRecorder()), &aStackTracedError}, true, "a stacktraced error"},
-	}
-
-	for _, tt := range tests {
-		t.Run(tt.name, func(t *testing.T) {
-			Error(tt.args.rw, tt.args.err)
-			if tt.withFields {
-				if rl, ok := tt.args.rw.(logging.ResponseLogger); ok {
-					fields := rl.Fields()
-					if fields["error"].(error).Error() != tt.want {
-						t.Errorf(`ResponseLogger["error"] = %s, wants %s`, fields["error"], tt.want)
-					}
-					// `stack-trace` expected to be set; not interested in actual output
-					if _, ok := fields["stack-trace"]; !ok {
-						t.Errorf(`ResponseLogger["stack-trace"] not set`)
-					}
-				} else {
-					t.Error("ResponseWriter does not implement logging.ResponseLogger")
-				}
-			}
-		})
-	}
-}