public void TestLargeStreamCopy()
{
FakeRandomGenerator frg = new FakeRandomGenerator();
using (MemoryStream source = new MemoryStream(frg.Generate(1000000)))
{
using (MemoryStream destination = new MemoryStream())
{
using (PipelineStream pipeline = new PipelineStream(new CancellationToken()))
{
Task producerTask = Task.Run(() =>
{
source.CopyTo(pipeline);
pipeline.Complete();
});
Task consumerTask = Task.Run(() =>
{
pipeline.CopyTo(destination);
});
Task.WaitAll(producerTask, consumerTask);
}
Assert.That(source.ToArray().IsEquivalentTo(destination.ToArray()), "The source and destination should be the same after passing through the pipeline.");
}
}
}
public void TestLargeWithVaryingChunkSizes()
{
FakeRandomGenerator frg = new FakeRandomGenerator();
byte[] source = frg.Generate(500000);
int[] chunkSizes = { 1, 2, 3, 5, 7, 11, 13, 17, 19, 64, 128, 256, 257, };
using (MemoryStream destination = new MemoryStream())
{
using (PipelineStream pipeline = new PipelineStream(new CancellationToken()))
{
Task producerTask = Task.Run(() =>
{
int i = 0;
int total = 0;
while (total < source.Length)
{
int count = chunkSizes[i++ % chunkSizes.Length];
count = total + count > source.Length ? source.Length - total : count;
pipeline.Write(source, total, count);
total += count;
}
pipeline.Complete();
});
Task consumerTask = Task.Run(() =>
{
byte[] read = new byte[17];
int count;
while ((count = pipeline.Read(read, 0, read.Length)) > 0)
{
destination.Write(read, 0, count);
}
});
Task.WaitAll(producerTask, consumerTask);
}
byte[] result = destination.ToArray();
Assert.That(source.IsEquivalentTo(result), "The source and result should be the same after passing through the pipeline.");
}
}
