sharpcompress/tests/SharpCompress.Test/Ace/AceReaderAsyncTests.cs

using System;
using System.Collections.Generic;
using System.IO;
using System.Linq;
using System.Threading.Tasks;
using SharpCompress.Common;
using SharpCompress.Readers;
using SharpCompress.Readers.Ace;
using SharpCompress.Test.Mocks;
using Xunit;

namespace SharpCompress.Test.Ace;

public class AceReaderAsyncTests : ReaderTests
{
    public AceReaderAsyncTests()
    {
        UseExtensionInsteadOfNameToVerify = true;
        UseCaseInsensitiveToVerify = true;
    }

    [Fact]
    public async ValueTask Ace_Uncompressed_Read_Async() =>
        await ReadAsync("Ace.store.ace", CompressionType.None);

    [Fact]
    public async ValueTask Ace_Encrypted_Read_Async()
    {
        var exception = await Assert.ThrowsAsync<CryptographicException>(() =>
            ReadAsync("Ace.encrypted.ace")
        );
    }

    [Theory]
    [InlineData("Ace.method1.ace", CompressionType.AceLZ77)]
    [InlineData("Ace.method1-solid.ace", CompressionType.AceLZ77)]
    [InlineData("Ace.method2.ace", CompressionType.AceLZ77)]
    [InlineData("Ace.method2-solid.ace", CompressionType.AceLZ77)]
    public async ValueTask Ace_Unsupported_ShouldThrow_Async(
        string fileName,
        CompressionType compressionType
    )
    {
        var exception = await Assert.ThrowsAsync<NotSupportedException>(() =>
            ReadAsync(fileName, compressionType)
        );
    }

    [Theory]
    [InlineData("Ace.store.largefile.ace", CompressionType.None)]
    public async ValueTask Ace_LargeFileTest_Read_Async(
        string fileName,
        CompressionType compressionType
    )
    {
        await ReadForBufferBoundaryCheckAsync(fileName, compressionType);
    }

    [Fact]
    public async ValueTask Ace_Multi_Reader_Async()
    {
        var exception = await Assert.ThrowsAsync<MultiVolumeExtractionException>(() =>
            DoMultiReaderAsync(
                new[] { "Ace.store.split.ace", "Ace.store.split.c01" },
                streams => AceReader.OpenAsyncReader(streams, null)
            )
        );
    }

    private async Task ReadAsync(string testArchive, CompressionType expectedCompression)
    {
        testArchive = Path.Combine(TEST_ARCHIVES_PATH, testArchive);
        using Stream stream = File.OpenRead(testArchive);
        await using var reader = await ReaderFactory.OpenAsyncReader(
            new AsyncOnlyStream(stream),
            new ReaderOptions()
        );
        while (await reader.MoveToNextEntryAsync())
        {
            if (!reader.Entry.IsDirectory)
            {
                Assert.Equal(expectedCompression, reader.Entry.CompressionType);
                await reader.WriteEntryToDirectoryAsync(
                    SCRATCH_FILES_PATH,
                    new ExtractionOptions { ExtractFullPath = true, Overwrite = true }
                );
            }
        }
        VerifyFiles();
    }

    private async Task ReadForBufferBoundaryCheckAsync(
        string testArchive,
        CompressionType expectedCompression
    )
    {
        testArchive = Path.Combine(TEST_ARCHIVES_PATH, testArchive);
        using Stream stream = File.OpenRead(testArchive);
        await using var reader = await ReaderFactory.OpenAsyncReader(
            new AsyncOnlyStream(stream),
            new ReaderOptions { LookForHeader = true }
        );
        while (await reader.MoveToNextEntryAsync())
        {
            if (!reader.Entry.IsDirectory)
            {
                Assert.Equal(expectedCompression, reader.Entry.CompressionType);
                await reader.WriteEntryToDirectoryAsync(
                    SCRATCH_FILES_PATH,
                    new ExtractionOptions { ExtractFullPath = true, Overwrite = true }
                );
            }
        }
        CompareFilesByPath(
            Path.Combine(SCRATCH_FILES_PATH, "alice29.txt"),
            Path.Combine(MISC_TEST_FILES_PATH, "alice29.txt")
        );
    }

    private async Task DoMultiReaderAsync(
        string[] archives,
        Func<IEnumerable<Stream>, IAsyncReader> readerFactory
    )
    {
        await using var reader = readerFactory(
            archives.Select(s => Path.Combine(TEST_ARCHIVES_PATH, s)).Select(File.OpenRead)
        );

        while (await reader.MoveToNextEntryAsync())
        {
            if (!reader.Entry.IsDirectory)
            {
                await reader.WriteEntryToDirectoryAsync(
                    SCRATCH_FILES_PATH,
                    new ExtractionOptions { ExtractFullPath = true, Overwrite = true }
                );
            }
        }
    }
}