Merge pull request #38 from jonfuller/customizable_output_stream

Adds option for setting the output TextWriter
khalidabuhakmeh · Dec 17, 2019 · 43d2631 · 43d2631
2 parents 1e9ad1e + 27ea990
commit 43d2631
Show file tree

Hide file tree

Showing 2 changed files with 43 additions and 4 deletions.
diff --git a/ConsoleTables.Tests/ConsoleTableTest.cs b/ConsoleTables.Tests/ConsoleTableTest.cs
@@ -1,4 +1,6 @@
+using System;
 using System.Collections.Generic;
+using System.IO;
 using Xunit;
 
 namespace ConsoleTables.Tests
@@ -86,6 +88,37 @@ public void NumberShouldBeRightAlignedOnMarkDown()
 ", table);
         }
 
+        [Fact]
+        public void OutputShouldDefaultToConsoleOut()
+        {
+            var users = new List<User>
+            {
+                new User { Name = "Alexandre" , Age = 36 }
+            };
+
+            var table = ConsoleTable.From(users);
+
+            Assert.Equal(table.Options.OutputTo, Console.Out);
+        }
+
+        [Fact]
+        public void OutputShouldGoToConfiguredOutputWriter()
+        {
+            var users = new List<User>
+            {
+                new User { Name = "Alexandre" , Age = 36 }
+            };
+
+            var testWriter = new StringWriter();
+
+            ConsoleTable
+               .From(users)
+               .Configure(o => o.OutputTo = testWriter)
+               .Write();
+
+            Assert.NotEmpty(testWriter.ToString());
+        }
+
         class User
         {
             public string Name { get; set; }

diff --git a/src/ConsoleTables/ConsoleTable.cs b/src/ConsoleTables/ConsoleTable.cs
@@ -1,5 +1,6 @@
 using System;
 using System.Collections.Generic;
+using System.IO;
 using System.Linq;
 using System.Reflection;
 using System.Text;
@@ -241,16 +242,16 @@ public void Write(Format format = ConsoleTables.Format.Default)
             switch (format)
             {
                 case ConsoleTables.Format.Default:
-                    Console.WriteLine(ToString());
+                    Options.OutputTo.WriteLine(ToString());
                     break;
                 case ConsoleTables.Format.MarkDown:
-                    Console.WriteLine(ToMarkDownString());
+                    Options.OutputTo.WriteLine(ToMarkDownString());
                     break;
                 case ConsoleTables.Format.Alternative:
-                    Console.WriteLine(ToStringAlternative());
+                    Options.OutputTo.WriteLine(ToStringAlternative());
                     break;
                 case ConsoleTables.Format.Minimal:
-                    Console.WriteLine(ToMinimalString());
+                    Options.OutputTo.WriteLine(ToMinimalString());
                     break;
                 default:
                     throw new ArgumentOutOfRangeException(nameof(format), format, null);
@@ -282,6 +283,11 @@ public class ConsoleTableOptions
         /// Enable only from a list of objects
         /// </summary>
         public Alignment NumberAlignment { get; set; } = Alignment.Left;
+
+        /// <summary>
+        /// The <see cref="TextWriter"/> to write to. Defaults to <see cref="Console.Out"/>.
+        /// </summary>
+        public TextWriter OutputTo { get; set; } = Console.Out;
     }
 
     public enum Format