fix: write running status not none on the first step

internal/agent/agent.go

@@ -85,6 +85,9 @@ func (a *Agent) Status() *models.Status {
 		&a.graph.StartedAt,
 		&a.graph.FinishedAt,
 	)
+	if status.Status == scheduler.SchedulerStatus_None {
+		status.Status = scheduler.SchedulerStatus_Running
+	}
 	status.RequestId = a.requestId
 	status.Log = a.logFilename
 	if node := a.scheduler.HandlerNode(constants.OnExit); node != nil {

internal/agent/agent_test.go

@@ -30,13 +30,16 @@ func TestMain(m *testing.M) {
 }
 
 func TestRunDAG(t *testing.T) {
-	dag, err := controller.FromConfig(testConfig("agent_run.yaml"))
-	require.NoError(t, err)
-
-	status, err := testDAG(t, dag)
-	require.NoError(t, err)
+	_, dag := testDAGAsync(t, testConfig("agent_run.yaml"))
+	time.Sleep(100 * time.Millisecond)
+	status, _ := controller.New(dag.Config).GetLastStatus()
+	require.Equal(t, status.Status, scheduler.SchedulerStatus_Running)
 
-	assert.Equal(t, scheduler.SchedulerStatus_Success, status.Status)
+	require.Eventually(t, func() bool {
+		status, err := controller.New(dag.Config).GetLastStatus()
+		require.NoError(t, err)
+		return status.Status == scheduler.SchedulerStatus_Success
+	}, time.Second*2, time.Millisecond*100)
 }
 
 func TestCheckRunning(t *testing.T) {

tests/testdata/agent_run.yaml

@@ -1,4 +1,4 @@
 name: "test"
 steps:
   - name: "1"
-    command: "true"
+    command: "sleep 1"