cs-codex-dist-tests/FileUtils/FileManager.cs

157 lines
4.6 KiB
C#
Raw Normal View History

2023-04-12 14:06:04 +00:00
using Logging;
using NUnit.Framework;
using Utils;
2023-04-12 14:06:04 +00:00
namespace FileUtils
2023-04-12 14:06:04 +00:00
{
public interface IFileManager
{
TestFile CreateEmptyTestFile(string label = "");
TestFile GenerateTestFile(ByteSize size, string label = "");
2023-04-12 14:06:04 +00:00
void DeleteAllTestFiles();
void ScopedFiles(Action action);
T ScopedFiles<T>(Func<T> action);
2023-04-12 14:06:04 +00:00
}
public class FileManager : IFileManager
{
2023-06-07 07:59:00 +00:00
public const int ChunkSize = 1024 * 1024 * 100;
private static NumberSource folderNumberSource = new NumberSource(0);
2023-04-12 14:06:04 +00:00
private readonly Random random = new Random();
private readonly BaseLog log;
2023-04-12 14:06:04 +00:00
private readonly string folder;
private readonly List<List<TestFile>> fileSetStack = new List<List<TestFile>>();
2023-04-12 14:06:04 +00:00
public FileManager(BaseLog log, string rootFolder)
2023-04-12 14:06:04 +00:00
{
folder = Path.Combine(rootFolder, folderNumberSource.GetNextNumber().ToString("D5"));
2023-04-12 14:06:04 +00:00
EnsureDirectory();
2023-04-12 14:06:04 +00:00
this.log = log;
}
public TestFile CreateEmptyTestFile(string label = "")
2023-04-12 14:06:04 +00:00
{
var path = Path.Combine(folder, Guid.NewGuid().ToString() + "_test.bin");
var result = new TestFile(log, path, label);
2023-04-12 14:06:04 +00:00
File.Create(result.Filename).Close();
if (fileSetStack.Any()) fileSetStack.Last().Add(result);
2023-04-12 14:06:04 +00:00
return result;
}
public TestFile GenerateTestFile(ByteSize size, string label)
2023-04-12 14:06:04 +00:00
{
2023-06-07 07:59:00 +00:00
var sw = Stopwatch.Begin(log);
var result = GenerateFile(size, label);
sw.End($"Generated file '{result.Describe()}'.");
2023-04-12 14:06:04 +00:00
return result;
}
public void DeleteAllTestFiles()
{
DeleteDirectory();
2023-04-12 14:06:04 +00:00
}
public void ScopedFiles(Action action)
{
PushFileSet();
action();
PopFileSet();
}
public T ScopedFiles<T>(Func<T> action)
{
PushFileSet();
var result = action();
PopFileSet();
return result;
}
private void PushFileSet()
{
fileSetStack.Add(new List<TestFile>());
}
private void PopFileSet()
{
if (!fileSetStack.Any()) return;
var pop = fileSetStack.Last();
fileSetStack.Remove(pop);
foreach (var file in pop)
{
try
{
File.Delete(file.Filename);
}
catch { }
}
}
2023-06-07 07:59:00 +00:00
private TestFile GenerateFile(ByteSize size, string label)
{
var result = CreateEmptyTestFile(label);
CheckSpaceAvailable(result, size);
GenerateFileBytes(result, size);
return result;
}
private void CheckSpaceAvailable(TestFile testFile, ByteSize size)
{
var file = new FileInfo(testFile.Filename);
var drive = new DriveInfo(file.Directory!.Root.FullName);
var spaceAvailable = drive.TotalFreeSpace;
if (spaceAvailable < size.SizeInBytes)
{
var msg = $"Inconclusive: Not enough disk space to perform test. " +
$"{Formatter.FormatByteSize(size.SizeInBytes)} required. " +
$"{Formatter.FormatByteSize(spaceAvailable)} available.";
log.Log(msg);
Assert.Inconclusive(msg);
}
}
2023-04-12 14:06:04 +00:00
private void GenerateFileBytes(TestFile result, ByteSize size)
{
long bytesLeft = size.SizeInBytes;
2023-06-07 07:59:00 +00:00
int chunkSize = ChunkSize;
2023-04-12 14:06:04 +00:00
while (bytesLeft > 0)
{
2023-06-07 07:59:00 +00:00
try
{
var length = Math.Min(bytesLeft, chunkSize);
AppendRandomBytesToFile(result, length);
bytesLeft -= length;
}
catch
{
chunkSize = chunkSize / 2;
if (chunkSize < 1024) throw;
}
2023-04-12 14:06:04 +00:00
}
}
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);
}
private void EnsureDirectory()
{
if (!Directory.Exists(folder)) Directory.CreateDirectory(folder);
}
private void DeleteDirectory()
{
Directory.Delete(folder, true);
}
2023-04-12 14:06:04 +00:00
}
}