RunnerSanta2019/Assets/Core/UnityTestTools/Assertions/Editor/GroupByExecutionMethodRenderer.cs

using System;
using System.Collections.Generic;
using System.Linq;

namespace UnityTest
{
	public class GroupByExecutionMethodRenderer : AssertionListRenderer<AssertionComponent.CheckMethod>
	{
		protected override IEnumerable<IGrouping<AssertionComponent.CheckMethod, AssertionComponent>> GroupResult (IEnumerable<AssertionComponent> assertionComponents)
		{
			var enumVals = Enum.GetValues (typeof (AssertionComponent.CheckMethod)).Cast<AssertionComponent.CheckMethod> ();
			var pairs = new List<CheckFunctionAssertionPair> ();

			foreach (var checkMethod in enumVals)
			{
				var components = assertionComponents.Where (c => (c.checkMethods & checkMethod) == checkMethod);
				var componentPairs = components.Select ((a) => new CheckFunctionAssertionPair () {checkMethod = checkMethod, assertionComponent = a});
				pairs.AddRange (componentPairs);
			}
			return pairs.GroupBy (pair => pair.checkMethod,
								pair => pair.assertionComponent);
		}

		private class CheckFunctionAssertionPair
		{
			public AssertionComponent assertionComponent;
			public AssertionComponent.CheckMethod checkMethod;
		}
	}
}