add more time before checking disconnects

SeldonIO · Nov 1, 2024 · ddae74f · ddae74f
1 parent 3235603
commit ddae74f
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/scheduler/pkg/agent/server_test.go b/scheduler/pkg/agent/server_test.go
@@ -1062,7 +1062,7 @@ func TestSubscribe(t *testing.T) {
 				}(idx, s)
 			}
 
-			time.Sleep(500 * time.Millisecond)
+			time.Sleep(1500 * time.Millisecond)
 
 			g.Expect(len(server.agents)).To(Equal(test.expectedAgentsCountAfterClose))
 

diff --git a/scheduler/pkg/kafka/dataflow/server_test.go b/scheduler/pkg/kafka/dataflow/server_test.go
@@ -737,7 +737,7 @@ func TestPipelineSubscribe(t *testing.T) {
 				}(a.id)
 			}
 
-			time.Sleep(500 * time.Millisecond)
+			time.Sleep(700 * time.Millisecond)
 
 			g.Expect(len(s.streams)).To(Equal(test.expectedAgentsCount))
 
@@ -749,7 +749,7 @@ func TestPipelineSubscribe(t *testing.T) {
 				}(idx, s)
 			}
 
-			time.Sleep(500 * time.Millisecond)
+			time.Sleep(1500 * time.Millisecond)
 
 			g.Expect(len(s.streams)).To(Equal(test.expectedAgentsCountAfterClose))