From 5e3d14f842ab970c58f24971ad58f0881a22dff1 Mon Sep 17 00:00:00 2001
From: Winni Neessen <wn@neessen.net>
Date: Mon, 11 Nov 2024 17:49:07 +0100
Subject: [PATCH] Add tests for Client's SetLogger method

This commit introduces unit tests for the Client's SetLogger method. It verifies the correct logger type is set and ensures setting a nil logger does not override an existing logger.
---
 smtp/smtp_test.go | 24 ++++++++++++++++++++++++
 1 file changed, 24 insertions(+)

diff --git a/smtp/smtp_test.go b/smtp/smtp_test.go
index 93cbf84..5b47291 100644
--- a/smtp/smtp_test.go
+++ b/smtp/smtp_test.go
@@ -3169,6 +3169,30 @@ func TestClient_SetDebugLog(t *testing.T) {
 	})
 }
 
+func TestClient_SetLogger(t *testing.T) {
+	t.Run("set logger to Stdlog logger", func(t *testing.T) {
+		client := &Client{}
+		client.SetLogger(log.New(os.Stderr, log.LevelDebug))
+		if !strings.EqualFold(fmt.Sprintf("%T", client.logger), "*log.Stdlog") {
+			t.Errorf("expected logger to be of type *log.Stdlog, got: %T", client.logger)
+		}
+	})
+	t.Run("set logger to JSONlog logger", func(t *testing.T) {
+		client := &Client{}
+		client.SetLogger(log.NewJSON(os.Stderr, log.LevelDebug))
+		if !strings.EqualFold(fmt.Sprintf("%T", client.logger), "*log.JSONlog") {
+			t.Errorf("expected logger to be of type *log.JSONlog, got: %T", client.logger)
+		}
+	})
+	t.Run("nil logger should just return and not set/override", func(t *testing.T) {
+		client := &Client{logger: log.NewJSON(os.Stderr, log.LevelDebug)}
+		client.SetLogger(nil)
+		if !strings.EqualFold(fmt.Sprintf("%T", client.logger), "*log.JSONlog") {
+			t.Errorf("expected logger to be of type *log.JSONlog, got: %T", client.logger)
+		}
+	})
+}
+
 // faker is a struct embedding io.ReadWriter to simulate network connections for testing purposes.
 type faker struct {
 	io.ReadWriter