Emit TraceTerminatedByGDDGovernor on CSClient.DensityTooLow

Follow-up to #1027
IntersectMBO · Mar 27, 2024 · 34a3689 · 34a3689
1 parent 54a9604
commit 34a3689
Showing 1 changed file with 2 additions and 5 deletions.
diff --git a/ouroboros-consensus-diffusion/test/consensus-test/Test/Consensus/PeerSimulator/ChainSync.hs b/ouroboros-consensus-diffusion/test/consensus-test/Test/Consensus/PeerSimulator/ChainSync.hs
@@ -12,8 +12,7 @@ module Test.Consensus.PeerSimulator.ChainSync (
   , runChainSyncServer
   ) where
 
-import           Control.Exception (AsyncException (ThreadKilled),
-                     SomeException)
+import           Control.Exception (SomeException)
 import           Control.Monad.Class.MonadTimer.SI (MonadTimer)
 import           Control.Tracer (Tracer (Tracer), nullTracer, traceWith)
 import           Data.Map.Strict (Map)
@@ -205,10 +204,8 @@ runChainSyncClient
             traceWith tracer $ TraceChainSyncClientTerminationEvent peerId TraceExceededTimeLimitCS
           Nothing -> pure ()
         case fromException exn of
-          Just ThreadKilled ->
+          Just CSClient.DensityTooLow ->
             traceWith tracer $ TraceChainSyncClientTerminationEvent peerId TraceTerminatedByGDDGovernor
-          _ -> pure ()
-        case fromException exn of
           Just CSClient.EmptyBucket ->
             traceWith tracer $ TraceChainSyncClientTerminationEvent peerId TraceTerminatedByLoP
           _ -> pure ()