fix incorrect reference to variable block

base_config.go

@@ -41,6 +41,9 @@ func (c *BaseConfig) EvalContext() *hcl.EvalContext {
 	}
 	for bt, bs := range c.blocksByTypes() {
 		sample := bs[0]
+		if s, ok := sample.(BlockCustomizedRefType); ok {
+			bt = s.CustomizedRefType()
+		}
 		if _, ok := sample.(SingleValueBlock); ok {
 			ctx.Variables[bt] = SingleValues(castBlock[SingleValueBlock](bs))
 			continue

block.go

@@ -13,6 +13,10 @@ import (
 	"strings"
 )
 
+type BlockCustomizedRefType interface {
+	CustomizedRefType() string
+}
+
 type BlockType interface {
 	BlockType() string
 }

config_test.go

@@ -337,3 +337,34 @@ func (s *configSuite) TestForEach_blocksWithIndexShouldHasNewBlockId() {
 	d1 := ruleBlocks[1].(Block)
 	s.NotEqual(d0.Id(), d1.Id())
 }
+
+func (s *configSuite) TestParseConfigWithVariable() {
+	content := `
+	variable "test" {
+		default = "hello"
+	}
+
+	data "dummy" sample {
+		data = {
+          key = var.test
+        }
+	}
+	`
+
+	s.dummyFsWithFiles(map[string]string{
+		"test.hcl": content,
+	})
+	t := s.T()
+
+	config, err := BuildDummyConfig("", "", nil, nil)
+	require.NoError(t, err)
+	_, err = RunDummyPlan(config)
+	require.NoError(t, err)
+	dataBlocks := Blocks[TestData](config)
+	assert.Len(t, dataBlocks, 1)
+	dummyData, ok := dataBlocks[0].(*DummyData)
+	require.True(t, ok)
+	assert.Equal(t, map[string]string{
+		"key": "hello",
+	}, dummyData.Tags)
+}

variable.go

@@ -17,6 +17,7 @@ var _ Variable = &VariableBlock{}
 var _ PrePlanBlock = &VariableBlock{}
 var _ PlanBlock = &VariableBlock{}
 var _ CustomDecode = &VariableBlock{}
+var _ BlockCustomizedRefType = &VariableBlock{}
 
 type Variable interface {
 	SingleValueBlock
@@ -52,6 +53,10 @@ func (v *VariableBlock) BlockType() string {
 	return "variable"
 }
 
+func (v *VariableBlock) CustomizedRefType() string {
+	return "var"
+}
+
 func (v *VariableBlock) AddressLength() int {
 	return 2
 }