예제 #1
0
        /// <summary>
        /// Gets a clone of the original mock for verification purposes.
        /// </summary>
        /// <param name="mock">The mock to be cloned.</param>
        /// <param name="notCalled">Whether to add a behavior that verifies the invocations performed on
        /// the clone were never performed on the original mock.
        /// </param>
        static T GetVerifier <T>(IMock <T> mock, bool notCalled = false) where T : class
        {
            // If the mock is already being verified, we don't need to clone again.
            if (mock.State.TryGetValue <bool>(typeof(Verify), out var verifying) && verifying)
            {
                return(mock.Object);
            }

            // Otherwise, we create a verification copy that does not record invocations
            // and has default behavior.
            var clone = mock.Clone();

            var recording = clone.Behaviors.OfType <MockRecordingBehavior>().FirstOrDefault();

            if (recording != null)
            {
                clone.Behaviors.Remove(recording);
            }

            if (notCalled)
            {
                clone.Behaviors.Insert(
                    clone.Behaviors.IndexOf(clone.Behaviors.OfType <MockContextBehavior>().First()) + 1,
                    new NotCalledBehavior());
            }

            // Sets up the right behaviors for a loose mock
            new Moq <T>(clone).Behavior = MockBehavior.Loose;

            clone.State.Set(typeof(Verify), true);

            return(clone.Object);
        }
예제 #2
0
파일: Verify.cs 프로젝트: weitzhandler/moq
        /// <summary>
        /// Gets a clone of the original mock for verification purposes.
        /// </summary>
        /// <param name="mock">The mock to be cloned.</param>
        /// <param name="notCalled">Whether to add a behavior that verifies the invocations performed on
        /// the clone were never performed on the original mock.
        /// </param>
        static T GetVerifier <T>(IMock <T> mock, bool notCalled = false) where T : class
        {
            // If the mock is already being verified, we don't need to clone again.
            if (mock.State.TryGetValue <bool>(typeof(Verify), out var verifying) && verifying)
            {
                return(mock.Object);
            }

            // Otherwise, we create a verification copy that does not record invocations
            // and has default behavior.
            var clone = mock.Clone();

            var recording = clone.Behaviors.OfType <MockRecordingBehavior>().FirstOrDefault();

            if (recording != null)
            {
                clone.Behaviors.Remove(recording);
            }

            // The target replacer is needed so that whenever we try to get the target object
            // and the IMock from it for occurrence verification, we actually get to the actual
            // target being verified, not the cloned mock. Otherwise, invocations won't match
            // with the setups, since the target would be different.
            clone.Behaviors.Insert(
                clone.Behaviors.IndexOf(clone.Behaviors.OfType <MockContextBehavior>().First()) + 1,
                new TargetReplacerBehavior(mock.Object));

            if (notCalled)
            {
                clone.Behaviors.Insert(
                    clone.Behaviors.IndexOf(clone.Behaviors.OfType <TargetReplacerBehavior>().First()) + 1,
                    new NotCalledBehavior());
            }

            // Sets up the right behaviors for a loose mock
            new Moq <T>(clone).Behavior = MockBehavior.Loose;

            clone.State.Set(typeof(Verify), true);

            return(clone.Object);
        }