public InternalProcessStreamReader(StreamReader processStreamReader)
{
this.processStream = processStreamReader.BaseStream;
this.pipe = new Pipe();
this.reader = new StreamReader(this.pipe.OutputStream);
this.task = Task.Run(() => this.BufferLoop());
}
public void TimeoutTest()
{
var pipe = new Pipe { OutputStream = { ReadTimeout = 0 } };
UnitTestHelpers.AssertThrows<TimeoutException>(() => pipe.OutputStream.ReadByte());
pipe.WriteText(new string('a', 2048));
pipe.ReadTextAsync(2048).Result.ShouldEqual(new string('a', 2048));
}
public void TestPartialWriteDoesNotTimeout()
{
var pipe = new Pipe { InputStream = { WriteTimeout = 0 }, OutputStream = { ReadTimeout = 1000 } };
pipe.SetFixedLength();
var text = Enumerable.Repeat((byte)'t', 3 * Constants.ByteBufferSize).ToArray();
var asyncWrite = pipe.InputStream.WriteAsync(text, 0, text.Length);
asyncWrite.Wait(TimeSpan.FromSeconds(.01)).ShouldEqual(false);
pipe.ReadTextAsync(text.Length).Result.ShouldEqual(new string(text.Select(b => (char)b).ToArray()));
}
public void TestWriteTimeout()
{
var pipe = new Pipe { InputStream = { WriteTimeout = 0 } };
pipe.SetFixedLength();
pipe.WriteText(new string('a', 2 * Constants.ByteBufferSize));
var asyncWrite = pipe.InputStream.WriteAsync(new byte[1], 0, 1);
asyncWrite.ContinueWith(_ => { }).Wait(TimeSpan.FromSeconds(.01)).ShouldEqual(true);
asyncWrite.IsFaulted.ShouldEqual(true);
Assert.IsInstanceOfType(asyncWrite.Exception.InnerException, typeof(TimeoutException));
}
public void TestLargeStreamWithFixedLength()
{
var pipe = new Pipe();
pipe.SetFixedLength();
var bytes = Enumerable.Range(0, 5 * Constants.ByteBufferSize)
.Select(b => (byte)(b % 256))
.ToArray();
var asyncWrite = pipe.InputStream.WriteAsync(bytes, 0, bytes.Length)
.ContinueWith(_ => pipe.InputStream.Close());
var memoryStream = new MemoryStream();
var buffer = new byte[777];
int bytesRead;
while ((bytesRead = pipe.OutputStream.Read(buffer, 0, buffer.Length)) > 0)
{
memoryStream.Write(buffer, 0, bytesRead);
}
asyncWrite.Wait(TimeSpan.FromSeconds(1)).ShouldEqual(true);
memoryStream.ToArray().SequenceEqual(bytes).ShouldEqual(true);
}
public void SimpleTest()
{
var pipe = new Pipe();
pipe.WriteText("abc");
pipe.ReadTextAsync(3).Result.ShouldEqual("abc");
pipe.WriteText("1");
pipe.WriteText("2");
pipe.WriteText("3");
pipe.ReadTextAsync(3).Result.ShouldEqual("123");
var asyncRead = pipe.ReadTextAsync(100);
asyncRead.Wait(TimeSpan.FromSeconds(.01)).ShouldEqual(false, asyncRead.IsCompleted ? "Found: " + (asyncRead.Result ?? "null") : "not complete");
pipe.WriteText("x");
asyncRead.Wait(TimeSpan.FromSeconds(.01)).ShouldEqual(false, asyncRead.IsCompleted ? "Found: " + (asyncRead.Result ?? "null") : "not complete");
pipe.WriteText(new string('y', 100));
asyncRead.Wait(TimeSpan.FromSeconds(5)).ShouldEqual(true, asyncRead.IsCompleted ? "Found: " + (asyncRead.Result ?? "null") : "not complete");
asyncRead.Result.ShouldEqual("x" + new string('y', 99));
}
public PipeOutputStream(Pipe pipe) { this.pipe = pipe; }
public PipeInputStream(Pipe pipe) { this.pipe = pipe; }
public void TestCancel()
{
var pipe = new Pipe();
var cancellationTokenSource = new CancellationTokenSource();
var asyncRead = pipe.ReadTextAsync(1, cancellationTokenSource.Token);
asyncRead.Wait(TimeSpan.FromSeconds(.01)).ShouldEqual(false);
cancellationTokenSource.Cancel();
asyncRead.ContinueWith(_ => { }).Wait(TimeSpan.FromSeconds(5)).ShouldEqual(true);
asyncRead.IsCanceled.ShouldEqual(true);
pipe.WriteText("aaa");
pipe.ReadTextAsync(2).Result.ShouldEqual("aa");
asyncRead = pipe.ReadTextAsync(1, cancellationTokenSource.Token);
asyncRead.ContinueWith(_ => { }).Wait(TimeSpan.FromSeconds(5)).ShouldEqual(true);
asyncRead.IsCanceled.ShouldEqual(true);
}
public void FuzzTest()
{
const int ByteCount = 100000;
var pipe = new Pipe();
var writeTask = Task.Run(async () =>
{
var memoryStream = new MemoryStream();
var random = new Random(1234);
var bytesWritten = 0;
while (bytesWritten < ByteCount)
{
//Console.WriteLine("Writing " + memoryStream.Length);
switch (random.Next(10))
{
case 1:
//Console.WriteLine("SETTING FIXED LENGTH");
pipe.SetFixedLength();
break;
case 2:
case 3:
await Task.Delay(1);
break;
default:
var bufferLength = random.Next(0, 5000);
var offset = random.Next(0, bufferLength + 1);
var count = random.Next(0, bufferLength - offset + 1);
var buffer = new byte[bufferLength];
random.NextBytes(buffer);
memoryStream.Write(buffer, offset, count);
//Console.WriteLine("WRITE START");
await pipe.InputStream.WriteAsync(buffer, offset, count);
//Console.WriteLine("WRITE END");
bytesWritten += count;
break;
}
}
//Console.WriteLine("WRITER ALL DONE");
pipe.InputStream.Close();
return memoryStream;
});
var readTask = Task.Run(async () =>
{
var memoryStream = new MemoryStream();
var random = new Random(5678);
while (true)
{
//Console.WriteLine("Reading " + memoryStream.Length);
if (random.Next(10) == 1)
{
await Task.Delay(1);
}
var bufferLength = random.Next(0, 5000);
var offset = random.Next(0, bufferLength + 1);
var count = random.Next(0, bufferLength - offset + 1);
var buffer = new byte[bufferLength];
//Console.WriteLine("READ START");
var bytesRead = await pipe.OutputStream.ReadAsync(buffer, offset, count);
//Console.WriteLine("READ END");
if (bytesRead == 0 && count > 0)
{
// if we tried to read more than 1 byte and we got 0, the pipe is done
//Console.WriteLine("READER ALL DONE");
return memoryStream;
}
memoryStream.Write(buffer, offset, bytesRead);
}
});
Task.WhenAll(writeTask, readTask).Wait(TimeSpan.FromSeconds(5)).ShouldEqual(true);
CollectionAssert.AreEqual(writeTask.Result.ToArray(), readTask.Result.ToArray());
}
public void TestConcurrentWrites()
{
var pipe = new Pipe();
pipe.SetFixedLength();
var longText = new string('x', (2 * Constants.ByteBufferSize) + 1);
var asyncWrite = pipe.WriteTextAsync(longText);
asyncWrite.Wait(TimeSpan.FromSeconds(.01)).ShouldEqual(false);
UnitTestHelpers.AssertThrows<InvalidOperationException>(() => pipe.InputStream.WriteByte(101));
pipe.OutputStream.Close();
asyncWrite.Wait(TimeSpan.FromSeconds(5)).ShouldEqual(true);
}
public void TestConcurrentReads()
{
var pipe = new Pipe();
var asyncRead = pipe.ReadTextAsync(1);
UnitTestHelpers.AssertThrows<InvalidOperationException>(() => pipe.OutputStream.ReadByte());
pipe.InputStream.Close();
asyncRead.Wait(TimeSpan.FromSeconds(5)).ShouldEqual(true);
}
public void TestCloseReadSide()
{
var pipe = new Pipe();
pipe.WriteText("abc");
pipe.ReadTextAsync(2).Result.ShouldEqual("ab");
pipe.OutputStream.Close();
UnitTestHelpers.AssertThrows<ObjectDisposedException>(() => pipe.OutputStream.ReadByte());
var largeBytes = new byte[10 * 1024];
var initialMemory = GC.GetTotalMemory(forceFullCollection: true);
for (var i = 0; i < int.MaxValue / 1024; ++i)
{
pipe.InputStream.Write(largeBytes, 0, largeBytes.Length);
}
var finalMemory = GC.GetTotalMemory(forceFullCollection: true);
Assert.IsTrue(finalMemory - initialMemory < 10 * largeBytes.Length, "final = " + finalMemory + " initial = " + initialMemory);
UnitTestHelpers.AssertThrows<ObjectDisposedException>(() => pipe.OutputStream.ReadByte());
pipe.InputStream.Close();
}
public void TestCloseWriteSide()
{
var pipe = new Pipe();
pipe.WriteText("123456");
pipe.InputStream.Close();
UnitTestHelpers.AssertThrows<ObjectDisposedException>(() => pipe.InputStream.WriteByte(1));
pipe.ReadTextAsync(5).Result.ShouldEqual("12345");
pipe.ReadTextAsync(2).Result.ShouldEqual(null);
pipe = new Pipe();
var asyncRead = pipe.ReadTextAsync(1);
asyncRead.Wait(TimeSpan.FromSeconds(.01)).ShouldEqual(false);
pipe.InputStream.Close();
asyncRead.Wait(TimeSpan.FromSeconds(5)).ShouldEqual(true);
}
public void TestPartialWriteDoesNotCancel()
{
var pipe = new Pipe();
var cancellationTokenSource = new CancellationTokenSource();
pipe.WriteText(new string('a', 2 * Constants.ByteBufferSize));
var asyncRead = pipe.ReadTextAsync(1);
asyncRead.Wait(TimeSpan.FromSeconds(5)).ShouldEqual(true);
asyncRead.Result.ShouldEqual("a");
pipe.SetFixedLength();
var bytes = Enumerable.Repeat((byte)'b', Constants.ByteBufferSize).ToArray();
var asyncWrite = pipe.InputStream.WriteAsync(bytes, 0, bytes.Length, cancellationTokenSource.Token);
asyncWrite.Wait(TimeSpan.FromSeconds(.01)).ShouldEqual(false);
cancellationTokenSource.Cancel();
asyncWrite.Wait(TimeSpan.FromSeconds(.01)).ShouldEqual(false);
asyncRead = pipe.ReadTextAsync((3 * Constants.ByteBufferSize) - 1);
asyncRead.Wait(TimeSpan.FromSeconds(5)).ShouldEqual(true);
asyncRead.Result.ShouldEqual(new string('a', (2 * Constants.ByteBufferSize) - 1) + new string('b', Constants.ByteBufferSize));
}
public void TestCancelWrite()
{
var pipe = new Pipe();
var cancellationTokenSource = new CancellationTokenSource();
pipe.WriteText(new string('a', 2 * Constants.ByteBufferSize));
pipe.SetFixedLength();
var bytes = Enumerable.Repeat((byte)'b', Constants.ByteBufferSize).ToArray();
var asyncWrite = pipe.InputStream.WriteAsync(bytes, 0, bytes.Length, cancellationTokenSource.Token);
asyncWrite.Wait(TimeSpan.FromSeconds(.01)).ShouldEqual(false);
cancellationTokenSource.Cancel();
asyncWrite.ContinueWith(_ => { }).Wait(TimeSpan.FromSeconds(5)).ShouldEqual(true);
asyncWrite.IsCanceled.ShouldEqual(true);
}
