Function TestNewAWSBeforeConnectSetsPassword

backend/store/dbauth/aws_test.go:170–192

(t *testing.T)

}

func TestNewAWSBeforeConnectSetsPassword(t *testing.T) {
	tokenProvider := &fakeTokenProvider{token: "generated-token"}
	hook := newAWSBeforeConnect(&awsConfig{
		endpoint: "example.us-east-1.rds.amazonaws.com:5432",
		region:   "us-east-1",
		user:     "bb_meta",
	}, tokenProvider)
	connConfig := &pgx.ConnConfig{}
	ctx := context.WithValue(context.Background(), contextKey{}, "marker")

	err := hook(ctx, connConfig)

	require.NoError(t, err)
	require.Equal(t, "generated-token", connConfig.Password)
	require.Equal(t, []fakeTokenProviderCall{
		{
			ctx:      ctx,
			endpoint: "example.us-east-1.rds.amazonaws.com:5432",
			region:   "us-east-1",
			user:     "bb_meta",
		},
	}, tokenProvider.calls)
}

func TestNewAWSBeforeConnectReturnsTokenError(t *testing.T) {
	tokenProvider := &fakeTokenProvider{err: errors.New("credential chain failed")}