Add completion test

kitagry · Jan 17, 2022 · bcdf576 · bcdf576
1 parent 2af4c6b
commit bcdf576
Showing 1 changed file with 28 additions and 0 deletions.
diff --git a/langserver/internal/source/completion_test.go b/langserver/internal/source/completion_test.go
@@ -61,6 +61,34 @@ import data.lib
 			},
 			expectItems: []source.CompletionItem{},
 		},
+		"completion for else": {
+			files: map[string]source.File{
+				"src.rego": {
+					RowText: `package src
+
+authorize = "allow" {
+	msg := "allow"
+	trace(msg)
+} else = "deny" {
+	ms := "deny"
+	ms
+}`,
+				},
+			},
+			location: &ast.Location{
+				Row: 8,
+				Col: 3,
+				Offset: len("package src\n\nauthorize = \"allow\" {\n	msg := \"allow\"\n	trace(msg)\n} else = \"deny\" {\n	ms := \"deny\"\n	ms"),
+				Text: []byte("s"),
+				File: "src.rego",
+			},
+			expectItems: []source.CompletionItem{
+				{
+					Label: "ms",
+					Kind:  source.VariableItem,
+				},
+			},
+		},
 	}
 
 	for n, tt := range tests {