public void GetByteArrayTest()
{
// Arrage
Header header = new Header
{
BlocksCount = 10,
BlockSize = 1024,
HashSize = 16,
HashType = HashTypes.MD5
};
// Act
MemoryStream stream = new MemoryStream();
BlockStreamWriter streamWriter = new BlockStreamWriter(stream);
streamWriter.WriteHeader(header);
var buffer = stream.ToArray();
// Assert
Assert.AreEqual(buffer.Length, 13);
}
public void GetByteArrayTest()
{
// Arrage
Block block = new Block
{
Id = 1,
OriginBlockSize = 1,
Hash = new byte[] { 1, 2, 3, 4 },
Data = new byte[] { 1, 2, 3, 4 }
};
// Act
MemoryStream stream = new MemoryStream();
BlockStreamWriter streamWriter = new BlockStreamWriter(stream);
streamWriter.WriteBlock(block);
var buffer = stream.ToArray();
// Assert
Assert.AreEqual(buffer.Length, 20);
}
public void ReadWriteCompressedTest()
{
// Assert
byte[] originBytes = { 0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 0, 1, 2, 3, 4, 5, 6, 7, 8, 9 };
BlocksHandler handler = new BlocksHandler(1, HashTypes.MD5, OperationType.Compress, Console.WriteLine);
DefaultStreamReader dReader = new DefaultStreamReader(new MemoryStream(originBytes), 11);
Header h1 = new Header
{
BlocksCount = 3,
BlockSize = 11,
HashSize = 16,
HashType = HashTypes.MD5
};
var b1 = dReader.GetNextBlock();
var b2 = dReader.GetNextBlock();
var b3 = dReader.GetNextBlock();
handler.AddUnhandledBlock(new Block
{
Data = b1.Data,
Hash = b1.Hash,
OriginBlockSize = b1.OriginBlockSize,
Id = b1.Id
});
handler.AddUnhandledBlock(new Block
{
Data = b3.Data,
Hash = b3.Hash,
OriginBlockSize = b3.OriginBlockSize,
Id = b3.Id
});
handler.AddUnhandledBlock(new Block
{
Data = b2.Data,
Hash = b2.Hash,
OriginBlockSize = b2.OriginBlockSize,
Id = b2.Id
});
handler.Start();
handler.Stop();
List<Block> array = handler.GetAvailableBlocks();
MemoryStream bufferStream = new MemoryStream();
BlockStreamWriter cryptedStreamWriter = new BlockStreamWriter(bufferStream);
cryptedStreamWriter.WriteHeader(h1);
for (int i = 0; i < array.Count(); i++)
{
cryptedStreamWriter.WriteBlock(array[i]);
}
bufferStream.Seek(0, SeekOrigin.Begin);
BlockStreamReader cryptedStreamReader = new BlockStreamReader(bufferStream);
Header h2 = cryptedStreamReader.GetHeader();
var b11 = cryptedStreamReader.GetNextBlock();
var b21 = cryptedStreamReader.GetNextBlock();
var b31 = cryptedStreamReader.GetNextBlock();
handler = new BlocksHandler(1, HashTypes.MD5, OperationType.Decompress, Console.WriteLine);
handler.AddUnhandledBlock(b11);
handler.AddUnhandledBlock(b21);
handler.AddUnhandledBlock(b31);
handler.Start();
handler.Stop();
List<Block> array2 = handler.GetAvailableBlocks();
Assert.AreEqual(array2.Count(), 3);
Assert.AreEqual(h1.BlocksCount, h2.BlocksCount);
Assert.AreEqual(h1.BlockSize, h2.BlockSize);
Assert.AreEqual(h1.HashSize, h2.HashSize);
Assert.AreEqual(h1.HashType, h2.HashType);
Assert.AreEqual(b1.Id, b11.Id);
Assert.AreEqual(b1.OriginBlockSize, b11.OriginBlockSize);
Assert.AreEqual(b1.Data, b11.Data);
Assert.AreEqual(b3.Id, b21.Id);
Assert.AreEqual(b3.OriginBlockSize, b21.OriginBlockSize);
Assert.AreEqual(b3.Data, b21.Data);
Assert.AreEqual(b2.Id, b31.Id);
Assert.AreEqual(b2.OriginBlockSize, b31.OriginBlockSize);
Assert.AreEqual(b2.Data, b31.Data);
}
