Attempt to fix flaky TestConnection

Signed-off-by: Rohit Nayak <rohit@planetscale.com>
vitessio · Oct 15, 2020 · e98892b · e98892b
1 parent dbd7ab6
commit e98892b
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/go/test/endtoend/messaging/message_test.go b/go/test/endtoend/messaging/message_test.go
@@ -331,6 +331,7 @@ func TestConnection(t *testing.T) {
  _, err = stream.MessageStream(userKeyspace, "", nil, name)
  require.Nil(t, err)
  // validate client count of vttablet
+ time.Sleep(time.Second)
  assert.Equal(t, 1, getClientCount(shard0Master))
  assert.Equal(t, 1, getClientCount(shard1Master))
  // second connection with vtgate, secont connection
@@ -340,6 +341,7 @@ func TestConnection(t *testing.T) {
  _, err = stream1.MessageStream(userKeyspace, "", nil, name)
  require.Nil(t, err)
  // validate client count of vttablet
+ time.Sleep(time.Second)
  assert.Equal(t, 2, getClientCount(shard0Master))
  assert.Equal(t, 2, getClientCount(shard1Master))