logos-storage-nim-cs-dist-tests/Tests/LogosStorageContinuousTests/TestHandle.cs

using System.Reflection;

namespace ContinuousTests
{
    public class TestHandle
    {
        private readonly List<MethodMoment> moments = new List<MethodMoment>();

        public TestHandle(ContinuousTest test)
        {
            Test = test;

            ReflectTestMoments();

            var testName = test.GetType().Name;
            if (!moments.Any()) throw new Exception($"Test '{testName}' has no moments.");
            if (moments.Count != moments.Select(m => m.Moment).Distinct().Count()) throw new Exception($"Test '{testName}' has duplicate moments");
        }

        public ContinuousTest Test { get; }

        public int GetEarliestMoment()
        {
            return moments.Min(m => m.Moment);
        }

        public int GetLastMoment()
        {
            return moments.Max(m => m.Moment);
        }

        public int? GetNextMoment(int currentMoment)
        {
            var remainingMoments = moments.Where(m => m.Moment > currentMoment).ToArray();
            if (!remainingMoments.Any()) return null;
            return remainingMoments.Min(m => m.Moment);
        }

        public void InvokeMoment(int currentMoment, Action<string> beforeInvoke)
        {
            var moment = moments.SingleOrDefault(m => m.Moment == currentMoment);
            if (moment == null) return;

            lock (MomentLock.Lock)
            {
                beforeInvoke(moment.Method.Name);
                moment.Method.Invoke(Test, Array.Empty<object>());
            }
        }

        private void ReflectTestMoments()
        {
            var methods = Test.GetType().GetMethods()
                      .Where(m => m.GetCustomAttributes(typeof(TestMomentAttribute), false).Length > 0)
                      .ToArray();

            foreach (var method in methods)
            {
                var moment = method.GetCustomAttribute<TestMomentAttribute>();
                if (moment != null && moment.T >= 0)
                {
                    moments.Add(new MethodMoment(method, moment.T));
                }
            }
        }
    }

    public class MethodMoment
    {
        public MethodMoment(MethodInfo method, int moment)
        {
            Method = method;
            Moment = moment;

            if (moment < 0) throw new Exception("Moment must be zero or greater.");
        }

        public MethodInfo Method { get; }
        public int Moment { get; }
    }

    public static class MomentLock
    {
        public static readonly object Lock = new();
    }
}