Added unit tests for MimeEntity.WriteTo(fileName) and Load(fileName)

Author: jstedfast

UnitTests/MimePartTests.cs

@@ -24,7 +24,9 @@
 // THE SOFTWARE.
 //
 
+using System.IO;
 using System.Text;
+using System.Buffers;
 
 using MimeKit;
 using MimeKit.Utils;
@@ -523,16 +525,16 @@ public async Task TestTranscodingAsync ()
 			}
 		}
 
-		[TestCase ("content", TestName = "TestWriteToNoNewLine")]
-		[TestCase ("content\r\n", TestName = "TestWriteToNewLine")]
+		[TestCase ("content", TestName = "TestWriteTo_NoNewLine")]
+		[TestCase ("content\r\n", TestName = "TestWriteTo_NewLine")]
 		public void TestWriteTo (string text)
 		{
 			var builder = new BodyBuilder ();
 
 			builder.Attachments.Add ("filename", new MemoryStream (Encoding.UTF8.GetBytes (text)));
 			builder.TextBody = "This is the text body.";
 
-			var body = builder.ToMessageBody ();
+			using var body = builder.ToMessageBody ();
 
 			using (var stream = new MemoryStream ()) {
 				var options = FormatOptions.Default.Clone ();
@@ -541,7 +543,7 @@ public void TestWriteTo (string text)
 				body.WriteTo (options, stream);
 				stream.Position = 0;
 
-				var multipart = (Multipart) MimeEntity.Load (stream);
+				using var multipart = (Multipart) MimeEntity.Load (stream);
 				using (var input = ((MimePart) multipart[1]).Content.Open ()) {
 					var buffer = new byte[1024];
 					int n;
@@ -555,16 +557,16 @@ public void TestWriteTo (string text)
 			}
 		}
 
-		[TestCase ("content", TestName = "TestWriteToNoNewLine")]
-		[TestCase ("content\r\n", TestName = "TestWriteToNewLine")]
+		[TestCase ("content", TestName = "TestWriteToAsync_NoNewLine")]
+		[TestCase ("content\r\n", TestName = "TestWriteToAsync_NewLine")]
 		public async Task TestWriteToAsync (string text)
 		{
 			var builder = new BodyBuilder ();
 
 			builder.Attachments.Add ("filename", new MemoryStream (Encoding.UTF8.GetBytes (text)));
 			builder.TextBody = "This is the text body.";
 
-			var body = builder.ToMessageBody ();
+			using var body = builder.ToMessageBody ();
 
 			using (var stream = new MemoryStream ()) {
 				var options = FormatOptions.Default.Clone ();
@@ -573,7 +575,7 @@ public async Task TestWriteToAsync (string text)
 				await body.WriteToAsync (options, stream);
 				stream.Position = 0;
 
-				var multipart = (Multipart) await MimeEntity.LoadAsync (stream);
+				using var multipart = (Multipart) await MimeEntity.LoadAsync (stream);
 				using (var input = ((MimePart) multipart[1]).Content.Open ()) {
 					var buffer = new byte[1024];
 					int n;
@@ -587,6 +589,62 @@ public async Task TestWriteToAsync (string text)
 			}
 		}
 
+		[TestCase ("content", TestName = "TestWriteToFile_NoNewLine")]
+		[TestCase ("content\r\n", TestName = "TestWriteToFile_NewLine")]
+		public void TestWriteToFile (string text)
+		{
+			var fileName = Path.GetTempFileName ();
+			using var body = new TextPart ("plain") {
+				Text = text
+			};
+
+			try {
+				var options = FormatOptions.Default.Clone ();
+				options.NewLineFormat = NewLineFormat.Dos;
+
+				body.WriteTo (options, fileName);
+
+				using (var loaded = (TextPart) MimeEntity.Load (fileName))
+					Assert.That (loaded.Text, Is.EqualTo (text));
+
+				options.NewLineFormat = NewLineFormat.Unix;
+				body.WriteTo (options, fileName);
+
+				using (var loaded = (TextPart) MimeEntity.Load (fileName))
+					Assert.That (loaded.Text, Is.EqualTo (text));
+			} finally {
+				File.Delete (fileName);
+			}
+		}
+
+		[TestCase ("content", TestName = "TestWriteToFileAsync_NoNewLine")]
+		[TestCase ("content\r\n", TestName = "TestWriteToFileAsync_NewLine")]
+		public async Task TestWriteToFileAsync (string text)
+		{
+			var fileName = Path.GetTempFileName ();
+			using var body = new TextPart ("plain") {
+				Text = text
+			};
+
+			try {
+				var options = FormatOptions.Default.Clone ();
+				options.NewLineFormat = NewLineFormat.Dos;
+
+				await body.WriteToAsync (options, fileName);
+
+				using (var loaded = (TextPart) await MimeEntity.LoadAsync (fileName))
+					Assert.That (loaded.Text, Is.EqualTo (text));
+
+				options.NewLineFormat = NewLineFormat.Unix;
+				await body.WriteToAsync (options, fileName);
+
+				using (var loaded = (TextPart) await MimeEntity.LoadAsync (fileName))
+					Assert.That (loaded.Text, Is.EqualTo (text));
+			} finally {
+				File.Delete (fileName);
+			}
+		}
+
 		[Test]
 		public void TestLoadHttpWebResponse ()
 		{