Fix TestParallelStdout: Wait for process to terminate

kanisterio · Oct 25, 2024 · a6bfb90 · a6bfb90
1 parent 227a815
commit a6bfb90
Showing 1 changed file with 8 additions and 3 deletions.
diff --git a/pkg/kanx/kanx_test.go b/pkg/kanx/kanx_test.go
@@ -10,6 +10,7 @@ import (
 	"testing"
 	"time"
 
+	"github.com/kanisterio/kanister/pkg/poll"
 	"google.golang.org/grpc"
 	. "gopkg.in/check.v1"
 )
@@ -257,9 +258,13 @@ func (s *KanXSuite) TestParallelStdout(c *C) {
 	err = sp.cmd.Process.Kill()
 	c.Assert(err, IsNil)
 
-	ps, err = ListProcesses(ctx, addr)
-	c.Assert(err, IsNil)
-	c.Assert(ps, HasLen, 1)
+	_ = poll.Wait(ctx, func(context.Context) (bool, error) {
+		ps, err = ListProcesses(ctx, addr)
+		c.Assert(err, IsNil)
+		c.Assert(ps, HasLen, 1)
+		return ps[0].GetState() != ProcessState_PROCESS_STATE_RUNNING, nil
+	})
+
 	c.Assert(ps[0].GetPid(), Equals, p.GetPid())
 	c.Assert(ps[0].GetState(), Equals, ProcessState_PROCESS_STATE_FAILED)
 	c.Assert(ps[0].GetExitErr(), Equals, "signal: killed")