Merge pull request #4 from madelson/short-flush-fix

Short flush fix
madelson · Jul 12, 2015 · 3cf74ff · 3cf74ff
2 parents 96b4d27 + aace803
commit 3cf74ff
Show file tree

Hide file tree

Showing 16 changed files with 1,512 additions and 743 deletions.
diff --git a/MedallionShell.Tests/GeneralTest.cs b/MedallionShell.Tests/GeneralTest.cs
@@ -238,6 +238,41 @@ public void TestVersioning()
             version.ShouldEqual(informationalVersion.InformationalVersion + ".0");
         }
 
+        [TestMethod]
+        public void TestShortFlush()
+        {
+            var command = Command.Run("SampleCommand", "shortflush", "a");
+            var readCommand = command.StandardOutput.ReadBlockAsync(new char[1], 0, 1);
+            //var readCommand = command.StandardOutput.BaseStream.ReadAsync(new byte[1], 0, 1);
+            readCommand.Wait(TimeSpan.FromSeconds(5)).ShouldEqual(true);
+
+            command.StandardInput.Close();
+            command.Task.Wait(TimeSpan.FromSeconds(5)).ShouldEqual(true);
+        }
+
+        [TestMethod]
+        public void TestAutoFlush()
+        {
+            var command = Command.Run("SampleCommand", "echo", "--per-char");
+            command.StandardInput.AutoFlush.ShouldEqual(true);
+            command.StandardInput.Write('a');
+
+            var buffer = new char[1];
+            var asyncRead = command.StandardOutput.ReadBlockAsync(buffer, 0, 1);
+            asyncRead.Wait(TimeSpan.FromSeconds(3)).ShouldEqual(true);
+            buffer[0].ShouldEqual('a');
+
+            command.StandardInput.AutoFlush = false;
+            command.StandardInput.Write('b');
+            asyncRead = command.StandardOutput.ReadBlockAsync(buffer, 0, 1);
+            asyncRead.Wait(TimeSpan.FromSeconds(.01)).ShouldEqual(false);
+            command.StandardInput.Flush();
+            asyncRead.Wait(TimeSpan.FromSeconds(3)).ShouldEqual(true);
+            buffer[0].ShouldEqual('b');
+
+            command.StandardInput.Close();
+        }
+
         [TestMethod]
         public void TestErrorEcho()
         {

diff --git a/MedallionShell.Tests/MedallionShell.Tests.csproj b/MedallionShell.Tests/MedallionShell.Tests.csproj
@@ -63,6 +63,7 @@
     <Compile Include="GeneralTest.cs" />
     <Compile Include="PipeTest.cs" />
     <Compile Include="Properties\AssemblyInfo.cs" />
+    <Compile Include="Streams\PipeTest.cs" />
     <Compile Include="SyntaxTest.cs" />
     <Compile Include="UnitTestHelpers.cs" />
   </ItemGroup>