mirror of
https://github.com/gravitational/teleport
synced 2024-10-20 01:03:40 +00:00
Fix Flaky TestDatabaseRootLeafIdleTimeout test (#18202)
This commit is contained in:
parent
cf426773c4
commit
a059d70ba8
|
@ -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)
|
||||
|
|
Loading…
Reference in a new issue