Skip to content

Commit 0a41146

Browse files
terencefanterencefan
committed
connection reroute test cases
1 parent 2b604b3 commit 0a41146

31 files changed

+984
-480
lines changed

‎.editorconfig‎

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -51,7 +51,7 @@ csharp_new_line_before_members_in_object_initializers = true
5151
csharp_new_line_before_members_in_anonymous_types = true
5252

5353
# Namespace settings
54-
csharp_style_namespace_declarations = file_scoped:silent
54+
csharp_style_namespace_declarations = file_scoped:error
5555

5656
# Brace settings
5757
csharp_prefer_braces = true:silent# Prefer curly braces even for one line of code
@@ -102,6 +102,7 @@ csharp_prefer_simple_default_expression = true:suggestion
102102
csharp_style_prefer_local_over_anonymous_function = true:suggestion
103103
csharp_style_prefer_index_operator = true:suggestion
104104
csharp_style_prefer_range_operator = true:suggestion
105+
csharp_space_around_binary_operators = before_and_after
105106

106107
[*.{xml,config,*proj,nuspec,props,resx,targets,yml,tasks}]
107108
indent_size = 2

‎src/Microsoft.Azure.SignalR.Common/DefaultHeaderProvider.cs‎

Lines changed: 11 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,11 @@
1+
// Copyright (c) Microsoft. All rights reserved.
2+
// Licensed under the MIT license. See LICENSE file in the project root for full license information.
3+
4+
using System.Collections.Generic;
5+
6+
namespace Microsoft.Azure.SignalR;
7+
8+
internal class DefaultHeaderProvider : ICustomHeaderProvider
9+
{
10+
public IEnumerable<(string key, string val)> Headers => [];
11+
}

‎src/Microsoft.Azure.SignalR.Common/Interfaces/ICustomHeaderProvider.cs‎

Lines changed: 11 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,11 @@
1+
// Copyright (c) Microsoft. All rights reserved.
2+
// Licensed under the MIT license. See LICENSE file in the project root for full license information.
3+
4+
using System.Collections.Generic;
5+
6+
namespace Microsoft.Azure.SignalR;
7+
8+
internal interface ICustomHeaderProvider
9+
{
10+
IEnumerable<(string key, string val)> Headers { get; }
11+
}

‎test/Microsoft.Azure.SignalR.E2ETests/Common/ITestHubConnection.cs‎

Lines changed: 29 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,29 @@
1+
// Copyright (c) Microsoft. All rights reserved.
2+
// Licensed under the MIT license. See LICENSE file in the project root for full license information.
3+
4+
using System.Threading.Tasks;
5+
6+
namespace Microsoft.Azure.SignalR.E2ETests.Common;
7+
8+
public interface ITestHubConnection
9+
{
10+
public string ConnectionId { get; }
11+
12+
public string User { get; }
13+
14+
public int MessageCount { get; }
15+
16+
public void Listen(params string[] methods);
17+
18+
public Task StartAsync();
19+
20+
public Task StopAsync();
21+
22+
public Task SendAsync(string method, params string[] messages);
23+
24+
public void ResetInvoke(string method);
25+
26+
public Task ExpectInvokeAsync(string method, params string[] message);
27+
28+
public void ResetMessageCount();
29+
}

‎test/Microsoft.Azure.SignalR.E2ETests/Common/ITestHubConnectionGroup.cs‎

Lines changed: 10 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,10 @@
1+
// Copyright (c) Microsoft. All rights reserved.
2+
// Licensed under the MIT license. See LICENSE file in the project root for full license information.
3+
4+
using System.Collections.Generic;
5+
6+
namespace Microsoft.Azure.SignalR.E2ETests.Common;
7+
8+
public interface ITestHubConnectionGroup : ITestHubConnection, IEnumerable<ITestHubConnection>
9+
{
10+
}

‎test/Microsoft.Azure.SignalR.E2ETests/Common/ITestServer.cs‎

