add test for import code actions

DetachHead · Feb 14, 2024 · 4cf6872 · 4cf6872
1 parent 589bdfc
commit 4cf6872
Showing 1 changed file with 26 additions and 0 deletions.
diff --git a/packages/pyright-internal/src/tests/fourslash/importSuggestion.invokeCodeAction.fourslash.ts b/packages/pyright-internal/src/tests/fourslash/importSuggestion.invokeCodeAction.fourslash.ts
@@ -0,0 +1,26 @@
+/// <reference path="fourslash.ts" />
+
+// @filename: pyrightconfig.json
+//// {}
+
+// @filename: test.py
+//// [|/*import*/|]foo: TracebackType[|/*marker*/|]
+
+const importRange = helper.getPositionRange('import');
+
+//@ts-expect-error https://github.com/DetachHead/basedpyright/issues/86
+await helper.verifyCodeActions('included', {
+    marker: {
+        codeActions: [
+            {
+                title: `from types import TracebackType`,
+                edit: {
+                    changes: {
+                        'file:///test.py': [{ range: importRange, newText: 'from types import TracebackType\n\n\n' }],
+                    },
+                },
+                kind: 'quickfix',
+            },
+        ],
+    },
+});