diff --git a/src/Runner.Listener/MessageListener.cs b/src/Runner.Listener/MessageListener.cs index 7be8b0bce84..ab91cff0508 100644 --- a/src/Runner.Listener/MessageListener.cs +++ b/src/Runner.Listener/MessageListener.cs @@ -188,12 +188,12 @@ public async Task DeleteSessionAsync() { using (var ts = new CancellationTokenSource(TimeSpan.FromSeconds(30))) { + await _runnerServer.DeleteAgentSessionAsync(_settings.PoolId, _session.SessionId, ts.Token); + if (_isBrokerSession) { await _brokerServer.DeleteSessionAsync(ts.Token); - return; } - await _runnerServer.DeleteAgentSessionAsync(_settings.PoolId, _session.SessionId, ts.Token); } } else diff --git a/src/Test/L0/Listener/MessageListenerL0.cs b/src/Test/L0/Listener/MessageListenerL0.cs index 57a1f60d800..b6837ad9a11 100644 --- a/src/Test/L0/Listener/MessageListenerL0.cs +++ b/src/Test/L0/Listener/MessageListenerL0.cs @@ -272,7 +272,7 @@ public async void DeleteSessionWithBrokerMigration() //Assert _runnerServer .Verify(x => x.DeleteAgentSessionAsync( - _settings.PoolId, expectedSession.SessionId, It.IsAny()), Times.Never()); + _settings.PoolId, expectedBrokerSession.SessionId, It.IsAny()), Times.Once()); _brokerServer .Verify(x => x.DeleteSessionAsync(It.IsAny()), Times.Once()); }