Lines changed: 15 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,15 @@
1+
// Copyright (c) Microsoft. All rights reserved.
2+
// Licensed under the MIT license. See LICENSE file in the project root for full license information.
3+
4+
using System;
5+
using System.Collections.Generic;
6+
using System.Threading.Tasks;
7+
8+
namespace Microsoft.Azure.SignalR.E2ETests.Common;
9+
10+
public interface ITestServer : IDisposable
11+
{
12+
Task<string> StartAsync(Dictionary<string, string?>? configuration = null);
13+
14+
Task StopAsync();
15+
}

‎test/Microsoft.Azure.SignalR.E2ETests/Common/IngressHeaderProvider.cs‎

Lines changed: 29 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,29 @@
1+
// Copyright (c) Microsoft. All rights reserved.
2+
// Licensed under the MIT license. See LICENSE file in the project root for full license information.
3+
4+
using System;
5+
using System.Collections.Generic;
6+
7+
namespace Microsoft.Azure.SignalR.E2ETests.Common;
8+
9+
internal sealed class IngressHeaderProvider : ICustomHeaderProvider
10+
{
11+
public const string AsrsInternalForwardedBy = "X-ASRS-Internal-Forwarded-By";
12+
13+
public const string PodA = "podA";
14+
15+
public const string PodB = "podB";
16+
17+
private static readonly string[] Ingresses = [PodA, PodB];
18+
19+
private static readonly Random Random = new();
20+
21+
public IEnumerable<(string key, string val)> Headers
22+
{
23+
get
24+
{
25+
var podName = Ingresses[Random.Next() % Ingresses.Length];
26+
yield return (AsrsInternalForwardedBy, podName);
27+
}
28+
}
29+
}

‎test/Microsoft.Azure.SignalR.E2ETests/Common/TestHubConnectionFactory.cs‎

