comment in

dagu-org · Feb 12, 2025 · f3751dc · f3751dc
1 parent b7b0e82
commit f3751dc
Showing 1 changed file with 33 additions and 33 deletions.
diff --git a/internal/digraph/scheduler/scheduler_test.go b/internal/digraph/scheduler/scheduler_test.go
@@ -317,39 +317,39 @@ func TestScheduler(t *testing.T) {
 		node := result.Node(t, "1")
 		require.Equal(t, 2, node.State().RetryCount) // 2 retry
 	})
-	// t.Run("RetryWithScript", func(t *testing.T) {
-	// 	sc := setup(t)
-
-	// 	const testEnv = "TEST_RETRY_WITH_SCRIPT"
-	// 	graph := sc.newGraph(t,
-	// 		newStep("1",
-	// 			withScript(`
-	// 				if [ "$TEST_RETRY_WITH_SCRIPT" -eq 1 ]; then
-	// 					exit 1
-	// 				fi
-	// 				exit 0
-	// 			`),
-	// 			withRetryPolicy(1, time.Millisecond*500),
-	// 		),
-	// 	)
-
-	// 	_ = os.Setenv(testEnv, "1")
-	// 	go func() {
-	// 		time.Sleep(time.Millisecond * 300)
-	// 		_ = os.Setenv(testEnv, "0")
-	// 		t.Cleanup(func() {
-	// 			_ = os.Unsetenv(testEnv)
-	// 		})
-	// 	}()
-
-	// 	result := graph.Schedule(t, scheduler.StatusSuccess)
-
-	// 	result.AssertNodeStatus(t, "1", scheduler.NodeStatusSuccess)
-
-	// 	node := result.Node(t, "1")
-	// 	require.Equal(t, 2, node.State().DoneCount)  // 2 executions
-	// 	require.Equal(t, 1, node.State().RetryCount) // 1 retry
-	// })
+	t.Run("RetryWithScript", func(t *testing.T) {
+		sc := setup(t)
+
+		const testEnv = "TEST_RETRY_WITH_SCRIPT"
+		graph := sc.newGraph(t,
+			newStep("1",
+				withScript(`
+					if [ "$TEST_RETRY_WITH_SCRIPT" -eq 1 ]; then
+						exit 1
+					fi
+					exit 0
+				`),
+				withRetryPolicy(1, time.Millisecond*500),
+			),
+		)
+
+		_ = os.Setenv(testEnv, "1")
+		go func() {
+			time.Sleep(time.Millisecond * 300)
+			_ = os.Setenv(testEnv, "0")
+			t.Cleanup(func() {
+				_ = os.Unsetenv(testEnv)
+			})
+		}()
+
+		result := graph.Schedule(t, scheduler.StatusSuccess)
+
+		result.AssertNodeStatus(t, "1", scheduler.NodeStatusSuccess)
+
+		node := result.Node(t, "1")
+		require.Equal(t, 2, node.State().DoneCount)  // 2 executions
+		require.Equal(t, 1, node.State().RetryCount) // 1 retry
+	})
 	t.Run("RetryPolicySuccess", func(t *testing.T) {
 		file := filepath.Join(
 			os.TempDir(), fmt.Sprintf("flag_test_retry_success_%s", uuid.Must(uuid.NewRandom()).String()),