mirror of
https://github.com/pomerium/pomerium.git
synced 2025-04-30 10:56:28 +02:00
526e2a58d6
* Initial test environment implementation * linter pass * wip: update request latency test * bugfixes * Fix logic race in envoy process monitor when canceling context * skip tests using test environment on non-linux
35 lines
1.1 KiB
Go
35 lines
1.1 KiB
Go
package snippets
|
|
|
|
import (
|
|
"context"
|
|
"time"
|
|
|
|
"github.com/pomerium/pomerium/internal/testenv"
|
|
"github.com/pomerium/pomerium/pkg/grpcutil"
|
|
"google.golang.org/grpc"
|
|
"google.golang.org/grpc/connectivity"
|
|
"google.golang.org/grpc/credentials/insecure"
|
|
)
|
|
|
|
func WaitStartupComplete(env testenv.Environment, timeout ...time.Duration) time.Duration {
|
|
start := time.Now()
|
|
recorder := env.NewLogRecorder()
|
|
if len(timeout) == 0 {
|
|
timeout = append(timeout, 1*time.Minute)
|
|
}
|
|
ctx, ca := context.WithTimeout(env.Context(), timeout[0])
|
|
defer ca()
|
|
recorder.WaitForMatch(map[string]any{
|
|
"syncer_id": "databroker",
|
|
"syncer_type": "type.googleapis.com/pomerium.config.Config",
|
|
"message": "listening for updates",
|
|
}, timeout...)
|
|
cc, err := grpc.Dial(env.DatabrokerURL().Value(),
|
|
grpc.WithTransportCredentials(insecure.NewCredentials()),
|
|
grpc.WithChainUnaryInterceptor(grpcutil.WithUnarySignedJWT(env.SharedSecret)),
|
|
grpc.WithChainStreamInterceptor(grpcutil.WithStreamSignedJWT(env.SharedSecret)),
|
|
)
|
|
env.Require().NoError(err)
|
|
env.Require().True(cc.WaitForStateChange(ctx, connectivity.Ready))
|
|
return time.Since(start)
|
|
}
|