fix tests

Author: pepeiborra

test/functional/FunctionalCodeAction.hs

@@ -372,7 +372,8 @@ redundantImportTests = testGroup "redundant import code actions" [
     , testCase "doesn't touch other imports" $ runSession hlsCommand noLiteralCaps "test/testdata/redundantImportTest/" $ do
         doc <- openDoc "src/MultipleImports.hs" "haskell"
         _   <- waitForDiagnosticsFromSource doc "typecheck"
-        InL cmd : _ <- getAllCodeActions doc
+        cas <- getAllCodeActions doc
+        cmd <- liftIO $ inspectCommand cas ["redundant import"]
         executeCommand cmd
         _ <- anyRequest
         contents <- documentContents doc
@@ -439,11 +440,12 @@ signatureTests = testGroup "missing top level signature code actions" [
         doc <- openDoc "TopLevelSignature.hs" "haskell"
 
         _ <- waitForDiagnosticsFromSource doc "typecheck"
-        cas <- map fromAction <$> getAllCodeActions doc
+        cas <- getAllCodeActions doc
 
-        liftIO $ "add signature: main :: IO ()" `elem` map (^. L.title) cas @? "Contains code action"
+        liftIO $ expectCodeAction cas ["add signature: main :: IO ()"]
 
-        executeCodeAction $ head cas
+        replaceWithStuff <- liftIO $ inspectCodeAction cas ["add signature"]
+        executeCodeAction replaceWithStuff
 
         contents <- documentContents doc