Clean up tracing of the next instruction from CSJ

IntersectMBO · Apr 23, 2024 · f6cfcff · f6cfcff
1 parent d260957
commit f6cfcff
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/ouroboros-consensus-diffusion/test/consensus-test/Test/Consensus/PeerSimulator/Trace.hs b/ouroboros-consensus-diffusion/test/consensus-test/Test/Consensus/PeerSimulator/Trace.hs
@@ -30,7 +30,7 @@ import           Ouroboros.Consensus.Genesis.Governor (DensityBounds (..),
 import           Ouroboros.Consensus.MiniProtocol.ChainSync.Client
                      (TraceChainSyncClientEvent (..))
 import           Ouroboros.Consensus.MiniProtocol.ChainSync.Client.Jumping
-                     (JumpResult (..))
+                     (Instruction (..), JumpResult (..))
 import           Ouroboros.Consensus.MiniProtocol.ChainSync.Client.State
                      (ChainSyncJumpingState (..), ChainSyncJumpingJumperState(..))
 import           Ouroboros.Consensus.Storage.ChainDB.API (LoE (..))
@@ -372,10 +372,16 @@ traceChainSyncClientEventTestBlockWith pid tracer = \case
     TraceJumpingWaitingForNextInstruction ->
       trace "Waiting for next instruction from the jumping governor"
     TraceJumpingInstructionIs instr ->
-      trace $ "Received instruction: " ++ show instr
+      trace $ "Received instruction: " ++ showInstr instr
   where
     trace = traceUnitWith tracer ("ChainSyncClient " ++ condense pid)
 
+    showInstr :: Instruction TestBlock -> String
+    showInstr = \case
+      JumpTo fragment -> "JumpTo " ++ tersePoint (castPoint $ headPoint fragment)
+      RunNormally -> "RunNormally"
+      Restart -> "Restart"
+
 traceChainSyncClientTerminationEventTestBlockWith ::
   PeerId ->
   Tracer m String ->