Retry code actions in Wingman test suite

Just trying to fix the flaky Windows CI

Author: pepeiborra

plugins/hls-tactics-plugin/test/Utils.hs

@@ -4,6 +4,7 @@
 {-# LANGUAGE ScopedTypeVariables   #-}
 {-# LANGUAGE TypeOperators         #-}
 {-# LANGUAGE ViewPatterns          #-}
+{-# LANGUAGE TypeApplications #-}
 
 module Utils where
 
@@ -116,20 +117,27 @@ mkGoldenTest eq tc occ line col input =
       -- wait for the entire build to finish, so that Tactics code actions that
       -- use stale data will get uptodate stuff
       void waitForBuildQueue
-      actions <- getCodeActions doc $ pointRange line col
-      case find ((== Just (tacticTitle tc occ)) . codeActionTitle) actions of
-        Just (InR CodeAction {_command = Just c}) -> do
-            executeCommand c
-            _resp <- skipManyTill anyMessage (message SWorkspaceApplyEdit)
-            edited <- documentContents doc
-            let expected_name = input <.> "expected" <.> "hs"
-            -- Write golden tests if they don't already exist
-            liftIO $ (doesFileExist expected_name >>=) $ flip unless $ do
-                T.writeFile expected_name edited
-            expected <- liftIO $ T.readFile expected_name
-            liftIO $ edited `eq` expected
-        _ -> error $ show actions
-
+      retryAction 4 $ do
+        actions <- getCodeActions doc $ pointRange line col
+        case find ((== Just (tacticTitle tc occ)) . codeActionTitle) actions of
+            Just (InR CodeAction {_command = Just c}) -> do
+                executeCommand c
+                _resp <- skipManyTill anyMessage (message SWorkspaceApplyEdit)
+                edited <- documentContents doc
+                let expected_name = input <.> "expected" <.> "hs"
+                -- Write golden tests if they don't already exist
+                liftIO $ (doesFileExist expected_name >>=) $ flip unless $ do
+                    T.writeFile expected_name edited
+                expected <- liftIO $ T.readFile expected_name
+                liftIO $ E.try (edited `eq` expected)
+            _ -> return $ Left $ E.toException $ E.ErrorCall $ show actions
+
+retryAction :: Int -> Session (Either E.SomeException a) -> Session a
+retryAction n act = do
+    res <- act
+    case (n, res) of
+        (_, Right x) -> return x
+        (_, Left e) -> if n>1 then retryAction (n-1) act else E.throw e
 
 mkCodeLensTest
     :: FilePath