runner/src/Test/L0/Sdk/RSWebApi/AgentJobRequestMessageL0.cs at 4796916d6a6a53528566e6c8b2efa31bd02465a3 · actions/runner · 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
using System.Collections.Generic;
using System.IO;
using System.Runtime.Serialization.Json;
using System.Text;
using Xunit;
using GitHub.DistributedTask.Pipelines;

namespace GitHub.Actions.RunService.WebApi.Tests;

public sealed class AgentJobRequestMessageL0
{
    [Fact]
    [Trait("Level", "L0")]
    [Trait("Category", "Common")]
    public void VerifyEnableDebuggerDeserialization_WithTrue()
    {
        // Arrange
        var serializer = new DataContractJsonSerializer(typeof(AgentJobRequestMessage));
        string jsonWithEnabledDebugger = DoubleQuotify("{'EnableDebugger': true}");

        // Act
        using var stream = new MemoryStream();
        stream.Write(Encoding.UTF8.GetBytes(jsonWithEnabledDebugger));
        stream.Position = 0;
        var recoveredMessage = serializer.ReadObject(stream) as AgentJobRequestMessage;

        // Assert
        Assert.NotNull(recoveredMessage);
        Assert.True(recoveredMessage.EnableDebugger, "EnableDebugger should be true when JSON contains 'EnableDebugger': true");
    }

    [Fact]
    [Trait("Level", "L0")]
    [Trait("Category", "Common")]
    public void VerifyEnableDebuggerDeserialization_DefaultToFalse()
    {
        // Arrange
        var serializer = new DataContractJsonSerializer(typeof(AgentJobRequestMessage));
        string jsonWithoutDebugger = DoubleQuotify("{'messageType': 'PipelineAgentJobRequest'}");

        // Act
        using var stream = new MemoryStream();
        stream.Write(Encoding.UTF8.GetBytes(jsonWithoutDebugger));
        stream.Position = 0;
        var recoveredMessage = serializer.ReadObject(stream) as AgentJobRequestMessage;

        // Assert
        Assert.NotNull(recoveredMessage);
        Assert.False(recoveredMessage.EnableDebugger, "EnableDebugger should default to false when JSON field is absent");
    }

    [Fact]
    [Trait("Level", "L0")]
    [Trait("Category", "Common")]
    public void VerifyEnableDebuggerDeserialization_WithFalse()
    {
        // Arrange
        var serializer = new DataContractJsonSerializer(typeof(AgentJobRequestMessage));
        string jsonWithDisabledDebugger = DoubleQuotify("{'EnableDebugger': false}");

        // Act
        using var stream = new MemoryStream();
        stream.Write(Encoding.UTF8.GetBytes(jsonWithDisabledDebugger));
        stream.Position = 0;
        var recoveredMessage = serializer.ReadObject(stream) as AgentJobRequestMessage;

        // Assert
        Assert.NotNull(recoveredMessage);
        Assert.False(recoveredMessage.EnableDebugger, "EnableDebugger should be false when JSON contains 'EnableDebugger': false");
    }

    [Fact]
    [Trait("Level", "L0")]
    [Trait("Category", "Common")]
    public void VerifyDebuggerTunnelDeserialization_WithTunnel()
    {
        // Arrange
        var serializer = new DataContractJsonSerializer(typeof(AgentJobRequestMessage), new DataContractJsonSerializerSettings
        {
            KnownTypes = new[] { typeof(DebuggerTunnelInfo) }
        });
        string json = DoubleQuotify(
            "{'EnableDebugger': true, 'DebuggerTunnel': {'TunnelId': 'tun-123', 'ClusterId': 'use2', 'HostToken': 'tok-abc', 'Port': 4711}}");

        // Act
        using var stream = new MemoryStream();
        stream.Write(Encoding.UTF8.GetBytes(json));
        stream.Position = 0;
        var recoveredMessage = serializer.ReadObject(stream) as AgentJobRequestMessage;

        // Assert
        Assert.NotNull(recoveredMessage);
        Assert.True(recoveredMessage.EnableDebugger);
        Assert.NotNull(recoveredMessage.DebuggerTunnel);
        Assert.Equal("tun-123", recoveredMessage.DebuggerTunnel.TunnelId);
        Assert.Equal("use2", recoveredMessage.DebuggerTunnel.ClusterId);
        Assert.Equal("tok-abc", recoveredMessage.DebuggerTunnel.HostToken);
        Assert.Equal(4711, recoveredMessage.DebuggerTunnel.Port);
    }

    [Fact]
    [Trait("Level", "L0")]
    [Trait("Category", "Common")]
    public void VerifyDebuggerTunnelDeserialization_WithoutTunnel()
    {
        // Arrange
        var serializer = new DataContractJsonSerializer(typeof(AgentJobRequestMessage));
        string json = DoubleQuotify("{'EnableDebugger': true}");

        // Act
        using var stream = new MemoryStream();
        stream.Write(Encoding.UTF8.GetBytes(json));
        stream.Position = 0;
        var recoveredMessage = serializer.ReadObject(stream) as AgentJobRequestMessage;

        // Assert
        Assert.NotNull(recoveredMessage);
        Assert.True(recoveredMessage.EnableDebugger);
        Assert.Null(recoveredMessage.DebuggerTunnel);
    }

    private static string DoubleQuotify(string text)
    {
        return text.Replace('\'', '"');
    }
}