ExpectantStrictMock-Moq

ExpectantStrictMock.cs

using Moq;
using Moq.Language;
using Moq.Language.Flow;
using System;
using System.Collections.Generic;
using System.Linq.Expressions;

namespace MoqHelper
{
    public class ExpectantStrictMock<T> : Mock<T> where T : class
    {
        private readonly List<Action> verifyFunctions;

        public ExpectantStrictMock() : base(MockBehavior.Strict)
        {
            verifyFunctions = new List<Action>();
        }

        public ISetup<T> SetupExpect(Expression<Action<T>> action, Times times)
        {
            verifyFunctions.Add(() => Verify(action, times));
            return Setup(action);
        }

        public ISetup<T> SetupExpect(Expression<Action<T>> action, Func<Times> times)
        {
            return SetupExpect(action, times());
        }

        public ISetup<T> SetupExpect(Expression<Action<T>> action)
        {
            return SetupExpect(action, Times.AtLeastOnce);
        }

        public ISetup<T, T2> SetupExpect<T2>(Expression<Func<T, T2>> action, Times times)
        {
            verifyFunctions.Add(() => Verify(action, times));
            return Setup(action);
        }

        public ISetup<T, T2> SetupExpect<T2>(Expression<Func<T, T2>> action, Func<Times> times)
        {
            return SetupExpect(action, times());
        }

        public ISetup<T, T2> SetupExpect<T2>(Expression<Func<T, T2>> action)
        {
            return SetupExpect(action, Times.AtLeastOnce);
        }

        public ISetupSequentialResult<T2> SetupSequenceExpect<T2>(Expression<Func<T, T2>> action, Times times)
        {
            verifyFunctions.Add(() => Verify(action, times));
            return this.SetupSequence(action);
        }

        public ISetupSequentialResult<T2> SetupSequenceExpect<T2>(Expression<Func<T, T2>> action, Func<Times> times)
        {
            return SetupSequenceExpect(action, times());
        }

        public ISetupSequentialResult<T2> SetupSequenceExpect<T2>(Expression<Func<T, T2>> action)
        {
            return SetupSequenceExpect(action, Times.AtLeastOnce);
        }

        public void VerifyAllExpectations()
        {
            foreach (var verifyFunction in verifyFunctions)
            {
                verifyFunction();
            }
        }
    }
}