test expression extension in a sub block

Author: jpogran

decoder/semantic_tokens_test.go

@@ -1404,3 +1404,234 @@ resource "vault_auth_backend" "blah" {
 		t.Fatalf("unexpected tokens: %s", diff)
 	}
 }
+
+func TestDecoder_SemanticTokensInFile_extensions_countIndexInSubBlock(t *testing.T) {
+	bodySchema := &schema.BodySchema{
+		Blocks: map[string]*schema.BlockSchema{
+			"resource": {
+				Body: &schema.BodySchema{
+					Extensions: &schema.BodyExtensions{
+						Count: true,
+					},
+					Attributes: map[string]*schema.AttributeSchema{
+						"count": {
+							Expr: schema.LiteralTypeOnly(cty.Number),
+						},
+					},
+					Blocks: map[string]*schema.BlockSchema{
+						"block": {
+							Body: &schema.BodySchema{
+								Attributes: map[string]*schema.AttributeSchema{
+									"attr": {
+										Expr: schema.LiteralTypeOnly(cty.Number),
+									},
+								},
+							},
+						},
+					},
+				},
+				Labels: []*schema.LabelSchema{
+					{
+						Name:     "type",
+						IsDepKey: true,
+						SemanticTokenModifiers: lang.SemanticTokenModifiers{
+							lang.TokenModifierDependent,
+						},
+					},
+					{Name: "name"},
+				},
+			},
+		},
+	}
+
+	testCfg := []byte(`
+resource "foobar" "name" {
+	count = 1
+	block {
+		attr = count.index
+	}
+}
+`)
+
+	f, pDiags := hclsyntax.ParseConfig(testCfg, "test.tf", hcl.InitialPos)
+	if len(pDiags) > 0 {
+		t.Fatal(pDiags)
+	}
+
+	d := testPathDecoder(t, &PathContext{
+		Schema: bodySchema,
+		Files: map[string]*hcl.File{
+			"test.tf": f,
+		},
+	})
+
+	ctx := context.Background()
+
+	tokens, err := d.SemanticTokensInFile(ctx, "test.tf")
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	expectedTokens := []lang.SemanticToken{
+		{ // resource
+			Type:      lang.TokenBlockType,
+			Modifiers: []lang.SemanticTokenModifier{},
+			Range: hcl.Range{
+				Filename: "test.tf",
+				Start: hcl.Pos{
+					Line:   2,
+					Column: 1,
+					Byte:   1,
+				},
+				End: hcl.Pos{
+					Line:   2,
+					Column: 9,
+					Byte:   9,
+				},
+			},
+		},
+		{ // foobar
+			Type: lang.TokenBlockLabel,
+			Modifiers: []lang.SemanticTokenModifier{
+				lang.TokenModifierDependent,
+			},
+			Range: hcl.Range{
+				Filename: "test.tf",
+				Start: hcl.Pos{
+					Line:   2,
+					Column: 10,
+					Byte:   10,
+				},
+				End: hcl.Pos{
+					Line:   2,
+					Column: 18,
+					Byte:   18,
+				},
+			},
+		},
+		{ // name
+			Type:      lang.TokenBlockLabel,
+			Modifiers: []lang.SemanticTokenModifier{},
+			Range: hcl.Range{
+				Filename: "test.tf",
+				Start: hcl.Pos{
+					Line:   2,
+					Column: 19,
+					Byte:   19,
+				},
+				End: hcl.Pos{
+					Line:   2,
+					Column: 25,
+					Byte:   25,
+				},
+			},
+		},
+		{ // count
+			Type:      lang.TokenAttrName,
+			Modifiers: lang.SemanticTokenModifiers{},
+			Range: hcl.Range{
+				Filename: "test.tf",
+				Start: hcl.Pos{
+					Line:   3,
+					Column: 2,
+					Byte:   29,
+				},
+				End: hcl.Pos{
+					Line:   3,
+					Column: 7,
+					Byte:   34,
+				},
+			},
+		},
+		{ // 1 number
+			Type:      lang.TokenNumber,
+			Modifiers: lang.SemanticTokenModifiers{},
+			Range: hcl.Range{
+				Filename: "test.tf",
+				Start: hcl.Pos{
+					Line:   3,
+					Column: 10,
+					Byte:   37,
+				},
+				End: hcl.Pos{
+					Line:   3,
+					Column: 11,
+					Byte:   38,
+				},
+			},
+		},
+		{ // block
+			Type:      lang.TokenBlockType,
+			Modifiers: lang.SemanticTokenModifiers{},
+			Range: hcl.Range{
+				Filename: "test.tf",
+				Start: hcl.Pos{
+					Line:   4,
+					Column: 2,
+					Byte:   40,
+				},
+				End: hcl.Pos{
+					Line:   4,
+					Column: 7,
+					Byte:   45,
+				},
+			},
+		},
+		{ // attr
+			Type:      lang.TokenAttrName,
+			Modifiers: lang.SemanticTokenModifiers{},
+			Range: hcl.Range{
+				Filename: "test.tf",
+				Start: hcl.Pos{
+					Line:   5,
+					Column: 3,
+					Byte:   50,
+				},
+				End: hcl.Pos{
+					Line:   5,
+					Column: 7,
+					Byte:   54,
+				},
+			},
+		},
+		{ // count
+			Type:      lang.TokenTraversalStep,
+			Modifiers: lang.SemanticTokenModifiers{},
+			Range: hcl.Range{
+				Filename: "test.tf",
+				Start: hcl.Pos{
+					Line:   5,
+					Column: 10,
+					Byte:   57,
+				},
+				End: hcl.Pos{
+					Line:   5,
+					Column: 15,
+					Byte:   62,
+				},
+			},
+		},
+		{ // index
+			Type:      lang.TokenTraversalStep,
+			Modifiers: lang.SemanticTokenModifiers{},
+			Range: hcl.Range{
+				Filename: "test.tf",
+				Start: hcl.Pos{
+					Line:   5,
+					Column: 16,
+					Byte:   63,
+				},
+				End: hcl.Pos{
+					Line:   5,
+					Column: 22,
+					Byte:   69,
+				},
+			},
+		},
+	}
+
+	diff := cmp.Diff(expectedTokens, tokens)
+	if diff != "" {
+		t.Fatalf("unexpected tokens: %s", diff)
+	}
+}