chore: Wiring expansion into the unit and stack parse (#6694)

Author: yhakbar

pkg/config/expansion_test.go

@@ -38,7 +38,7 @@ unit "app" {
   }
 
   source = "./modules/app"
-  path   = "app"
+  path   = "app/${each.key}"
 }
 `
 
@@ -55,7 +55,7 @@ stack "team" {
   }
 
   source = "./stacks/team"
-  path   = "team"
+  path   = "team/${count.index}"
 }
 `
 
@@ -633,29 +633,31 @@ bogus "x" {
 	require.Error(t, err)
 }
 
-func TestUnitAndStackDecodeExpansionBlock(t *testing.T) {
+func TestUnitAndStackExpandPerIterationElement(t *testing.T) {
 	t.Parallel()
 
 	stackCfg, err := parseStackString(t, unitWithExpansionHCL+stackWithExpansionHCL)
 	require.NoError(t, err)
-	require.Len(t, stackCfg.Units, 1)
-	require.Len(t, stackCfg.Stacks, 1)
-
-	unitExpansion := stackCfg.Units[0].Expansion
-	require.NotNil(t, unitExpansion)
-	require.NotNil(t, unitExpansion.ForEach)
-	assert.Equal(
-		t,
-		cty.SetVal([]cty.Value{cty.StringVal("web"), cty.StringVal("api")}),
-		*unitExpansion.ForEach,
-	)
+	require.Len(t, stackCfg.Units, 2)
+	require.Len(t, stackCfg.Stacks, 2)
+
+	unitPaths := map[string]string{}
+
+	for _, unit := range stackCfg.Units {
+		require.NotNil(t, unit.Expansion)
+		unitPaths[unit.Expansion.Key()] = unit.Path
+	}
+
+	assert.Equal(t, map[string]string{"web": "app/web", "api": "app/api"}, unitPaths)
+
+	stackPaths := map[string]string{}
+
+	for _, stack := range stackCfg.Stacks {
+		require.NotNil(t, stack.Expansion)
+		stackPaths[stack.Expansion.Key()] = stack.Path
+	}
 
-	stackExpansion := stackCfg.Stacks[0].Expansion
-	require.NotNil(t, stackExpansion)
-	require.NotNil(t, stackExpansion.Count)
-	// Decoding and the literal build the number at different big.Float precisions,
-	// which assert.Equal reads as a diff.
-	assert.True(t, stackExpansion.Count.RawEquals(cty.NumberIntVal(2)))
+	assert.Equal(t, map[string]string{"0": "team/0", "1": "team/1"}, stackPaths)
 }
 
 func TestBlocksWithoutExpansionDecodeUnchanged(t *testing.T) {