From d4dc212dd3e30e275aed443b5c1574e8cf476aa3 Mon Sep 17 00:00:00 2001
From: Winni Neessen <wn@neessen.net>
Date: Wed, 23 Oct 2024 16:14:05 +0200
Subject: [PATCH] Refactor client tests and add SetSMTPAuthCustom test

Removed outdated tests for TLSConfig, Username, Password, and SMTPAuth that were commented out. Added a new test for the SetSMTPAuthCustom method, specifically for PLAIN authentication, ensuring proper behavior and type checks.
---
 client_test.go | 128 ++++++++++---------------------------------------
 1 file changed, 24 insertions(+), 104 deletions(-)

diff --git a/client_test.go b/client_test.go
index adeeec5..0c762a8 100644
--- a/client_test.go
+++ b/client_test.go
@@ -1379,113 +1379,33 @@ func TestClient_SetSMTPAuth(t *testing.T) {
 	})
 }
 
+func TestClient_SetSMTPAuthCustom(t *testing.T) {
+	t.Run("SetSMTPAuthCustom with PLAIN auth", func(t *testing.T) {
+		client, err := NewClient(DefaultHost)
+		if err != nil {
+			t.Fatalf("failed to create new client: %s", err)
+		}
+		client.SetSMTPAuthCustom(
+			smtp.PlainAuth("", "", "", "", false),
+		)
+		if client.smtpAuth == nil {
+			t.Errorf("failed to set custom SMTP auth, expected auth method but got nil")
+		}
+		if client.smtpAuthType != SMTPAuthCustom {
+			t.Errorf("failed to set custom SMTP auth, want auth type: %s, got: %s", SMTPAuthCustom,
+				client.smtpAuthType)
+		}
+		authType := reflect.TypeOf(client.smtpAuth).String()
+		if authType != "*smtp.plainAuth" {
+			t.Errorf("failed to set custom SMTP auth, expected auth method type: %s, got: %s",
+				"*smtp.plainAuth", authType)
+		}
+	})
+}
+
 /*
 
 
-// TestSetTLSConfig tests the SetTLSConfig() method for the Client object
-func TestSetTLSConfig(t *testing.T) {
-	tests := []struct {
-		name  string
-		value *tls.Config
-		sf    bool
-	}{
-		{"default config", &tls.Config{}, false},
-		{"nil config", nil, true},
-	}
-	for _, tt := range tests {
-		t.Run(tt.name, func(t *testing.T) {
-			c, err := NewClient(DefaultHost)
-			if err != nil {
-				t.Errorf("failed to create new client: %s", err)
-				return
-			}
-			if err := c.SetTLSConfig(tt.value); err != nil && !tt.sf {
-				t.Errorf("failed to set TLSConfig: %s", err)
-				return
-			}
-		})
-	}
-}
-
-// TestSetUsername tests the SetUsername method for the Client object
-func TestSetUsername(t *testing.T) {
-	tests := []struct {
-		name  string
-		value string
-		want  string
-		sf    bool
-	}{
-		{"normal username", "testuser", "testuser", false},
-		{"empty username", "", "", false},
-	}
-	for _, tt := range tests {
-		t.Run(tt.name, func(t *testing.T) {
-			c, err := NewClient(DefaultHost)
-			if err != nil {
-				t.Errorf("failed to create new client: %s", err)
-				return
-			}
-			c.SetUsername(tt.value)
-			if c.user != tt.want {
-				t.Errorf("failed to set username. Expected %s, got: %s", tt.want, c.user)
-			}
-		})
-	}
-}
-
-// TestSetPassword tests the SetPassword method for the Client object
-func TestSetPassword(t *testing.T) {
-	tests := []struct {
-		name  string
-		value string
-		want  string
-		sf    bool
-	}{
-		{"normal password", "testpass", "testpass", false},
-		{"empty password", "", "", false},
-	}
-	for _, tt := range tests {
-		t.Run(tt.name, func(t *testing.T) {
-			c, err := NewClient(DefaultHost)
-			if err != nil {
-				t.Errorf("failed to create new client: %s", err)
-				return
-			}
-			c.SetPassword(tt.value)
-			if c.pass != tt.want {
-				t.Errorf("failed to set password. Expected %s, got: %s", tt.want, c.pass)
-			}
-		})
-	}
-}
-
-// TestSetSMTPAuth tests the SetSMTPAuth method for the Client object
-func TestSetSMTPAuth(t *testing.T) {
-	tests := []struct {
-		name  string
-		value SMTPAuthType
-		want  string
-		sf    bool
-	}{
-		{"SMTPAuth: LOGIN", SMTPAuthLogin, "LOGIN", false},
-		{"SMTPAuth: PLAIN", SMTPAuthPlain, "PLAIN", false},
-		{"SMTPAuth: CRAM-MD5", SMTPAuthCramMD5, "CRAM-MD5", false},
-	}
-	for _, tt := range tests {
-		t.Run(tt.name, func(t *testing.T) {
-			c, err := NewClient(DefaultHost)
-			if err != nil {
-				t.Errorf("failed to create new client: %s", err)
-				return
-			}
-			c.SetSMTPAuth(tt.value)
-			if string(c.smtpAuthType) != tt.want {
-				t.Errorf("failed to set SMTP auth type. Expected %s, got: %s", tt.want, string(c.smtpAuthType))
-			}
-		})
-	}
-}
-
 // TestWithDSN tests the WithDSN method for the Client object
 func TestWithDSN(t *testing.T) {
 	c, err := NewClient(DefaultHost, WithDSN())