public void ShouldTakeLessThanTotal()
{
using (var work = new WorkRemaining <int>(Enumerable.Repeat(5, 5)))
{
work.AddAdditional(Enumerable.Repeat(3, 3));
var dataReturned = work.Take(4).ToArray();
var expected = Enumerable.Repeat(3, 3).Concat(Enumerable.Repeat(5, 1)).ToArray();
Assert.That(dataReturned, Is.EqualTo(expected));
}
}
public void ShouldReturnFromBufferBeforeSequence()
{
using (var work = new WorkRemaining <int>(Enumerable.Repeat(5, 5)))
{
work.AddAdditional(Enumerable.Repeat(3, 3));
var dataReturned = work.Take(8).ToArray();
var expected = Enumerable.Repeat(3, 3).Concat(Enumerable.Repeat(5, 5)).ToArray();
Assert.That(dataReturned, Is.EqualTo(expected));
}
}
