ImageSharp/tests/ImageSharp.Tests/Image/ImageTests.Decode_Cancellation.cs

// Copyright (c) Six Labors.
// Licensed under the Six Labors Split License.

using SixLabors.ImageSharp.Formats;
using SixLabors.ImageSharp.Tests.TestUtilities;

namespace SixLabors.ImageSharp.Tests;

public partial class ImageTests
{
    public class Decode_Cancellation : ImageLoadTestBase
    {
        private bool isTestStreamSeekable;
        private readonly SemaphoreSlim notifyWaitPositionReachedSemaphore = new(0);
        private readonly SemaphoreSlim continueSemaphore = new(0);

        public Decode_Cancellation() => this.TopLevelConfiguration.StreamProcessingBufferSize = 128;

        public static readonly TheoryData<string> TestFiles = new()
        {
            TestImages.Png.BikeSmall,
            TestImages.Jpeg.Baseline.Jpeg420Small,
            TestImages.Bmp.Car,
            TestImages.Tiff.RgbUncompressed,
            TestImages.Gif.Kumin,
            TestImages.Tga.Bit32PalRleBottomLeft,
            TestImages.Webp.TestPatternOpaqueSmall,
            TestImages.Pbm.RgbPlainMagick
        };

        [Theory]
        [MemberData(nameof(TestFiles))]
        public async Task IdentifyAsync_IsCancellable(string file)
        {
            CancellationTokenSource cts = new();
            string path = Path.Combine(TestEnvironment.InputImagesDirectoryFullPath, file);
            using PausedStream pausedStream = new(path);
            pausedStream.OnWaiting(_ =>
            {
                cts.Cancel();
                pausedStream.Release();
            });

            Configuration configuration = Configuration.CreateDefaultInstance();
            configuration.FileSystem = new SingleStreamFileSystem(pausedStream);
            DecoderOptions options = new()
            {
                Configuration = configuration
            };

            await Assert.ThrowsAsync<TaskCanceledException>(async () => await Image.IdentifyAsync(options, "someFakeFile", cts.Token));
        }

        [Theory]
        [MemberData(nameof(TestFiles))]
        public async Task DecodeAsync_IsCancellable(string file)
        {
            CancellationTokenSource cts = new();
            string path = Path.Combine(TestEnvironment.InputImagesDirectoryFullPath, file);
            using PausedStream pausedStream = new(path);
            pausedStream.OnWaiting(_ =>
            {
                cts.Cancel();
                pausedStream.Release();
            });

            Configuration configuration = Configuration.CreateDefaultInstance();
            configuration.FileSystem = new SingleStreamFileSystem(pausedStream);
            DecoderOptions options = new()
            {
                Configuration = configuration
            };

            await Assert.ThrowsAsync<TaskCanceledException>(async () =>
            {
                using Image image = await Image.LoadAsync(options, "someFakeFile", cts.Token);
            });
        }

        protected override Stream CreateStream() => this.TestFormat.CreateAsyncSemaphoreStream(this.notifyWaitPositionReachedSemaphore, this.continueSemaphore, this.isTestStreamSeekable);
    }
}