forked from LykosAI/StabilityMatrix
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathPriorityOrderer.cs
57 lines (48 loc) · 1.6 KB
/
PriorityOrderer.cs
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
using StabilityMatrix.UITests.Attributes;
using Xunit.Abstractions;
using Xunit.Sdk;
namespace StabilityMatrix.UITests;
public class PriorityOrderer : ITestCaseOrderer
{
public IEnumerable<TTestCase> OrderTestCases<TTestCase>(IEnumerable<TTestCase> testCases)
where TTestCase : ITestCase
{
var sortedMethods = new SortedDictionary<int, List<TTestCase>>();
foreach (var testCase in testCases)
{
var priority = 0;
foreach (
var attr in testCase.TestMethod.Method.GetCustomAttributes(
typeof(TestPriorityAttribute).AssemblyQualifiedName
)
)
{
priority = attr.GetNamedArgument<int>("Priority");
}
GetOrCreate(sortedMethods, priority).Add(testCase);
}
foreach (var list in sortedMethods.Keys.Select(priority => sortedMethods[priority]))
{
list.Sort(
(x, y) =>
StringComparer.OrdinalIgnoreCase.Compare(
x.TestMethod.Method.Name,
y.TestMethod.Method.Name
)
);
foreach (var testCase in list)
{
yield return testCase;
}
}
}
private static TValue GetOrCreate<TKey, TValue>(IDictionary<TKey, TValue> dictionary, TKey key)
where TValue : new()
{
if (dictionary.TryGetValue(key, out var result))
return result;
result = new TValue();
dictionary[key] = result;
return result;
}
}