114 lines
3.7 KiB
C#
114 lines
3.7 KiB
C#
using Logging;
|
|
using NUnit.Framework;
|
|
|
|
namespace DistTestCore
|
|
{
|
|
public interface IFileManager
|
|
{
|
|
TestFile CreateEmptyTestFile();
|
|
TestFile GenerateTestFile(ByteSize size);
|
|
void DeleteAllTestFiles();
|
|
}
|
|
|
|
public class FileManager : IFileManager
|
|
{
|
|
public const int ChunkSize = 1024 * 1024;
|
|
private readonly Random random = new Random();
|
|
private readonly List<TestFile> activeFiles = new List<TestFile>();
|
|
private readonly TestLog log;
|
|
private readonly string folder;
|
|
|
|
public FileManager(TestLog log, Configuration configuration)
|
|
{
|
|
folder = configuration.GetFileManagerFolder();
|
|
|
|
if (!Directory.Exists(folder)) Directory.CreateDirectory(folder);
|
|
this.log = log;
|
|
}
|
|
|
|
public TestFile CreateEmptyTestFile()
|
|
{
|
|
var result = new TestFile(Path.Combine(folder, Guid.NewGuid().ToString() + "_test.bin"));
|
|
File.Create(result.Filename).Close();
|
|
activeFiles.Add(result);
|
|
return result;
|
|
}
|
|
|
|
public TestFile GenerateTestFile(ByteSize size)
|
|
{
|
|
var result = CreateEmptyTestFile();
|
|
GenerateFileBytes(result, size);
|
|
log.Log($"Generated {size.SizeInBytes} bytes of content for file '{result.Filename}'.");
|
|
return result;
|
|
}
|
|
|
|
public void DeleteAllTestFiles()
|
|
{
|
|
foreach (var file in activeFiles) File.Delete(file.Filename);
|
|
activeFiles.Clear();
|
|
}
|
|
|
|
private void GenerateFileBytes(TestFile result, ByteSize size)
|
|
{
|
|
long bytesLeft = size.SizeInBytes;
|
|
while (bytesLeft > 0)
|
|
{
|
|
var length = Math.Min(bytesLeft, ChunkSize);
|
|
AppendRandomBytesToFile(result, length);
|
|
bytesLeft -= length;
|
|
}
|
|
}
|
|
|
|
private void AppendRandomBytesToFile(TestFile result, long length)
|
|
{
|
|
var bytes = new byte[length];
|
|
random.NextBytes(bytes);
|
|
using var stream = new FileStream(result.Filename, FileMode.Append);
|
|
stream.Write(bytes, 0, bytes.Length);
|
|
}
|
|
}
|
|
|
|
public class TestFile
|
|
{
|
|
public TestFile(string filename)
|
|
{
|
|
Filename = filename;
|
|
}
|
|
|
|
public string Filename { get; }
|
|
|
|
public long GetFileSize()
|
|
{
|
|
var info = new FileInfo(Filename);
|
|
return info.Length;
|
|
}
|
|
|
|
public void AssertIsEqual(TestFile? actual)
|
|
{
|
|
if (actual == null) Assert.Fail("TestFile is null.");
|
|
if (actual == this || actual!.Filename == Filename) Assert.Fail("TestFile is compared to itself.");
|
|
|
|
Assert.That(actual.GetFileSize(), Is.EqualTo(GetFileSize()), "Files are not of equal length.");
|
|
|
|
using var streamExpected = new FileStream(Filename, FileMode.Open, FileAccess.Read);
|
|
using var streamActual = new FileStream(actual.Filename, FileMode.Open, FileAccess.Read);
|
|
|
|
var bytesExpected = new byte[FileManager.ChunkSize];
|
|
var bytesActual = new byte[FileManager.ChunkSize];
|
|
|
|
var readExpected = 0;
|
|
var readActual = 0;
|
|
|
|
while (true)
|
|
{
|
|
readExpected = streamExpected.Read(bytesExpected, 0, FileManager.ChunkSize);
|
|
readActual = streamActual.Read(bytesActual, 0, FileManager.ChunkSize);
|
|
|
|
if (readExpected == 0 && readActual == 0) return;
|
|
Assert.That(readActual, Is.EqualTo(readExpected), "Unable to read buffers of equal length.");
|
|
CollectionAssert.AreEqual(bytesExpected, bytesActual, "Files are not binary-equal.");
|
|
}
|
|
}
|
|
}
|
|
}
|