chore: Refactor test code to use vscode.languages.getDiagnostics

openfga · Sep 14, 2024 · bdac52f · bdac52f
1 parent f7c151e
commit bdac52f
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 5 deletions.
diff --git a/client/src/test/diagnostics.test.ts b/client/src/test/diagnostics.test.ts
@@ -268,6 +268,8 @@ async function testAutofixSuggestions(docUri: vscode.Uri, expectedDiagnostics: D
     assert.deepEqual(actualDiagnostic.range, expectedDiagnostic.range);
     assert.equal(actualDiagnostic.severity, expectedDiagnostic.severity);
     assert.equal(actualDiagnostic.source, expectedDiagnostic.source);
-    assert.equal(actualDiagnostic.autofix, expectedDiagnostic.autofix);
+    if (expectedDiagnostic.autofix) {
+      assert.equal(actualDiagnostic.autofix, expectedDiagnostic.autofix);
+    }
   });
 }
diff --git a/client/src/test/extension.test.ts b/client/src/test/extension.test.ts
@@ -57,10 +57,7 @@ suite("Should execute command", () => {
 
     await activate(docUri);
 
-    const diagnostics = await commands.executeCommand<vscode.Diagnostic[]>(
-      "vscode.executeDiagnosticProvider",
-      docUri,
-    );
+    const diagnostics = vscode.languages.getDiagnostics(docUri);
 
     const autofixSuggestions = diagnostics
       .filter((diagnostic) => diagnostic.severity === vscode.DiagnosticSeverity.Error)