Fix Flaky TestDatabaseRootLeafIdleTimeout test (#18202)

This commit is contained in:
Marek Smoliński 2022-11-14 10:09:33 +01:00 committed by GitHub
parent cf426773c4
commit a059d70ba8
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23

View file

@ -28,6 +28,7 @@ import (
"github.com/gravitational/trace"
"github.com/jackc/pgconn"
"github.com/jonboulle/clockwork"
"github.com/stretchr/testify/assert"
"github.com/stretchr/testify/require"
"go.mongodb.org/mongo-driver/bson"
@ -563,6 +564,13 @@ func TestDatabaseRootLeafIdleTimeout(t *testing.T) {
t.Run("root role with idle timeout", func(t *testing.T) {
setRoleIdleTimeout(t, rootAuthServer, rootRole, idleTimeout)
require.Eventually(t, func() bool {
role, err := rootAuthServer.GetRole(context.Background(), rootRole.GetName())
assert.NoError(t, err)
return time.Duration(role.GetOptions().ClientIdleTimeout) == idleTimeout
}, time.Second, time.Millisecond*100, "role idle timeout propagation filed")
client := mkMySQLLeafDBClient(t)
_, err := client.Execute("select 1")
require.NoError(t, err)
@ -578,6 +586,13 @@ func TestDatabaseRootLeafIdleTimeout(t *testing.T) {
t.Run("leaf role with idle timeout", func(t *testing.T) {
setRoleIdleTimeout(t, leafAuthServer, leafRole, idleTimeout)
require.Eventually(t, func() bool {
role, err := leafAuthServer.GetRole(context.Background(), leafRole.GetName())
assert.NoError(t, err)
return time.Duration(role.GetOptions().ClientIdleTimeout) == idleTimeout
}, time.Second, time.Millisecond*100, "role idle timeout propagation filed")
client := mkMySQLLeafDBClient(t)
_, err := client.Execute("select 1")
require.NoError(t, err)