diff --git a/Content.IntegrationTests/Tests/Networking/ReconnectTest.cs b/Content.IntegrationTests/Tests/Networking/ReconnectTest.cs index bbecd71254..aea01d41c5 100644 --- a/Content.IntegrationTests/Tests/Networking/ReconnectTest.cs +++ b/Content.IntegrationTests/Tests/Networking/ReconnectTest.cs @@ -18,51 +18,29 @@ namespace Content.IntegrationTests.Tests.Networking await Task.WhenAll(client.WaitIdleAsync(), server.WaitIdleAsync()); // Connect. - client.SetConnectTarget(server); - client.Post(() => IoCManager.Resolve().ClientConnect(null, 0, null)); + await client.WaitPost(() => IoCManager.Resolve().ClientConnect(null, 0, null)); // Run some ticks for the handshake to complete and such. - - for (var i = 0; i < 10; i++) - { - server.RunTicks(1); - await server.WaitIdleAsync(); - client.RunTicks(1); - await client.WaitIdleAsync(); - } + await RunTicksSync(client, server, 10); await Task.WhenAll(client.WaitIdleAsync(), server.WaitIdleAsync()); - client.Post(() => IoCManager.Resolve().ProcessCommand("disconnect")); + await client.WaitPost(() => IoCManager.Resolve().ProcessCommand("disconnect")); // Run some ticks for the disconnect to complete and such. - for (var i = 0; i < 5; i++) - { - server.RunTicks(1); - await server.WaitIdleAsync(); - client.RunTicks(1); - await client.WaitIdleAsync(); - } + await RunTicksSync(client, server, 5); await Task.WhenAll(client.WaitIdleAsync(), server.WaitIdleAsync()); // Reconnect. - client.SetConnectTarget(server); - client.Post(() => IoCManager.Resolve().ClientConnect(null, 0, null)); + await client.WaitPost(() => IoCManager.Resolve().ClientConnect(null, 0, null)); // Run some ticks for the handshake to complete and such. - - for (var i = 0; i < 10; i++) - { - server.RunTicks(1); - await server.WaitIdleAsync(); - client.RunTicks(1); - await client.WaitIdleAsync(); - } + await RunTicksSync(client, server, 10); await Task.WhenAll(client.WaitIdleAsync(), server.WaitIdleAsync()); }