Added unit tests for SMTP and IMAP SetCustomAuthentication methods.

Author: danzuep

source/MailKitSimplified.Receiver/Services/ImapReceiver.cs

@@ -85,6 +85,12 @@ public ImapReceiver SetFolder(string mailFolderName)
             return this;
         }
 
+        public ImapReceiver SetCustomAuthentication(Func<IImapClient, Task> customAuthenticationMethod)
+        {
+            _customAuthenticationMethod = customAuthenticationMethod;
+            return this;
+        }
+
         public IMailFolderReader ReadFrom(string mailFolderName)
         {
             _receiverOptions.MailFolderName = mailFolderName;
@@ -103,19 +109,6 @@ public IMailFolderMonitor Monitor(string mailFolderName)
 
         public IMailFolderMonitor MonitorFolder => new MailFolderMonitor(this);
 
-        public IImapReceiver CustomAuthentication(Func<IImapClient, Task> customAuthenticationMethod)
-        {
-            _customAuthenticationMethod = customAuthenticationMethod;
-            return this;
-        }
-
-        public IImapReceiver CustomAuthentication(Action<IImapClient> customAuthenticationMethod) =>
-            CustomAuthentication((imapClient) =>
-            {
-                customAuthenticationMethod(imapClient);
-                return Task.CompletedTask;
-            });
-
         public async ValueTask<IImapClient> ConnectAuthenticatedImapClientAsync(CancellationToken cancellationToken = default)
         {
             await ConnectImapClientAsync(cancellationToken).ConfigureAwait(false);

source/MailKitSimplified.Sender/Services/SmtpSender.cs

@@ -78,6 +78,12 @@ public SmtpSender SetCredential(string username, string password)
             return this;
         }
 
+        public SmtpSender SetCustomAuthentication(Func<ISmtpClient, Task> customAuthenticationMethod)
+        {
+            _customAuthenticationMethod = customAuthenticationMethod;
+            return this;
+        }
+
         public IEmailWriter WriteEmail => new EmailWriter(this);
 
         public static IProtocolLogger GetProtocolLogger(string logFilePath = null, bool append = false, IFileSystem fileSystem = null)
@@ -103,19 +109,6 @@ public static IProtocolLogger GetProtocolLogger(string logFilePath = null, bool
             return protocolLogger;
         }
 
-        public ISmtpSender CustomAuthentication(Func<ISmtpClient, Task> customAuthenticationMethod)
-        {
-            _customAuthenticationMethod = customAuthenticationMethod;
-            return this;
-        }
-
-        public ISmtpSender CustomAuthentication(Action<ISmtpClient> customAuthenticationMethod) =>
-            CustomAuthentication((imapClient) =>
-            {
-                customAuthenticationMethod(imapClient);
-                return Task.CompletedTask;
-            });
-
         public async ValueTask<ISmtpClient> ConnectSmtpClientAsync(CancellationToken cancellationToken = default) =>
             await GetConnectedAuthenticatedAsync(cancellationToken).ConfigureAwait(false);
 

tests/MailKitSimplified.Receiver.Tests/ImapReceiverUnitTests.cs

@@ -10,6 +10,7 @@
 using MailKitSimplified.Receiver.Services;
 using MailKitSimplified.Receiver.Abstractions;
 using MailKitSimplified.Receiver.Models;
+using MailKit.Net.Smtp;
 
 namespace MailKitSimplified.Receiver.Tests
 {
@@ -73,7 +74,8 @@ public void CreateImapReceiver_WithFluentMethods_ReturnsImapReceiver()
                 .SetPort(It.IsAny<ushort>())
                 .SetCredential(It.IsAny<string>(), It.IsAny<string>())
                 .SetProtocolLog(It.IsAny<string>())
-                .SetFolder(It.IsAny<string>());
+                .SetFolder(It.IsAny<string>())
+                .SetCustomAuthentication(It.IsAny<Func<IImapClient, Task>>());
             Assert.NotNull(imapReceiver);
             Assert.IsAssignableFrom<IImapReceiver>(imapReceiver);
         }

tests/MailKitSimplified.Sender.Tests/SmtpSenderUnitTests.cs

@@ -12,6 +12,7 @@
 using MailKitSimplified.Sender.Models;
 using MailKitSimplified.Sender.Abstractions;
 using System.IO.Abstractions.TestingHelpers;
+using MailKit.Net.Imap;
 
 namespace MailKitSimplified.Sender.Tests
 {
@@ -73,10 +74,13 @@ public void CreateSmtpSender_WithAnyHostAndCredential_ReturnsSmtpSender()
         [Fact]
         public void CreateSmtpSender_WithFluentMethods_ReturnsSmtpSender()
         {
+            // Act
             using var smtpSender = SmtpSender.Create(_localhost)
                 .SetPort(It.IsAny<ushort>())
                 .SetCredential(It.IsAny<string>(), It.IsAny<string>())
-                .SetProtocolLog(It.IsAny<string>());
+                .SetProtocolLog(It.IsAny<string>())
+                .SetCustomAuthentication(It.IsAny<Func<ISmtpClient, Task>>());
+            // Assert
             Assert.NotNull(smtpSender);
             Assert.IsAssignableFrom<ISmtpSender>(smtpSender);
         }