Always Delete Actions Service Session (#3214)

* Delete Actions Service session always * update tes
actions · Mar 21, 2024 · 8ebf298 · 8ebf298
1 parent 4b85145
commit 8ebf298
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git 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
@@ -272,7 +272,7 @@ public async void DeleteSessionWithBrokerMigration()
  //Assert
  _runnerServer
  .Verify(x => x.DeleteAgentSessionAsync(
- _settings.PoolId, expectedSession.SessionId, It.IsAny<CancellationToken>()), Times.Never());
+ _settings.PoolId, expectedBrokerSession.SessionId, It.IsAny<CancellationToken>()), Times.Once());
  _brokerServer
  .Verify(x => x.DeleteSessionAsync(It.IsAny<CancellationToken>()), Times.Once());
  }