outbound-governor: fixed shrinking of local root peers

IntersectMBO · Apr 25, 2024 · 33093ce · 33093ce
1 parent d96248b
commit 33093ce
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/ouroboros-network/sim-tests-lib/Test/Ouroboros/Network/PeerSelection/MockEnvironment.hs b/ouroboros-network/sim-tests-lib/Test/Ouroboros/Network/PeerSelection/MockEnvironment.hs
@@ -879,16 +879,15 @@ instance Arbitrary GovernorMockEnvironment where
          publicRootPeers' <- shrinkPublicRootPeers publicRootPeers
       ]
     where
-      shrinkLocalRootPeers (LocalRootPeers m g) =
-        [ LocalRootPeers m' g'
-          | (m', g') <- shrink (m, g)
+      shrinkLocalRootPeers a =
+        [ LocalRootPeers.fromGroups g
+          | g <- shrink (LocalRootPeers.toGroups a)
         ]
       shrinkPublicRootPeers (PublicRootPeers pp bsp lp blp) =
         [ PublicRootPeers pp' bsp' lp' blp'
           | (pp', bsp', lp', blp') <- shrink (pp, bsp, lp, blp)
         ]
 
-
 --
 -- Tests for the QC Arbitrary instances
 --