Fix tests

n8n-io · Dec 9, 2024 · 67c4321 · 67c4321
1 parent 3e385e1
commit 67c4321
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/internal/env/env_test.go b/internal/env/env_test.go
@@ -158,6 +158,7 @@ func TestPrepareRunnerEnv(t *testing.T) {
 			name: "includes default and allowed env vars",
 			config: &config.Config{
 				AutoShutdownTimeout: "15",
+				TaskTimeout:         "60",
 				Runner: &config.RunnerConfig{
 					AllowedEnv: []string{"CUSTOM_VAR1", "CUSTOM_VAR2"},
 				},
@@ -177,6 +178,7 @@ func TestPrepareRunnerEnv(t *testing.T) {
 				"LANG=en_US.UTF-8",
 				"N8N_RUNNERS_AUTO_SHUTDOWN_TIMEOUT=15",
 				"N8N_RUNNERS_HEALTH_CHECK_SERVER_ENABLED=true",
+				"N8N_RUNNERS_TASK_TIMEOUT=60",
 				"PATH=/usr/bin",
 				"TERM=xterm",
 				"TZ=UTC",
@@ -186,6 +188,7 @@ func TestPrepareRunnerEnv(t *testing.T) {
 			name: "handles empty allowed env list",
 			config: &config.Config{
 				AutoShutdownTimeout: "15",
+				TaskTimeout:         "60",
 				Runner: &config.RunnerConfig{
 					AllowedEnv: []string{},
 				},
@@ -199,13 +202,15 @@ func TestPrepareRunnerEnv(t *testing.T) {
 				"LANG=en_US.UTF-8",
 				"N8N_RUNNERS_AUTO_SHUTDOWN_TIMEOUT=15",
 				"N8N_RUNNERS_HEALTH_CHECK_SERVER_ENABLED=true",
+				"N8N_RUNNERS_TASK_TIMEOUT=60",
 				"PATH=/usr/bin",
 			},
 		},
 		{
 			name: "handles custom auto-shutdown timeout",
 			config: &config.Config{
 				AutoShutdownTimeout: "30",
+				TaskTimeout:         "60",
 				Runner: &config.RunnerConfig{
 					AllowedEnv: []string{},
 				},
@@ -217,6 +222,7 @@ func TestPrepareRunnerEnv(t *testing.T) {
 			expected: []string{
 				"N8N_RUNNERS_AUTO_SHUTDOWN_TIMEOUT=30",
 				"N8N_RUNNERS_HEALTH_CHECK_SERVER_ENABLED=true",
+				"N8N_RUNNERS_TASK_TIMEOUT=60",
 				"PATH=/usr/bin",
 			},
 		},