diff --git a/Assets/Mirror/Tests/Editor/Transports/MiddlewareTransportTest.cs b/Assets/Mirror/Tests/Editor/Transports/MiddlewareTransportTest.cs index 32af6cbf7..858e1050e 100644 --- a/Assets/Mirror/Tests/Editor/Transports/MiddlewareTransportTest.cs +++ b/Assets/Mirror/Tests/Editor/Transports/MiddlewareTransportTest.cs @@ -292,13 +292,13 @@ public void TestClientExceptionCallback() } [Test] - [TestCase(0)] - [TestCase(1)] - [TestCase(19)] - public void TestServerConnectedCallback(int id) + [TestCase(0, "")] + [TestCase(1, "")] + [TestCase(19, "")] + public void TestServerConnectedCallback(int id, string remoteClientAddress) { int called = 0; - middleware.OnServerConnected = (i) => + middleware.OnServerConnectedWithAddress = (i, remoteClientAddress) => { called++; Assert.That(i, Is.EqualTo(id)); @@ -306,10 +306,10 @@ public void TestServerConnectedCallback(int id) // start to give callback to inner middleware.ServerStart(); - inner.OnServerConnected.Invoke(id); + inner.OnServerConnectedWithAddress.Invoke(id, remoteClientAddress); Assert.That(called, Is.EqualTo(1)); - inner.OnServerConnected.Invoke(id); + inner.OnServerConnectedWithAddress.Invoke(id, remoteClientAddress); Assert.That(called, Is.EqualTo(2)); } diff --git a/Assets/Mirror/Transports/Middleware/MiddlewareTransport.cs b/Assets/Mirror/Transports/Middleware/MiddlewareTransport.cs index fc7c9cd00..21d2ac3ef 100644 --- a/Assets/Mirror/Transports/Middleware/MiddlewareTransport.cs +++ b/Assets/Mirror/Transports/Middleware/MiddlewareTransport.cs @@ -43,6 +43,7 @@ public override void ClientConnect(string address) public override void ServerStart() { inner.OnServerConnected = OnServerConnected; + inner.OnServerConnectedWithAddress = OnServerConnectedWithAddress; inner.OnServerDataReceived = OnServerDataReceived; inner.OnServerDisconnected = OnServerDisconnected; inner.OnServerError = OnServerError;