Multi component test: wait for reference ready

haskell · Feb 19, 2022 · 1d9b6a1 · 1d9b6a1
1 parent 6c7fd99
commit 1d9b6a1
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/ghcide/test/exe/Main.hs b/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)