123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960 |
- using System;
- using System.Collections;
- using System.Reflection;
- using NUnit.Framework;
- using NUnit.Framework.Interfaces;
- using NUnit.Framework.Internal;
- using NUnit.Framework.Internal.Commands;
- using UnityEngine.TestRunner.NUnitExtensions.Runner;
- namespace UnityEngine.TestTools
- {
- internal class EnumerableRepeatedTestCommand : DelegatingTestCommand, IEnumerableTestMethodCommand
- {
- private int repeatCount;
-
- public EnumerableRepeatedTestCommand(RepeatAttribute.RepeatedTestCommand commandToReplace) : base(commandToReplace.GetInnerCommand())
- {
- repeatCount = (int) typeof(RepeatAttribute.RepeatedTestCommand)
- .GetField("repeatCount", BindingFlags.NonPublic | BindingFlags.Instance)
- .GetValue(commandToReplace);
- }
- public override TestResult Execute(ITestExecutionContext context)
- {
- throw new NotImplementedException("Use ExecuteEnumerable");
- }
- public IEnumerable ExecuteEnumerable(ITestExecutionContext context)
- {
- var unityContext = (UnityTestExecutionContext)context;
- int count = unityContext.EnumerableRepeatedTestState;
- while (count < repeatCount)
- {
- count++;
- unityContext.EnumerableRepeatedTestState = count;
-
- if (innerCommand is IEnumerableTestMethodCommand)
- {
- var executeEnumerable = ((IEnumerableTestMethodCommand)innerCommand).ExecuteEnumerable(context);
- foreach (var iterator in executeEnumerable)
- {
- yield return iterator;
- }
- }
- else
- {
- context.CurrentResult = innerCommand.Execute(context);
- }
- if (context.CurrentResult.ResultState != ResultState.Success)
- {
- break;
- }
- }
- unityContext.EnumerableRepeatedTestState = 0;
- }
- }
- }
|