From 4674e3f759f57dc9bbfe53d1210abfe3a05450ab Mon Sep 17 00:00:00 2001
From: Matt Calhoun <matt@cloudposse.com>
Date: Wed, 29 Jan 2025 14:11:34 -0500
Subject: [PATCH] set COMPONENT_HELPER_STATE_DIR in test helper (#48)

---
 pkg/atmos/component-helper/atmos_options.go | 7 +++++--
 1 file changed, 5 insertions(+), 2 deletions(-)

diff --git a/pkg/atmos/component-helper/atmos_options.go b/pkg/atmos/component-helper/atmos_options.go
index e98b5e9..2c08b0b 100644
--- a/pkg/atmos/component-helper/atmos_options.go
+++ b/pkg/atmos/component-helper/atmos_options.go
@@ -28,8 +28,11 @@ func getAtmosOptions(t *testing.T, config *c.Config, componentName string, stack
 		BackendConfig: map[string]interface{}{
 			"workspace_key_prefix": strings.Join([]string{config.RandomIdentifier, stackName}, "-"),
 		},
-		Vars:    mergedVars,
-		EnvVars: map[string]string{"ATMOS_BASE_PATH": config.TempDir},
+		Vars: mergedVars,
+		EnvVars: map[string]string{
+			"ATMOS_BASE_PATH":            config.TempDir,
+			"COMPONENT_HELPER_STATE_DIR": config.StateDir,
+		},
 	}
 	return atmosOptions
 }