process TODO.md step6
This commit is contained in:
@@ -0,0 +1,182 @@
|
||||
using System;
|
||||
using System.Collections.Generic;
|
||||
using System.Net;
|
||||
using System.Threading.Tasks;
|
||||
using Google.Protobuf;
|
||||
using Network.Defines;
|
||||
using Network.NetworkApplication;
|
||||
using Network.NetworkHost;
|
||||
using Network.NetworkTransport;
|
||||
using NUnit.Framework;
|
||||
|
||||
namespace Tests.EditMode.Network
|
||||
{
|
||||
public class ServerRuntimeEntryPointTests
|
||||
{
|
||||
private static readonly IPEndPoint Peer = new(IPAddress.Loopback, 9100);
|
||||
|
||||
[Test]
|
||||
public void StartServerRuntimeAsync_ReliableOnly_StartsAndExposesServerHost()
|
||||
{
|
||||
var createdTransports = new Dictionary<int, FakeTransport>();
|
||||
var configuration = new ServerRuntimeConfiguration(9000)
|
||||
{
|
||||
TransportFactory = port => CreateTransport(createdTransports, port)
|
||||
};
|
||||
|
||||
using var runtime = NetworkIntegrationFactory.StartServerRuntimeAsync(configuration).GetAwaiter().GetResult();
|
||||
|
||||
Assert.That(createdTransports.Keys, Is.EquivalentTo(new[] { 9000 }));
|
||||
Assert.That(runtime.IsRunning, Is.True);
|
||||
Assert.That(runtime.Host.Transport, Is.SameAs(createdTransports[9000]));
|
||||
Assert.That(runtime.Host.SyncTransport, Is.Null);
|
||||
Assert.That(createdTransports[9000].StartCallCount, Is.EqualTo(1));
|
||||
}
|
||||
|
||||
[Test]
|
||||
public void StartServerRuntimeAsync_DualTransport_StartsBothConfiguredLanes()
|
||||
{
|
||||
var createdTransports = new Dictionary<int, FakeTransport>();
|
||||
var configuration = new ServerRuntimeConfiguration(9000)
|
||||
{
|
||||
SyncPort = 9001,
|
||||
TransportFactory = port => CreateTransport(createdTransports, port)
|
||||
};
|
||||
|
||||
using var runtime = ServerRuntimeEntryPoint.StartAsync(configuration).GetAwaiter().GetResult();
|
||||
|
||||
Assert.That(createdTransports.Keys, Is.EquivalentTo(new[] { 9000, 9001 }));
|
||||
Assert.That(runtime.Host.Transport, Is.SameAs(createdTransports[9000]));
|
||||
Assert.That(runtime.Host.SyncTransport, Is.SameAs(createdTransports[9001]));
|
||||
Assert.That(createdTransports[9000].StartCallCount, Is.EqualTo(1));
|
||||
Assert.That(createdTransports[9001].StartCallCount, Is.EqualTo(1));
|
||||
}
|
||||
|
||||
[Test]
|
||||
public void StartServerRuntimeAsync_SyncTransportStartFails_RollsBackStartedResources()
|
||||
{
|
||||
var createdTransports = new Dictionary<int, FakeTransport>();
|
||||
var configuration = new ServerRuntimeConfiguration(9000)
|
||||
{
|
||||
SyncPort = 9001,
|
||||
TransportFactory = port =>
|
||||
{
|
||||
var transport = CreateTransport(createdTransports, port);
|
||||
if (port == 9001)
|
||||
{
|
||||
transport.StartException = new InvalidOperationException("sync failed");
|
||||
}
|
||||
|
||||
return transport;
|
||||
}
|
||||
};
|
||||
|
||||
var exception = Assert.Throws<InvalidOperationException>(() =>
|
||||
ServerRuntimeEntryPoint.StartAsync(configuration).GetAwaiter().GetResult());
|
||||
|
||||
Assert.That(exception.Message, Is.EqualTo("sync failed"));
|
||||
Assert.That(createdTransports[9000].StartCallCount, Is.EqualTo(1));
|
||||
Assert.That(createdTransports[9000].StopCallCount, Is.EqualTo(1));
|
||||
Assert.That(createdTransports[9001].StartCallCount, Is.EqualTo(1));
|
||||
Assert.That(createdTransports[9001].StopCallCount, Is.EqualTo(1));
|
||||
}
|
||||
|
||||
[Test]
|
||||
public void ServerRuntimeHandle_DrainsMessages_ExposesManagedSessions_AndStopsIdempotently()
|
||||
{
|
||||
var createdTransports = new Dictionary<int, FakeTransport>();
|
||||
var configuration = new ServerRuntimeConfiguration(9000)
|
||||
{
|
||||
Dispatcher = new MainThreadNetworkDispatcher(),
|
||||
TransportFactory = port => CreateTransport(createdTransports, port)
|
||||
};
|
||||
var runtime = ServerRuntimeEntryPoint.StartAsync(configuration).GetAwaiter().GetResult();
|
||||
var handled = false;
|
||||
|
||||
runtime.Host.MessageManager.RegisterHandler(MessageType.Heartbeat, (payload, sender) =>
|
||||
{
|
||||
handled = true;
|
||||
});
|
||||
|
||||
createdTransports[9000].EmitReceive(CreateEnvelope(MessageType.Heartbeat, new Heartbeat()), Peer);
|
||||
|
||||
Assert.That(runtime.ManagedSessions.Count, Is.EqualTo(1));
|
||||
Assert.That(runtime.TryGetSession(Peer, out var session), Is.True);
|
||||
Assert.That(session.SessionManager.State, Is.EqualTo(ConnectionState.TransportConnected));
|
||||
Assert.That(runtime.Host.MessageManager.PendingMessageCount, Is.EqualTo(1));
|
||||
|
||||
runtime.DrainPendingMessagesAsync().GetAwaiter().GetResult();
|
||||
runtime.UpdateLifecycle();
|
||||
|
||||
Assert.That(handled, Is.True);
|
||||
Assert.That(runtime.Host.MessageManager.PendingMessageCount, Is.EqualTo(0));
|
||||
|
||||
runtime.Stop();
|
||||
runtime.Stop();
|
||||
|
||||
Assert.That(runtime.IsRunning, Is.False);
|
||||
Assert.That(runtime.ManagedSessions.Count, Is.EqualTo(0));
|
||||
Assert.That(createdTransports[9000].StopCallCount, Is.EqualTo(1));
|
||||
}
|
||||
|
||||
private static FakeTransport CreateTransport(IDictionary<int, FakeTransport> createdTransports, int port)
|
||||
{
|
||||
var transport = new FakeTransport();
|
||||
createdTransports.Add(port, transport);
|
||||
return transport;
|
||||
}
|
||||
|
||||
private static byte[] CreateEnvelope(MessageType type, IMessage payload)
|
||||
{
|
||||
return new Envelope
|
||||
{
|
||||
Type = (int)type,
|
||||
Payload = payload.ToByteString()
|
||||
}.ToByteArray();
|
||||
}
|
||||
|
||||
private sealed class FakeTransport : ITransport
|
||||
{
|
||||
public Exception StartException { get; set; }
|
||||
|
||||
public int StartCallCount { get; private set; }
|
||||
|
||||
public int StopCallCount { get; private set; }
|
||||
|
||||
public event Action<byte[], IPEndPoint> OnReceive;
|
||||
|
||||
public Task StartAsync()
|
||||
{
|
||||
StartCallCount++;
|
||||
if (StartException != null)
|
||||
{
|
||||
throw StartException;
|
||||
}
|
||||
|
||||
return Task.CompletedTask;
|
||||
}
|
||||
|
||||
public void Stop()
|
||||
{
|
||||
StopCallCount++;
|
||||
}
|
||||
|
||||
public void Send(byte[] data)
|
||||
{
|
||||
}
|
||||
|
||||
public void SendTo(byte[] data, IPEndPoint target)
|
||||
{
|
||||
}
|
||||
|
||||
public void SendToAll(byte[] data)
|
||||
{
|
||||
}
|
||||
|
||||
public void EmitReceive(byte[] data, IPEndPoint sender)
|
||||
{
|
||||
OnReceive?.Invoke(data, sender);
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
@@ -0,0 +1,11 @@
|
||||
fileFormatVersion: 2
|
||||
guid: 0a5c5d260e11429dafcee61f53a2f2d7
|
||||
MonoImporter:
|
||||
externalObjects: {}
|
||||
serializedVersion: 2
|
||||
defaultReferences: []
|
||||
executionOrder: 0
|
||||
icon: {instanceID: 0}
|
||||
userData:
|
||||
assetBundleName:
|
||||
assetBundleVariant:
|
||||
Reference in New Issue
Block a user