SshClient: replace DisconnectedAsync API with Disconnected API.

tmds · Dec 11, 2024 · d4ab315 · d4ab315
1 parent 2999ea0
commit d4ab315
Show file tree

Hide file tree

Showing 3 changed files with 21 additions and 28 deletions.
diff --git 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
@@ -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
@@ -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);
     }
 }