mirror of
https://github.com/adamhathcock/sharpcompress.git
synced 2026-09-23 07:25:11 +00:00
add AOT smoke and missing tests
This commit is contained in:
Adam Hathcock
312
tests/SharpCompress.Test/AsyncParityAndCancellationTests.cs
Normal file
312
tests/SharpCompress.Test/AsyncParityAndCancellationTests.cs
Normal file
@@ -0,0 +1,312 @@
|
||||
#if NET8_0_OR_GREATER
|
||||
using System;
|
||||
using System.Collections.Generic;
|
||||
using System.IO;
|
||||
using System.Linq;
|
||||
using System.Threading;
|
||||
using System.Threading.Tasks;
|
||||
using SharpCompress.Archives;
|
||||
using SharpCompress.Common;
|
||||
using SharpCompress.Readers;
|
||||
using SharpCompress.Test.Mocks;
|
||||
using SharpCompress.Writers;
|
||||
using Xunit;
|
||||
|
||||
namespace SharpCompress.Test;
|
||||
|
||||
public class AsyncParityAndCancellationTests : TestBase
|
||||
{
|
||||
[Theory]
|
||||
[InlineData("Zip.deflate.zip")]
|
||||
[InlineData("Tar.tar")]
|
||||
[InlineData("Tar.tar.gz")]
|
||||
[InlineData("Rar.rar")]
|
||||
[InlineData("7Zip.nonsolid.7z")]
|
||||
public async Task ArchiveAsyncEntries_ShouldMatchSyncEntries(string archiveName)
|
||||
{
|
||||
var archivePath = Path.Combine(TEST_ARCHIVES_PATH, archiveName);
|
||||
|
||||
var syncEntries = ReadArchiveEntries(archivePath);
|
||||
var asyncEntries = await ReadArchiveEntriesAsync(archivePath);
|
||||
|
||||
Assert.Equal(syncEntries, asyncEntries);
|
||||
}
|
||||
|
||||
[Theory]
|
||||
[InlineData("Zip.deflate.zip")]
|
||||
[InlineData("Tar.tar")]
|
||||
[InlineData("Tar.tar.gz")]
|
||||
[InlineData("Rar.rar")]
|
||||
public async Task ReaderAsyncEntries_ShouldMatchSyncEntries(string archiveName)
|
||||
{
|
||||
var archivePath = Path.Combine(TEST_ARCHIVES_PATH, archiveName);
|
||||
|
||||
var syncEntries = ReadReaderEntries(archivePath);
|
||||
var asyncEntries = await ReadReaderEntriesAsync(archivePath);
|
||||
|
||||
Assert.Equal(syncEntries, asyncEntries);
|
||||
}
|
||||
|
||||
[Fact]
|
||||
public async Task AsyncReaderExtraction_ShouldRespectCancellationBeforeStart()
|
||||
{
|
||||
var archivePath = Path.Combine(TEST_ARCHIVES_PATH, "Zip.deflate.zip");
|
||||
await using var stream = File.OpenRead(archivePath);
|
||||
await using var reader = await ReaderFactory.OpenAsyncReader(stream);
|
||||
using var cts = new CancellationTokenSource();
|
||||
cts.Cancel();
|
||||
|
||||
await Assert.ThrowsAnyAsync<OperationCanceledException>(async () =>
|
||||
await reader.WriteAllToDirectoryAsync(SCRATCH_FILES_PATH, cancellationToken: cts.Token)
|
||||
);
|
||||
}
|
||||
|
||||
[Fact]
|
||||
public async Task AsyncArchiveExtraction_ShouldRespectCancellationBeforeStart()
|
||||
{
|
||||
var archivePath = Path.Combine(TEST_ARCHIVES_PATH, "Zip.deflate.zip");
|
||||
await using var archive = await ArchiveFactory.OpenAsyncArchive(archivePath);
|
||||
using var cts = new CancellationTokenSource();
|
||||
cts.Cancel();
|
||||
|
||||
await Assert.ThrowsAnyAsync<OperationCanceledException>(async () =>
|
||||
await archive.WriteToDirectoryAsync(SCRATCH_FILES_PATH, cancellationToken: cts.Token)
|
||||
);
|
||||
}
|
||||
|
||||
[Fact]
|
||||
public async Task AsyncReaderExtraction_ShouldRespectCancellationDuringRead()
|
||||
{
|
||||
var archiveBytes = CreateLargeTarArchive();
|
||||
using var cts = new CancellationTokenSource();
|
||||
await using var stream = new CancelAfterBytesReadStream(
|
||||
new MemoryStream(archiveBytes),
|
||||
cts,
|
||||
cancelAfterBytes: 2048
|
||||
);
|
||||
await Assert.ThrowsAnyAsync<OperationCanceledException>(async () =>
|
||||
{
|
||||
await using var reader = await ReaderFactory.OpenAsyncReader(
|
||||
stream,
|
||||
cancellationToken: cts.Token
|
||||
);
|
||||
await reader.WriteAllToDirectoryAsync(SCRATCH_FILES_PATH, cancellationToken: cts.Token);
|
||||
});
|
||||
}
|
||||
|
||||
[Fact]
|
||||
public async Task OpenAsyncReader_CallerProvidedStream_ShouldRemainOpenByDefault()
|
||||
{
|
||||
var archiveBytes = await File.ReadAllBytesAsync(
|
||||
Path.Combine(TEST_ARCHIVES_PATH, "Zip.deflate.zip")
|
||||
);
|
||||
var stream = new TestStream(new MemoryStream(archiveBytes));
|
||||
|
||||
await using (var reader = await ReaderFactory.OpenAsyncReader(stream))
|
||||
{
|
||||
Assert.True(await reader.MoveToNextEntryAsync());
|
||||
}
|
||||
|
||||
Assert.False(stream.IsDisposed);
|
||||
}
|
||||
|
||||
[Fact]
|
||||
public async Task OpenAsyncArchive_CallerProvidedStream_ShouldRemainOpenByDefault()
|
||||
{
|
||||
var archiveBytes = await File.ReadAllBytesAsync(
|
||||
Path.Combine(TEST_ARCHIVES_PATH, "Zip.deflate.zip")
|
||||
);
|
||||
var stream = new TestStream(new MemoryStream(archiveBytes));
|
||||
|
||||
await using (var archive = await ArchiveFactory.OpenAsyncArchive(stream))
|
||||
{
|
||||
await foreach (var _ in archive.EntriesAsync)
|
||||
{
|
||||
break;
|
||||
}
|
||||
}
|
||||
|
||||
Assert.False(stream.IsDisposed);
|
||||
}
|
||||
|
||||
private static List<EntrySnapshot> ReadArchiveEntries(string archivePath)
|
||||
{
|
||||
using var archive = ArchiveFactory.OpenArchive(archivePath);
|
||||
return archive
|
||||
.Entries.Where(entry => !entry.IsDirectory)
|
||||
.Select(entry =>
|
||||
{
|
||||
using var stream = entry.OpenEntryStream();
|
||||
using var memory = new MemoryStream();
|
||||
stream.CopyTo(memory);
|
||||
return new EntrySnapshot(
|
||||
entry.Key ?? string.Empty,
|
||||
entry.Size,
|
||||
entry.CompressionType,
|
||||
Convert.ToBase64String(memory.ToArray())
|
||||
);
|
||||
})
|
||||
.OrderBy(entry => entry.Key, StringComparer.Ordinal)
|
||||
.ToList();
|
||||
}
|
||||
|
||||
private static async Task<List<EntrySnapshot>> ReadArchiveEntriesAsync(string archivePath)
|
||||
{
|
||||
await using var archive = await ArchiveFactory.OpenAsyncArchive(archivePath);
|
||||
var entries = new List<EntrySnapshot>();
|
||||
await foreach (var entry in archive.EntriesAsync)
|
||||
{
|
||||
if (entry.IsDirectory)
|
||||
{
|
||||
continue;
|
||||
}
|
||||
|
||||
await using var stream = await entry.OpenEntryStreamAsync();
|
||||
using var memory = new MemoryStream();
|
||||
await stream.CopyToAsync(memory);
|
||||
entries.Add(
|
||||
new EntrySnapshot(
|
||||
entry.Key ?? string.Empty,
|
||||
entry.Size,
|
||||
entry.CompressionType,
|
||||
Convert.ToBase64String(memory.ToArray())
|
||||
)
|
||||
);
|
||||
}
|
||||
|
||||
return entries.OrderBy(entry => entry.Key, StringComparer.Ordinal).ToList();
|
||||
}
|
||||
|
||||
private static List<EntrySnapshot> ReadReaderEntries(string archivePath)
|
||||
{
|
||||
using var stream = File.OpenRead(archivePath);
|
||||
using var reader = ReaderFactory.OpenReader(stream);
|
||||
var entries = new List<EntrySnapshot>();
|
||||
while (reader.MoveToNextEntry())
|
||||
{
|
||||
if (reader.Entry.IsDirectory)
|
||||
{
|
||||
continue;
|
||||
}
|
||||
|
||||
using var memory = new MemoryStream();
|
||||
reader.WriteEntryTo(memory);
|
||||
entries.Add(
|
||||
new EntrySnapshot(
|
||||
reader.Entry.Key ?? string.Empty,
|
||||
reader.Entry.Size,
|
||||
reader.Entry.CompressionType,
|
||||
Convert.ToBase64String(memory.ToArray())
|
||||
)
|
||||
);
|
||||
}
|
||||
|
||||
return entries.OrderBy(entry => entry.Key, StringComparer.Ordinal).ToList();
|
||||
}
|
||||
|
||||
private static async Task<List<EntrySnapshot>> ReadReaderEntriesAsync(string archivePath)
|
||||
{
|
||||
await using var stream = File.OpenRead(archivePath);
|
||||
await using var reader = await ReaderFactory.OpenAsyncReader(stream);
|
||||
var entries = new List<EntrySnapshot>();
|
||||
while (await reader.MoveToNextEntryAsync())
|
||||
{
|
||||
if (reader.Entry.IsDirectory)
|
||||
{
|
||||
continue;
|
||||
}
|
||||
|
||||
using var memory = new MemoryStream();
|
||||
await reader.WriteEntryToAsync(memory);
|
||||
entries.Add(
|
||||
new EntrySnapshot(
|
||||
reader.Entry.Key ?? string.Empty,
|
||||
reader.Entry.Size,
|
||||
reader.Entry.CompressionType,
|
||||
Convert.ToBase64String(memory.ToArray())
|
||||
)
|
||||
);
|
||||
}
|
||||
|
||||
return entries.OrderBy(entry => entry.Key, StringComparer.Ordinal).ToList();
|
||||
}
|
||||
|
||||
private static byte[] CreateLargeTarArchive()
|
||||
{
|
||||
using var stream = new MemoryStream();
|
||||
using (
|
||||
var writer = WriterFactory.OpenWriter(
|
||||
stream,
|
||||
ArchiveType.Tar,
|
||||
new WriterOptions(CompressionType.None)
|
||||
)
|
||||
)
|
||||
{
|
||||
writer.Write("large.bin", new MemoryStream(new byte[64 * 1024]));
|
||||
}
|
||||
return stream.ToArray();
|
||||
}
|
||||
|
||||
private sealed record EntrySnapshot(
|
||||
string Key,
|
||||
long Size,
|
||||
CompressionType CompressionType,
|
||||
string Content
|
||||
);
|
||||
|
||||
private sealed class CancelAfterBytesReadStream(
|
||||
Stream stream,
|
||||
CancellationTokenSource cancellationTokenSource,
|
||||
long cancelAfterBytes
|
||||
) : Stream
|
||||
{
|
||||
private long _bytesRead;
|
||||
|
||||
public override bool CanRead => stream.CanRead;
|
||||
public override bool CanSeek => stream.CanSeek;
|
||||
public override bool CanWrite => false;
|
||||
public override long Length => stream.Length;
|
||||
public override long Position
|
||||
{
|
||||
get => stream.Position;
|
||||
set => stream.Position = value;
|
||||
}
|
||||
|
||||
public override void Flush() => stream.Flush();
|
||||
|
||||
public override int Read(byte[] buffer, int offset, int count) =>
|
||||
throw new NotSupportedException("Use async reads for this test stream.");
|
||||
|
||||
public override async ValueTask<int> ReadAsync(
|
||||
Memory<byte> buffer,
|
||||
CancellationToken cancellationToken = default
|
||||
)
|
||||
{
|
||||
cancellationToken.ThrowIfCancellationRequested();
|
||||
var read = await stream.ReadAsync(buffer, cancellationToken).ConfigureAwait(false);
|
||||
_bytesRead = read;
|
||||
if (_bytesRead > cancelAfterBytes)
|
||||
{
|
||||
cancellationTokenSource.Cancel();
|
||||
cancellationToken.ThrowIfCancellationRequested();
|
||||
}
|
||||
|
||||
return read;
|
||||
}
|
||||
|
||||
public override Task<int> ReadAsync(
|
||||
byte[] buffer,
|
||||
int offset,
|
||||
int count,
|
||||
CancellationToken cancellationToken
|
||||
) => ReadAsync(buffer.AsMemory(offset, count), cancellationToken).AsTask();
|
||||
|
||||
public override long Seek(long offset, SeekOrigin origin) => stream.Seek(offset, origin);
|
||||
|
||||
public override void SetLength(long value) => throw new NotSupportedException();
|
||||
|
||||
public override void Write(byte[] buffer, int offset, int count) =>
|
||||
throw new NotSupportedException();
|
||||
}
|
||||
}
|
||||
#endif
|
||||
Reference in New Issue
Block a user