diff --git a/winPEAS/winPEASexe/Tests/SmokeTests.cs b/winPEAS/winPEASexe/Tests/SmokeTests.cs index 9542bf8..a7716c3 100644 --- a/winPEAS/winPEASexe/Tests/SmokeTests.cs +++ b/winPEAS/winPEASexe/Tests/SmokeTests.cs @@ -1,5 +1,6 @@ using Microsoft.VisualStudio.TestTools.UnitTesting; using System; +using System.IO; namespace winPEAS.Tests { @@ -25,17 +26,28 @@ namespace winPEAS.Tests [TestMethod] public void ShouldDisplayHelp() { + var originalOut = Console.Out; + var sw = new StringWriter(); try { + Console.SetOut(sw); string[] args = new string[] { "help", }; Program.Main(args); + + string output = sw.ToString(); + Assert.IsTrue(output.Contains("WinPEAS is a binary"), + "Help output did not contain expected text."); } catch (Exception e) { Assert.Fail($"Exception thrown: {e.Message}"); } + finally + { + Console.SetOut(originalOut); + } } } }