Lines changed: 195 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,195 @@
1+
// Copyright (c) Microsoft. All rights reserved.
2+
// Licensed under the MIT license. See LICENSE file in the project root for full license information.
3+
4+
using System;
5+
using System.Collections;
6+
using System.Collections.Concurrent;
7+
using System.Collections.Generic;
8+
using System.Diagnostics.CodeAnalysis;
9+
using System.Linq;
10+
using System.Threading;
11+
using System.Threading.Tasks;
12+
13+
using Microsoft.AspNetCore.SignalR.Client;
14+
15+
using Xunit;
16+
17+
namespace Microsoft.Azure.SignalR.E2ETests.Common;
18+
19+
public class TestHubConnectionFactory
20+
{
21+
public bool EnableStatefulReconnect { get; set; }
22+
23+
public ITestHubConnection NewConnection(string host, string? hub = null, string userId = "foo")
24+
{
25+
return new TestHubConnection(host, hub)
26+
{
27+
User = userId,
28+
EnableStatefulReconnect = EnableStatefulReconnect
29+
};
30+
}
31+
32+
public ITestHubConnectionGroup NewConnectionGroup(string host, int count, string? hub = null, string userId = "foo")
33+
{
34+
return new TestHubConnectionGroup(host, count, hub)
35+
{
36+
User = userId,
37+
EnableStatefulReconnect = EnableStatefulReconnect
38+
};
39+
}
40+
41+
private sealed class TestHubConnection(string host, string? hub = null) : ITestHubConnection
42+
{
43+
private readonly ConcurrentDictionary<string, TaskCompletionSource<string[]>> _expectedInvokes = new();
44+
45+
private HubConnection? _hubConnection;
46+
47+
private volatile int _messageCount;
48+
49+
public string User { get; init; } = "foo";
50+
51+
public int MessageCount => _messageCount;
52+
53+
public bool EnableStatefulReconnect { get; init; }
54+
55+
public string ConnectionId => _hubConnection?.ConnectionId ?? throw NotReady;
56+
57+
private static Exception NotReady { get; } = new InvalidOperationException("HubConnection is not in connected state.");
58+
59+
public Task StartAsync()
60+
{
61+
BuildConnectionIfNull();
62+
return _hubConnection.StartAsync();
63+
}
64+
65+
public Task StopAsync() => _hubConnection?.StopAsync() ?? Task.CompletedTask;
66+
67+
public Task SendAsync(string method, params string[] messages)
68+
{
69+
if (_hubConnection == null || _hubConnection.State != HubConnectionState.Connected)
70+
{
71+
throw NotReady;
72+
}
73+
return _hubConnection.SendCoreAsync(method, messages);
74+
}
75+
76+
public void Listen(params string[] methods)
77+
{
78+
BuildConnectionIfNull();
79+
foreach (var method in methods)
80+
{
81+
_expectedInvokes.TryAdd(method, new TaskCompletionSource<string[]>());
82+
_hubConnection.On(method, (Action<string>)(message => Invoke(method, message)));
83+
}
84+
}
85+
86+
public void ResetInvoke(string method)
87+
{
88+
_expectedInvokes.AddOrUpdate(method,
89+
new TaskCompletionSource<string[]>(),
90+
(method, ov) => ov.Task.IsCompleted ? new TaskCompletionSource<string[]>() : ov);
91+
}
92+
93+
public async Task ExpectInvokeAsync(string method, params string[] messages)
94+
{
95+
Assert.Equal(messages, await _expectedInvokes[method].Task);
96+
}
97+
98+
public void ResetMessageCount()
99+
{
100+
_messageCount = 0;
101+
}
102+
103+
private void Invoke(string method, params string[] messages)
104+
{
105+
Interlocked.Increment(ref _messageCount);
106+
if (_expectedInvokes.TryGetValue(method, out var source))
107+
{
108+
source.TrySetResult(messages);
109+
}
110+
}
111+
112+
[MemberNotNull(nameof(_hubConnection))]
113+
private void BuildConnectionIfNull()
114+
{
115+
if (_hubConnection == null)
116+
{
117+
hub ??= nameof(TestHub);
118+
var url = $"{host}/{hub}?user={User}";
119+
var builder = new HubConnectionBuilder().WithUrl(url);
120+
121+
if (EnableStatefulReconnect)
122+
{
123+
builder = builder.WithStatefulReconnect();
124+
}
125+
_hubConnection = builder.Build();
126+
}
127+
}
128+
}
129+
130+
private sealed class TestHubConnectionGroup(string host, int count, string? hub = null) : ITestHubConnectionGroup
131+
{
132+
private List<ITestHubConnection>? _connections;
133+
134+
public bool EnableStatefulReconnect { get; init; }
135+
136+
public IEnumerable<ITestHubConnection> Connections
137+
{
138+
get
139+
{
140+
_connections ??= (from i in Enumerable.Range(0, count)
141+
select new TestHubConnection(host, hub)
142+
{
143+
User = User,
144+
EnableStatefulReconnect = EnableStatefulReconnect,
145+
} as ITestHubConnection).ToList();
146+
return _connections;
147+
}
148+
}
149+
150+
public string User { get; init; } = string.Empty;
151+
152+
public int MessageCount => Connections.Select(x => x.MessageCount).Sum();
153+
154+
public string ConnectionId => throw new NotImplementedException("Connection group does not have ConnectionId.");
155+
156+
public void Listen(params string[] methods)
157+
{
158+
foreach (var connection in Connections)
159+
{
160+
connection.Listen(methods);
161+
}
162+
}
163+
164+
public async Task StartAsync() => await Task.WhenAll(Connections.Select(x => x.StartAsync()));
165+
166+
public IEnumerator<ITestHubConnection> GetEnumerator() => Connections.GetEnumerator();
167+
168+
IEnumerator IEnumerable.GetEnumerator() => GetEnumerator();
169+
170+
public async Task StopAsync() => await Task.WhenAll(Connections.Select(x => x.StopAsync()));
171+
172+
public void ResetInvoke(string method)
173+
{
174+
foreach (var connection in Connections)
175+
{
176+
connection.ResetInvoke(method);
177+
}
178+
}
179+
180+
public Task ExpectInvokeAsync(string method, params string[] messages)
181+
{
182+
return Task.WhenAll(Connections.Select(x => x.ExpectInvokeAsync(method, messages)));
183+
}
184+
185+
public Task SendAsync(string method, params string[] messages) => Task.WhenAll(Connections.Select(x => x.SendAsync(method, messages)));
186+
187+
public void ResetMessageCount()
188+
{
189+
foreach (var connection in Connections)
190+
{
191+
connection.ResetMessageCount();
192+
}
193+
}
194+
}
195+
}

0 commit comments

Comments
 (0)