fix a flaky test

smithy-lang · Jun 13, 2024 · 2ae801a · 2ae801a
1 parent c9c751d
commit 2ae801a
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/src/test/java/software/amazon/smithy/lsp/SmithyLanguageServerTest.java b/src/test/java/software/amazon/smithy/lsp/SmithyLanguageServerTest.java
@@ -430,7 +430,7 @@ public void didChange() throws Exception {
         server.didChange(changeBuilder.range(rangeAdapter.shiftRight().build()).text(" ").build());
         server.didChange(changeBuilder.range(rangeAdapter.shiftRight().build()).text("G").build());
 
-        server.getLifecycleManager().getTask(uri).get();
+        server.getLifecycleManager().waitForAllTasks();
 
         // mostly so you can see what it looks like
         assertThat(server.getProject().getDocument(uri).copyText(), equalTo(safeString("$version: \"2\"\n" +
@@ -451,8 +451,7 @@ public void didChange() throws Exception {
                 .buildCompletion();
         List<CompletionItem> completions = server.completion(completionParams).get().getLeft();
 
-        // TODO: Somehow this has become flaky
-        assertThat(completions, containsInAnyOrder(hasLabel("GetFoo"), hasLabel("GetFooInput")));
+        assertThat(completions, hasItem(hasLabel("GetFooInput")));
     }
 
     @Test