ImageSharp/tests/ImageSharp.Tests/Processing/ImageProcessingContextTests.cs

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

using Moq;
using SixLabors.ImageSharp.PixelFormats;
using SixLabors.ImageSharp.Processing;
using SixLabors.ImageSharp.Processing.Processors;

namespace SixLabors.ImageSharp.Tests.Processing;

/// <summary>
/// Contains test cases for default <see cref="IImageProcessingContext"/> implementation.
/// </summary>
public class ImageProcessingContextTests : IDisposable
{
    private readonly Image image = new Image<Rgba32>(10, 10);

    private readonly Mock<IImageProcessor> processorDefinition;

    private readonly Mock<ICloningImageProcessor> cloningProcessorDefinition;

    private readonly Mock<IImageProcessor<Rgba32>> regularProcessorImpl;

    private readonly Mock<ICloningImageProcessor<Rgba32>> cloningProcessorImpl;

    private static readonly Rectangle Bounds = new(3, 3, 5, 5);

    public ImageProcessingContextTests()
    {
        this.processorDefinition = new Mock<IImageProcessor>();
        this.cloningProcessorDefinition = new Mock<ICloningImageProcessor>();
        this.regularProcessorImpl = new Mock<IImageProcessor<Rgba32>>();
        this.cloningProcessorImpl = new Mock<ICloningImageProcessor<Rgba32>>();
    }

    // bool throwException, bool useBounds
    public static readonly TheoryData<bool, bool> ProcessorTestData = new()
    {
        { false, false },
        { false, true },
        { true, false },
        { true, true }
    };

    [Theory]
    [MemberData(nameof(ProcessorTestData))]
    public void Mutate_RegularProcessor(bool throwException, bool useBounds)
    {
        this.SetupRegularProcessor(throwException);

        if (throwException)
        {
            Assert.Throws<ImageProcessingException>(() => this.MutateRegularApply(useBounds));
        }
        else
        {
            this.MutateRegularApply(useBounds);
        }

        this.regularProcessorImpl.Verify(p => p.Execute(), Times.Once());
        this.regularProcessorImpl.Verify(p => p.Dispose(), Times.Once());
    }

    [Theory]
    [MemberData(nameof(ProcessorTestData))]
    public void Clone_RegularProcessor(bool throwException, bool useBounds)
    {
        this.SetupRegularProcessor(throwException);

        if (throwException)
        {
            Assert.Throws<ImageProcessingException>(() => this.CloneRegularApply(useBounds));
        }
        else
        {
            this.CloneRegularApply(useBounds);
        }

        this.regularProcessorImpl.Verify(p => p.Execute(), Times.Once);
        this.regularProcessorImpl.Verify(p => p.Dispose(), Times.Once);
    }

    [Theory]
    [MemberData(nameof(ProcessorTestData))]
    public void Mutate_CloningProcessor(bool throwException, bool useBounds)
    {
        this.SetupCloningProcessor(throwException);

        if (throwException)
        {
            Assert.Throws<ImageProcessingException>(() => this.MutateCloneApply(useBounds));
        }
        else
        {
            this.MutateCloneApply(useBounds);
        }

        this.cloningProcessorImpl.Verify(p => p.Execute(), Times.Once());
        this.cloningProcessorImpl.Verify(p => p.Dispose(), Times.Once());
    }

    [Theory]
    [MemberData(nameof(ProcessorTestData))]
    public void Clone_CloningProcessor(bool throwException, bool useBounds)
    {
        this.SetupCloningProcessor(throwException);

        if (throwException)
        {
            Assert.Throws<ImageProcessingException>(() => this.CloneCloneApply(useBounds));
        }
        else
        {
            this.CloneCloneApply(useBounds);
        }

        this.cloningProcessorImpl.Verify(p => p.CloneAndExecute(), Times.Once());
        this.cloningProcessorImpl.Verify(p => p.Dispose(), Times.Once());
    }

    private void MutateRegularApply(bool useBounds)
    {
        if (useBounds)
        {
            this.image.Mutate(c => c.ApplyProcessor(this.processorDefinition.Object, Bounds));
        }
        else
        {
            this.image.Mutate(c => c.ApplyProcessor(this.processorDefinition.Object));
        }
    }

    private void MutateCloneApply(bool useBounds)
    {
        if (useBounds)
        {
            this.image.Mutate(c => c.ApplyProcessor(this.cloningProcessorDefinition.Object, Bounds));
        }
        else
        {
            this.image.Mutate(c => c.ApplyProcessor(this.cloningProcessorDefinition.Object));
        }
    }

    private void CloneRegularApply(bool useBounds)
    {
        if (useBounds)
        {
            this.image.Clone(c => c.ApplyProcessor(this.processorDefinition.Object, Bounds)).Dispose();
        }
        else
        {
            this.image.Clone(c => c.ApplyProcessor(this.processorDefinition.Object)).Dispose();
        }
    }

    private void CloneCloneApply(bool useBounds)
    {
        if (useBounds)
        {
            this.image.Clone(c => c.ApplyProcessor(this.cloningProcessorDefinition.Object, Bounds)).Dispose();
        }
        else
        {
            this.image.Clone(c => c.ApplyProcessor(this.cloningProcessorDefinition.Object)).Dispose();
        }
    }

    private void SetupRegularProcessor(bool throwsException)
    {
        if (throwsException)
        {
            this.regularProcessorImpl.Setup(p => p.Execute()).Throws(new ImageProcessingException("Test"));
        }

        this.processorDefinition
            .Setup(p => p.CreatePixelSpecificProcessor(Configuration.Default, It.IsAny<Image<Rgba32>>(), It.IsAny<Rectangle>()))
            .Returns(this.regularProcessorImpl.Object);
    }

    private void SetupCloningProcessor(bool throwsException)
    {
        if (throwsException)
        {
            this.cloningProcessorImpl.Setup(p => p.Execute()).Throws(new ImageProcessingException("Test"));
            this.cloningProcessorImpl.Setup(p => p.CloneAndExecute()).Throws(new ImageProcessingException("Test"));
        }

        this.cloningProcessorDefinition
            .Setup(p => p.CreatePixelSpecificCloningProcessor(Configuration.Default, It.IsAny<Image<Rgba32>>(), It.IsAny<Rectangle>()))
            .Returns(this.cloningProcessorImpl.Object);

        this.cloningProcessorDefinition
            .Setup(p => p.CreatePixelSpecificProcessor(Configuration.Default, It.IsAny<Image<Rgba32>>(), It.IsAny<Rectangle>()))
            .Returns(this.cloningProcessorImpl.Object);
    }

    public void Dispose() => this.image?.Dispose();
}