tests: begin the march to test coverage
This commit is contained in:
Stevan Freeborn
@@ -138,252 +138,6 @@ public sealed class DiscordGatewayClientTests : IDisposable
|
||||
);
|
||||
}
|
||||
|
||||
[Fact]
|
||||
public async Task ConnectAsync_WhenCalledAndWebSocketIsClosed_ItShouldStopReceivingMessages()
|
||||
{
|
||||
_mockDiscordRestClient
|
||||
.Setup(static x => x.GetGatewayUrlAsync(It.IsAny<CancellationToken>()))
|
||||
.ReturnsAsync("wss://gateway.discord.gg");
|
||||
|
||||
var mockWebSocket = new Mock<IWebSocket>();
|
||||
|
||||
mockWebSocket
|
||||
.Setup(static x => x.State)
|
||||
.Returns(WebSocketState.Closed);
|
||||
|
||||
_mockWebSocketFactory
|
||||
.Setup(static x => x.Create())
|
||||
.Returns(mockWebSocket.Object);
|
||||
|
||||
await _discordGatewayClient.ConnectAsync(CancellationToken.None);
|
||||
|
||||
mockWebSocket.Verify(
|
||||
static x => x.ReceiveAsync(It.IsAny<ArraySegment<byte>>(), It.IsAny<CancellationToken>()),
|
||||
Times.Never
|
||||
);
|
||||
}
|
||||
|
||||
[Fact]
|
||||
public async Task ConnectAsync_WhenCalledAndCancellationIsRequested_ItShouldStopReceivingMessages()
|
||||
{
|
||||
_mockDiscordRestClient
|
||||
.Setup(static x => x.GetGatewayUrlAsync(It.IsAny<CancellationToken>()))
|
||||
.ReturnsAsync("wss://gateway.discord.gg");
|
||||
|
||||
var mockWebSocket = new Mock<IWebSocket>();
|
||||
|
||||
_mockWebSocketFactory
|
||||
.Setup(static x => x.Create())
|
||||
.Returns(mockWebSocket.Object);
|
||||
|
||||
using var cts = new CancellationTokenSource();
|
||||
|
||||
await _discordGatewayClient.ConnectAsync(cts.Token);
|
||||
await Task.Delay(100);
|
||||
await cts.CancelAsync();
|
||||
|
||||
mockWebSocket.Verify(static x => x.State, Times.AtMostOnce);
|
||||
mockWebSocket.Verify(
|
||||
static x => x.ReceiveAsync(It.IsAny<ArraySegment<byte>>(), It.IsAny<CancellationToken>()),
|
||||
Times.AtMostOnce
|
||||
);
|
||||
}
|
||||
|
||||
[Fact]
|
||||
public async Task ConnectAsync_WhenCalledAndHelloEventReceived_ItShouldStartSendingHeartbeats()
|
||||
{
|
||||
_mockDiscordRestClient
|
||||
.Setup(static x => x.GetGatewayUrlAsync(It.IsAny<CancellationToken>()))
|
||||
.ReturnsAsync("wss://gateway.discord.gg");
|
||||
|
||||
var mockWebSocket = new Mock<IWebSocket>();
|
||||
|
||||
mockWebSocket
|
||||
.Setup(static x => x.State)
|
||||
.Returns(WebSocketState.Open);
|
||||
|
||||
var messageQueue = new Queue<(WebSocketReceiveResult, byte[])>();
|
||||
|
||||
var heartbeatInterval = 100;
|
||||
|
||||
var helloEventPayload = CreateEventPayload(new
|
||||
{
|
||||
op = 10,
|
||||
d = new
|
||||
{
|
||||
heartbeat_interval = heartbeatInterval,
|
||||
}
|
||||
});
|
||||
|
||||
messageQueue.Enqueue((
|
||||
new(helloEventPayload.Bytes.Length, WebSocketMessageType.Text, true),
|
||||
helloEventPayload.Bytes
|
||||
));
|
||||
|
||||
SetupReceiveMessageSequence(mockWebSocket, messageQueue);
|
||||
|
||||
_mockWebSocketFactory
|
||||
.Setup(static x => x.Create())
|
||||
.Returns(mockWebSocket.Object);
|
||||
|
||||
var cts = new CancellationTokenSource();
|
||||
await _discordGatewayClient.ConnectAsync(cts.Token);
|
||||
|
||||
await Task.Delay((int)(heartbeatInterval * 1.5));
|
||||
await cts.CancelAsync();
|
||||
|
||||
var expectedHeartbeatPayload = CreateEventPayload(new HeartbeatDiscordEvent(null));
|
||||
|
||||
mockWebSocket.Verify(
|
||||
x => x.SendAsync(
|
||||
It.Is<ArraySegment<byte>>(b => expectedHeartbeatPayload.Bytes.SequenceEqual(b)),
|
||||
It.IsAny<WebSocketMessageType>(),
|
||||
It.IsAny<bool>(),
|
||||
It.IsAny<CancellationToken>()
|
||||
),
|
||||
Times.Once
|
||||
);
|
||||
}
|
||||
|
||||
[Fact]
|
||||
public async Task ConnectAsync_WhenCalledAndHeartbeatIsNotAcknowledged_ItShouldDisconnectAndAttemptToResume()
|
||||
{
|
||||
_mockDiscordRestClient
|
||||
.Setup(static x => x.GetGatewayUrlAsync(It.IsAny<CancellationToken>()))
|
||||
.ReturnsAsync("wss://gateway.discord.gg");
|
||||
|
||||
var mockWebSocket = new Mock<IWebSocket>();
|
||||
|
||||
mockWebSocket
|
||||
.Setup(static x => x.State)
|
||||
.Returns(WebSocketState.Open);
|
||||
|
||||
var messageQueue = new Queue<(WebSocketReceiveResult, byte[])>();
|
||||
|
||||
var heartbeatInterval = 100;
|
||||
|
||||
var helloEventPayload = CreateEventPayload(new
|
||||
{
|
||||
op = 10,
|
||||
d = new
|
||||
{
|
||||
heartbeat_interval = heartbeatInterval,
|
||||
}
|
||||
});
|
||||
|
||||
messageQueue.Enqueue((
|
||||
new(helloEventPayload.Bytes.Length, WebSocketMessageType.Text, true),
|
||||
helloEventPayload.Bytes
|
||||
));
|
||||
|
||||
SetupReceiveMessageSequence(mockWebSocket, messageQueue);
|
||||
|
||||
_mockWebSocketFactory
|
||||
.Setup(static x => x.Create())
|
||||
.Returns(mockWebSocket.Object);
|
||||
|
||||
var cts = new CancellationTokenSource();
|
||||
await _discordGatewayClient.ConnectAsync(cts.Token);
|
||||
|
||||
await Task.Delay((int)(heartbeatInterval * 2.5));
|
||||
await cts.CancelAsync();
|
||||
|
||||
var expectedHeartbeatPayload = CreateEventPayload(new HeartbeatDiscordEvent(null));
|
||||
|
||||
mockWebSocket.Verify(
|
||||
x => x.SendAsync(
|
||||
It.Is<ArraySegment<byte>>(b => expectedHeartbeatPayload.Bytes.SequenceEqual(b)),
|
||||
It.IsAny<WebSocketMessageType>(),
|
||||
It.IsAny<bool>(),
|
||||
It.IsAny<CancellationToken>()
|
||||
),
|
||||
Times.Once
|
||||
);
|
||||
|
||||
mockWebSocket.Verify(
|
||||
static x => x.CloseAsync(
|
||||
It.Is<WebSocketCloseStatus>(
|
||||
x => x != WebSocketCloseStatus.NormalClosure && x != WebSocketCloseStatus.EndpointUnavailable
|
||||
),
|
||||
It.IsAny<string>(),
|
||||
It.IsAny<CancellationToken>()
|
||||
),
|
||||
Times.Once
|
||||
);
|
||||
}
|
||||
|
||||
[Fact]
|
||||
public async Task ConnectAsync_WhenCalledAndHeartbeatNotAcknowlegedAndAlreadyClosed_ItShouldNotDisconnect()
|
||||
{
|
||||
_mockDiscordRestClient
|
||||
.Setup(static x => x.GetGatewayUrlAsync(It.IsAny<CancellationToken>()))
|
||||
.ReturnsAsync("wss://gateway.discord.gg");
|
||||
|
||||
var mockWebSocket = new Mock<IWebSocket>();
|
||||
|
||||
mockWebSocket
|
||||
.SetupSequence(static x => x.State)
|
||||
.Returns(WebSocketState.Open)
|
||||
.Returns(WebSocketState.Open)
|
||||
.Returns(WebSocketState.Open)
|
||||
.Returns(WebSocketState.Open)
|
||||
.Returns(WebSocketState.Open)
|
||||
.Returns(WebSocketState.Open)
|
||||
.Returns(WebSocketState.Open)
|
||||
.Returns(WebSocketState.Closed);
|
||||
|
||||
var messageQueue = new Queue<(WebSocketReceiveResult, byte[])>();
|
||||
|
||||
var heartbeatInterval = 100;
|
||||
|
||||
var helloEventPayload = CreateEventPayload(new
|
||||
{
|
||||
op = 10,
|
||||
d = new
|
||||
{
|
||||
heartbeat_interval = heartbeatInterval,
|
||||
}
|
||||
});
|
||||
|
||||
messageQueue.Enqueue((
|
||||
new(helloEventPayload.Bytes.Length, WebSocketMessageType.Text, true),
|
||||
helloEventPayload.Bytes
|
||||
));
|
||||
|
||||
SetupReceiveMessageSequence(mockWebSocket, messageQueue);
|
||||
|
||||
_mockWebSocketFactory
|
||||
.Setup(static x => x.Create())
|
||||
.Returns(mockWebSocket.Object);
|
||||
|
||||
var cts = new CancellationTokenSource();
|
||||
await _discordGatewayClient.ConnectAsync(cts.Token);
|
||||
|
||||
await Task.Delay((int)(heartbeatInterval * 2.5));
|
||||
await cts.CancelAsync();
|
||||
|
||||
var expectedHeartbeatPayload = CreateEventPayload(new HeartbeatDiscordEvent(null));
|
||||
|
||||
mockWebSocket.Verify(
|
||||
x => x.SendAsync(
|
||||
It.Is<ArraySegment<byte>>(b => expectedHeartbeatPayload.Bytes.SequenceEqual(b)),
|
||||
It.IsAny<WebSocketMessageType>(),
|
||||
It.IsAny<bool>(),
|
||||
It.IsAny<CancellationToken>()
|
||||
),
|
||||
Times.Once
|
||||
);
|
||||
|
||||
mockWebSocket.Verify(
|
||||
static x => x.CloseAsync(
|
||||
It.IsAny<WebSocketCloseStatus>(),
|
||||
It.IsAny<string>(),
|
||||
It.IsAny<CancellationToken>()
|
||||
),
|
||||
Times.Never
|
||||
);
|
||||
}
|
||||
|
||||
private static void SetupReceiveMessageSequence(
|
||||
Mock<IWebSocket> mockWebSocket,
|
||||
Queue<(WebSocketReceiveResult, byte[])> messageQueue
|
||||
|
||||
Reference in New Issue
Block a user