public void TestBlockAllocation()
{
var repository = new MockRepository();
var mockDiskAccess = repository.DynamicMock<IDirectDiskAccess>();
var mockAllocator = repository.StrictMock<IBlockAllocator>();
int count = 0;
Expect.Call(mockAllocator.AllocateBlocks(1)).Return(new []{new Address(0)}).Repeat.Any().WhenCalled(f => count++);
Expect.Call(() => mockAllocator.FreeBlocks(null)).IgnoreArguments().Repeat.Any().WhenCalled(f => count--);
var storage = new BlockAddressStorage(mockDiskAccess, mockAllocator, new Address(0), new Address(0));
repository.ReplayAll();
for(int i = 0; i < 10000; ++i)
{
storage.AddBlocks(1);
}
Assert.AreEqual(10000, storage.NumBlocksAllocated);
Assert.IsNotNull(storage.GetBlockStartAddress(0));
Assert.IsNotNull(storage.GetBlockStartAddress(9999));
storage.FreeLastBlocks(5000);
Assert.AreEqual(5000, storage.NumBlocksAllocated);
Assert.IsNotNull(storage.GetBlockStartAddress(4999));
Assert.Throws<IndexOutOfRangeException>(() => storage.GetBlockStartAddress(5000));
storage.FreeLastBlocks(5000);
Assert.AreEqual(0, storage.NumBlocksAllocated);
Assert.Throws<IndexOutOfRangeException>(() => storage.GetBlockStartAddress(0));
Assert.Throws<ArgumentOutOfRangeException>(() => storage.FreeLastBlocks(1));
Assert.AreEqual(0, count);
repository.VerifyAll();
}