GitHubActionsTestLogger/GitHubActionsTestLogger.Tests/Mtp/TestNodeBuilder.cs at 17598b61b4055a6f7eb81c76253e687f253beb9b · Tyrrrz/GitHubActionsTestLogger · GitHub

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
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
using System;
using GitHubActionsTestLogger.Tests.Utils.Extensions;
using Microsoft.Testing.Platform.Extensions.Messages;

namespace GitHubActionsTestLogger.Tests.Mtp;

internal class TestNodeBuilder
{
    private string _displayName = "FakeTest";
    private string _assemblyName = "FakeTests.dll";
    private string _namespace = "FakeTests";
    private string _typeName = "FakeTestClass";
    private string _methodName = "FakeTestMethod";
    private string? _sourceFilePath;
    private int? _sourceLineNumber;
    private TestOutcome _testOutcome;
    private string? _errorMessage;
    private string? _errorStackTrace;

    public TestNodeBuilder SetDisplayName(string displayName)
    {
        _displayName = displayName;
        return this;
    }

    public TestNodeBuilder SetAssemblyName(string assemblyName)
    {
        _assemblyName = assemblyName;
        return this;
    }

    public TestNodeBuilder SetNamespace(string @namespace)
    {
        _namespace = @namespace;
        return this;
    }

    public TestNodeBuilder SetTypeName(string typeName)
    {
        _typeName = typeName;
        return this;
    }

    public TestNodeBuilder SetMethodName(string methodName)
    {
        _methodName = methodName;
        return this;
    }

    public TestNodeBuilder SetSourceFilePath(string sourceFilePath)
    {
        _sourceFilePath = sourceFilePath;
        return this;
    }

    public TestNodeBuilder SetSourceLineNumber(int sourceLineNumber)
    {
        _sourceLineNumber = sourceLineNumber;
        return this;
    }

    public TestNodeBuilder SetOutcome(TestOutcome testOutcome)
    {
        _testOutcome = testOutcome;
        return this;
    }

    public TestNodeBuilder SetErrorMessage(string errorMessage)
    {
        _errorMessage = errorMessage;
        return this;
    }

    public TestNodeBuilder SetErrorStackTrace(string errorStackTrace)
    {
        _errorStackTrace = errorStackTrace;
        return this;
    }

    public TestNode Build()
    {
        var properties = new PropertyBag();

        // State
        properties.Add(
            _testOutcome switch
            {
                TestOutcome.None => DiscoveredTestNodeStateProperty.CachedInstance,
                TestOutcome.Passed => PassedTestNodeStateProperty.CachedInstance,
                TestOutcome.Failed => !string.IsNullOrWhiteSpace(_errorStackTrace)
                    ? new FailedTestNodeStateProperty(
                        new Exception(_errorMessage).ReplaceStackTrace(_errorStackTrace),
                        _errorMessage
                    )
                    : new FailedTestNodeStateProperty(_errorMessage ?? "Test failed."),
                TestOutcome.Skipped => SkippedTestNodeStateProperty.CachedInstance,
                _ => DiscoveredTestNodeStateProperty.CachedInstance,
            }
        );

        // Method info
        properties.Add(
            new TestMethodIdentifierProperty(
                _assemblyName,
                _namespace,
                _typeName,
                _methodName,
                0,
                [],
                "void"
            )
        );

        // Location info
        if (_sourceFilePath is not null && _sourceLineNumber is not null)
        {
            properties.Add(
                new TestFileLocationProperty(
                    _sourceFilePath,
                    new LinePositionSpan(
                        new LinePosition(_sourceLineNumber.Value, 0),
                        new LinePosition(_sourceLineNumber.Value, 0)
                    )
                )
            );
        }

        return new TestNode
        {
            Uid = Guid.NewGuid().ToString(),
            DisplayName = _displayName,
            Properties = properties,
        };
    }
}