From d4ab3156690d9a1fb7b12c88c9ed1466f33820ae Mon Sep 17 00:00:00 2001
From: Tom Deseyn <tom.deseyn@gmail.com>
Date: Wed, 11 Dec 2024 08:13:39 +0100
Subject: [PATCH] SshClient: replace DisconnectedAsync API with Disconnected
 API.

---
 README.md                           |  3 +--
 src/Tmds.Ssh/SshClient.cs           | 11 +++++----
 test/Tmds.Ssh.Tests/ConnectTests.cs | 35 +++++++++++------------------
 3 files changed, 21 insertions(+), 28 deletions(-)

diff --git a/README.md b/README.md
index 997047b..302c315 100644
--- a/README.md
+++ b/README.md
@@ -59,8 +59,7 @@ class SshClient : IDisposable
 
   // Not usable with AutoReconnect.
   // A connection must be established before calling this method.
-  // Throws if the connection was closed for another reason than explicit close by the user.
-  Task DisconnectedAsync(CancellationToken cancellationToken = default);
+  CancellationToken Disconnected { get; }
 
   Task<RemoteProcess> ExecuteAsync(string command, CancellationToken cancellationToken);
   Task<RemoteProcess> ExecuteAsync(string command, ExecuteOptions? options = null, CancellationToken cancellationToken = default);
diff --git a/src/Tmds.Ssh/SshClient.cs b/src/Tmds.Ssh/SshClient.cs
index 03bc27b..be088c1 100644
--- a/src/Tmds.Ssh/SshClient.cs
+++ b/src/Tmds.Ssh/SshClient.cs
@@ -89,13 +89,16 @@ public async Task ConnectAsync(CancellationToken cancellationToken = default)
         await GetSessionAsync(cancellationToken, explicitConnect: true).ConfigureAwait(false);
     }
 
-    public Task DisconnectedAsync(CancellationToken cancellationToken = default)
+    public CancellationToken Disconnected
     {
-        SshSession session = GetConnectedSesion(cancellationToken);
-        return session.DisconnectedAsync(cancellationToken);
+        get
+        {
+            SshSession session = GetConnectedSesion();
+            return session.ConnectionClosed;
+        }
     }
 
-    private SshSession GetConnectedSesion(CancellationToken cancellationToken, [CallerMemberNameAttribute]string? callerMethod = null)
+    private SshSession GetConnectedSesion([CallerMemberNameAttribute]string? callerMethod = null)
     {
         if (_autoReconnect)
         {
diff --git a/test/Tmds.Ssh.Tests/ConnectTests.cs b/test/Tmds.Ssh.Tests/ConnectTests.cs
index 315a4b9..49c160c 100644
--- a/test/Tmds.Ssh.Tests/ConnectTests.cs
+++ b/test/Tmds.Ssh.Tests/ConnectTests.cs
@@ -461,62 +461,53 @@ public async Task SshConfig_Options()
     }
 
     [Fact]
-    public async Task DisconnectedAsync()
+    public async Task Disconnected()
     {
         using var client = await _sshServer.CreateClientAsync();
 
-        Task disconnectedTask = client.DisconnectedAsync();
+        CancellationToken disconnected = client.Disconnected;
 
         client.Dispose();
 
-        await disconnectedTask;
+        Assert.True(disconnected.IsCancellationRequested);
     }
 
     [Fact]
-    public async Task DisconnectedAsync_WithError()
+    public async Task Disconnected_WithError()
     {
         using var client = await _sshServer.CreateClientAsync();
 
-        Task disconnectedTask = client.DisconnectedAsync();
+        CancellationToken disconnected = client.Disconnected;
 
         client.ForceConnectionClose();
 
-        await Assert.ThrowsAsync<SshConnectionClosedException>(() => disconnectedTask);
+        Assert.True(disconnected.IsCancellationRequested);
+        Assert.True(client.Disconnected.IsCancellationRequested);
     }
 
     [Fact]
-    public async Task DisconnectedAsync_RequiresConnect()
+    public async Task Disconnected_RequiresConnect()
     {
         using var client = await _sshServer.CreateClientAsync(connect: false);
 
-        await Assert.ThrowsAsync<InvalidOperationException>(() => client.DisconnectedAsync());
+        Assert.Throws<InvalidOperationException>(() => client.Disconnected);
     }
 
     [Fact]
-    public async Task DisconnectedAsync_NotWhenAutoReconnect()
+    public async Task Disconnected_NotWhenAutoReconnect()
     {
         using var client = await _sshServer.CreateClientAsync(configure: settings => settings.AutoReconnect = true);
 
-        await Assert.ThrowsAsync<InvalidOperationException>(() => client.DisconnectedAsync());
+        Assert.Throws<InvalidOperationException>(() => client.Disconnected);
     }
 
     [Fact]
-    public async Task DisconnectedAsync_NotAfterDispose()
+    public async Task Disconnected_NotAfterDispose()
     {
         using var client = await _sshServer.CreateClientAsync();
 
         client.Dispose();
 
-        await Assert.ThrowsAsync<ObjectDisposedException>(() => client.DisconnectedAsync());
-    }
-
-    [Fact]
-    public async Task DisconnectedAsync_CanCancel()
-    {
-        using var client = await _sshServer.CreateClientAsync();
-
-        var cts = new CancellationTokenSource(50);
-
-        await Assert.ThrowsAnyAsync<OperationCanceledException>(() => client.DisconnectedAsync(cts.Token));
+        Assert.Throws<ObjectDisposedException>(() => client.Disconnected);
     }
 }