Multi component test: wait for reference ready

Author: pepeiborra

ghcide/test/exe/Main.hs

@@ -5513,8 +5513,8 @@ simpleMultiTest2 = testCase "simple-multi-test2" $ runWithExtraFiles "multi" $ \
         bPath = dir </> "b/B.hs"
     bdoc <- openDoc bPath "haskell"
     WaitForIdeRuleResult {} <- waitForAction "TypeCheck" bdoc
-    adoc@(TextDocumentIdentifier auri) <- openDoc aPath "haskell"
-    WaitForIdeRuleResult {} <- waitForAction "TypeCheck" adoc
+    TextDocumentIdentifier auri <- openDoc aPath "haskell"
+    skipManyTill anyMessage $ isReferenceReady aPath
     locs <- getDefinitions bdoc (Position 2 7)
     let fooL = mkL auri 2 0 2 3
     checkDefs locs (pure [fooL])
@@ -5528,8 +5528,8 @@ simpleMultiTest3 = testCase "simple-multi-test3" $ runWithExtraFiles "multi" $ \
         cPath = dir </> "c/C.hs"
     bdoc <- openDoc bPath "haskell"
     WaitForIdeRuleResult {} <- waitForAction "TypeCheck" bdoc
-    adoc@(TextDocumentIdentifier auri) <- openDoc aPath "haskell"
-    WaitForIdeRuleResult {} <- waitForAction "TypeCheck" adoc
+    TextDocumentIdentifier auri <- openDoc aPath "haskell"
+    skipManyTill anyMessage $ isReferenceReady aPath
     cdoc <- openDoc cPath "haskell"
     WaitForIdeRuleResult {} <- waitForAction "TypeCheck" cdoc
     locs <- getDefinitions cdoc (Position 2 7)