AuthenticationProvider.Authenticate doesn't have access to RemotingSe…

…ssion.Current.
theRainbird · Dec 6, 2024 · 30ee709 · 30ee709
1 parent 22703c7
commit 30ee709
Showing 1 changed file with 43 additions and 4 deletions.
diff --git a/CoreRemoting.Tests/RpcTests.cs b/CoreRemoting.Tests/RpcTests.cs
@@ -3,6 +3,7 @@
 using System.Diagnostics;
 using System.Threading;
 using System.Threading.Tasks;
+using CoreRemoting.Authentication;
 using CoreRemoting.Channels;
 using CoreRemoting.Serialization;
 using CoreRemoting.Tests.ExternalTypes;
@@ -832,8 +833,10 @@ public void Authentication_is_taken_into_account_and_RejectCall_event_is_fired()
             void RejectCall(object sender, ServerRpcContext e) =>
                 rejectedMethod = e.MethodCallMessage.MethodName;
 
-            _serverFixture.Server.Config.AuthenticationRequired = true;
-            _serverFixture.Server.RejectCall += RejectCall;
+            var server = _serverFixture.Server;
+            server.RejectCall += RejectCall;
+            server.Config.AuthenticationRequired = true;
+
             try
             {
                 using var client = new RemotingClient(new ClientConfig()
@@ -859,8 +862,44 @@ void RejectCall(object sender, ServerRpcContext e) =>
             }
             finally
             {
-                _serverFixture.Server.Config.AuthenticationRequired = false;
-                _serverFixture.Server.RejectCall -= RejectCall;
+                server.Config.AuthenticationRequired = false;
+                server.RejectCall -= RejectCall;
+            }
+        }
+
+        [Fact]
+        public void Authentication_handler_has_access_to_the_current_session()
+        {
+            var server = _serverFixture.Server;
+            var authProvider = server.Config.AuthenticationProvider;
+            server.Config.AuthenticationRequired = true;
+            server.Config.AuthenticationProvider = new FakeAuthProvider
+            {
+                AuthenticateFake = c => RemotingSession.Current != null
+            };
+
+            try
+            {
+                using var client = new RemotingClient(new ClientConfig()
+                {
+                    ConnectionTimeout = 0,
+                    InvocationTimeout = 0,
+                    SendTimeout = 0,
+                    Channel = ClientChannel,
+                    MessageEncryption = false,
+                    ServerPort = _serverFixture.Server.Config.NetworkPort,
+                    Credentials = [new Credential()],
+                });
+
+                client.Connect();
+
+                var proxy = client.CreateProxy<ITestService>();
+                Assert.Equal("123", proxy.Reverse("321"));
+            }
+            finally
+            {
+                server.Config.AuthenticationProvider = authProvider;
+                server.Config.AuthenticationRequired = false;
             }
         }
     }