fix: Fixing TestDiscoveryWithSingleCustomConfigFilename

Author: yhakbar

internal/component/unit.go

@@ -24,6 +24,7 @@ type Unit struct {
 	discoveryContext *DiscoveryContext
 	Execution        *UnitExecution
 	path             string
+	configFile       string
 	reading          []string
 	dependencies     Components
 	dependents       Components
@@ -45,6 +46,7 @@ type UnitExecution struct {
 func NewUnit(path string) *Unit {
 	return &Unit{
 		path:             path,
+		configFile:       config.DefaultTerragruntConfigPath,
 		discoveryContext: &DiscoveryContext{},
 		dependencies:     make(Components, 0),
 		dependents:       make(Components, 0),
@@ -83,6 +85,16 @@ func (u *Unit) StoreConfig(cfg *config.TerragruntConfig) {
 	u.cfg = cfg
 }
 
+// ConfigFile returns the discovered config filename for this unit.
+func (u *Unit) ConfigFile() string {
+	return u.configFile
+}
+
+// SetConfigFile sets the discovered config filename for this unit.
+func (u *Unit) SetConfigFile(filename string) {
+	u.configFile = filename
+}
+
 // Kind returns the kind of component (always Unit for Unit).
 func (u *Unit) Kind() Kind {
 	return UnitKind

internal/discovery/discovery.go

@@ -373,6 +373,12 @@ func Parse(
 	case *component.Stack:
 		configFilename = config.DefaultStackFile
 	default:
+		if unit, ok := c.(*component.Unit); ok && unit.ConfigFile() != "" {
+			configFilename = unit.ConfigFile()
+
+			break
+		}
+
 		if opts.TerragruntConfigPath != "" && !util.IsDir(opts.TerragruntConfigPath) {
 			configFilename = filepath.Base(opts.TerragruntConfigPath)
 		}
@@ -691,11 +697,10 @@ func (d *Discovery) processFile(
 		// If the directory is hidden, allow it only if it matches an include pattern
 		allowHidden := false
 
-		if canErr == nil {
-			// Always allow .terragrunt-stack contents
-			cleanDir := util.CleanPath(canonicalDir)
-			allowHidden = isInStackDirectory(cleanDir)
-		}
+		// Always allow .terragrunt-stack contents. We can safely use the non-canonical
+		// dir here because this branch is only reached when canonical path computation
+		// failed, or when we couldn't early-return with a concrete component.
+		allowHidden = isInStackDirectory(filepath.ToSlash(dir))
 
 		if !allowHidden {
 			return nil
@@ -725,6 +730,9 @@ func (d *Discovery) createComponentFromPath(path string, filenames []string) com
 		}
 
 		c := componentOfBase(dir, base)
+		if unit, ok := c.(*component.Unit); ok {
+			unit.SetConfigFile(base)
+		}
 
 		if d.discoveryContext != nil {
 			c.SetDiscoveryContext(d.discoveryContext)