// Copyright (c) Six Labors.
// Licensed under the Apache License, Version 2.0.

using System;
using System.Diagnostics;
using System.Reflection;
using Xunit.Sdk;

namespace SixLabors.ImageSharp.Tests
{
    [AttributeUsage(AttributeTargets.Class | AttributeTargets.Method, AllowMultiple = false, Inherited = true)]
    public class ValidateDisposedMemoryAllocationsAttribute : BeforeAfterTestAttribute
    {
        private readonly int expected = 0;

        public ValidateDisposedMemoryAllocationsAttribute()
            : this(0)
        {
        }

        public ValidateDisposedMemoryAllocationsAttribute(int expected)
            => this.expected = expected;

        public override void Before(MethodInfo methodUnderTest)
            => MemoryAllocatorValidator.MonitorAllocations();

        public override void After(MethodInfo methodUnderTest)
        {
            MemoryAllocatorValidator.ValidateAllocations(this.expected);
            MemoryAllocatorValidator.StopMonitoringAllocations();
        }
    }
}