all: do not depend on the testing package

Discovered while looking for something else.

Updates tailscale/corp#18935

Signed-off-by: Maisem Ali <maisem@tailscale.com>
This commit is contained in:
Maisem Ali
2024-05-23 23:13:32 -07:00
committed by Brad Fitzpatrick
parent 4214e5f71b
commit 9a64c06a20
12 changed files with 20 additions and 9 deletions

View File

@@ -6,7 +6,6 @@ package syspolicy
import (
"errors"
"sync/atomic"
"testing"
)
var (
@@ -69,7 +68,14 @@ func RegisterHandler(h Handler) {
}
}
func SetHandlerForTest(tb testing.TB, h Handler) {
// TB is a subset of testing.TB that we use to set up test helpers.
// It's defined here to avoid pulling in the testing package.
type TB interface {
Helper()
Cleanup(func())
}
func SetHandlerForTest(tb TB, h Handler) {
tb.Helper()
oldHandler := handler
handler = h