internal/lsp: fix TestCompatibleCodeActions

9fans · Apr 6, 2024 · 57f8560 · 57f8560
1 parent c4eff1c
commit 57f8560
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 1 deletion.
diff --git a/internal/lsp/utils.go b/internal/lsp/utils.go
@@ -35,7 +35,7 @@ func CompatibleCodeActions(cap *protocol.ServerCapabilities, kinds []protocol.Co
 		log.Printf("CompatibleCodeActions: unexpected CodeActionProvider type %T", ap)
 	case bool:
 		if ap {
-			allowed = kinds
+			return kinds
 		}
 		return nil
 	case protocol.CodeActionOptions:

diff --git a/internal/lsp/utils_test.go b/internal/lsp/utils_test.go
@@ -63,6 +63,19 @@ func TestCompatibleCodeActions(t *testing.T) {
 			[]protocol.CodeActionKind{protocol.SourceOrganizeImports},
 			[]protocol.CodeActionKind{protocol.SourceOrganizeImports},
 		},
+		{
+			"OneFoundMap",
+			protocol.ServerCapabilities{
+				CodeActionProvider: map[string]any{
+					"codeActionKinds": []any{
+						"quickfix",
+						"source.organizeImports",
+					},
+				},
+			},
+			[]protocol.CodeActionKind{protocol.SourceOrganizeImports},
+			[]protocol.CodeActionKind{protocol.SourceOrganizeImports},
+		},
 	} {
 		t.Run(tc.name, func(t *testing.T) {
 			got := CompatibleCodeActions(&tc.cap, tc.kinds)