-
Notifications
You must be signed in to change notification settings - Fork 22
/
test_controls.go
41 lines (35 loc) · 1.24 KB
/
test_controls.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
// Copyright 2023 The Cockroach Authors
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
//
// SPDX-License-Identifier: Apache-2.0
package stdpool
import "context"
// TestControls allows certain behaviors to be tweaked for testing.
type TestControls struct {
// WaitForStartup makes certain connection or authentication errors
// result in a delay and retry, instead of failing.
WaitForStartup bool
}
// WithTestControls allows certain behaviors to be tweaked for testing.
func WithTestControls(tc TestControls) Option {
return &withTestControls{tc}
}
type withTestControls struct {
tc TestControls
}
func (o *withTestControls) option() {}
func (o *withTestControls) testControls(ctx context.Context, tc *TestControls) error {
*tc = o.tc
return nil
}