mirror of
https://github.com/tailscale/tailscale.git
synced 2025-12-01 07:22:04 +01:00
ac0b15356d
And fix up the TestAutoUpdateDefaults integration tests as they weren't testing reality: the DefaultAutoUpdate is supposed to only be relevant on the first MapResponse in the stream, but the tests weren't testing that. They were instead injecting a 2nd+ MapResponse. This changes the test control server to add a hook to modify the first map response, and then makes the test control when the node goes up and down to make new map responses. Also, the test now runs on macOS where the auto-update feature being disabled would've previously t.Skipped the whole test. Updates #11502 Change-Id: If2319bd1f71e108b57d79fe500b2acedbc76e1a6 Signed-off-by: Brad Fitzpatrick <bradfitz@tailscale.com>
111 lines
2.7 KiB
Go
111 lines
2.7 KiB
Go
// Copyright (c) Tailscale Inc & AUTHORS
|
|
// SPDX-License-Identifier: BSD-3-Clause
|
|
|
|
// Package feature tracks which features are linked into the binary.
|
|
package feature
|
|
|
|
import (
|
|
"errors"
|
|
"reflect"
|
|
|
|
"tailscale.com/util/testenv"
|
|
)
|
|
|
|
var ErrUnavailable = errors.New("feature not included in this build")
|
|
|
|
var in = map[string]bool{}
|
|
|
|
// Registered reports the set of registered features.
|
|
//
|
|
// The returned map should not be modified by the caller,
|
|
// not accessed concurrently with calls to Register.
|
|
func Registered() map[string]bool { return in }
|
|
|
|
// Register notes that the named feature is linked into the binary.
|
|
func Register(name string) {
|
|
if _, ok := in[name]; ok {
|
|
panic("duplicate feature registration for " + name)
|
|
}
|
|
in[name] = true
|
|
}
|
|
|
|
// Hook is a func that can only be set once.
|
|
//
|
|
// It is not safe for concurrent use.
|
|
type Hook[Func any] struct {
|
|
f Func
|
|
ok bool
|
|
}
|
|
|
|
// IsSet reports whether the hook has been set.
|
|
func (h *Hook[Func]) IsSet() bool {
|
|
return h.ok
|
|
}
|
|
|
|
// Set sets the hook function, panicking if it's already been set
|
|
// or f is the zero value.
|
|
//
|
|
// It's meant to be called in init.
|
|
func (h *Hook[Func]) Set(f Func) {
|
|
if h.ok {
|
|
panic("Set on already-set feature hook")
|
|
}
|
|
if reflect.ValueOf(f).IsZero() {
|
|
panic("Set with zero value")
|
|
}
|
|
h.f = f
|
|
h.ok = true
|
|
}
|
|
|
|
// SetForTest sets the hook function for tests, blowing
|
|
// away any previous value. It will panic if called from
|
|
// non-test code.
|
|
//
|
|
// It returns a restore function that resets the hook
|
|
// to its previous value.
|
|
func (h *Hook[Func]) SetForTest(f Func) (restore func()) {
|
|
testenv.AssertInTest()
|
|
old := *h
|
|
h.f, h.ok = f, true
|
|
return func() { *h = old }
|
|
}
|
|
|
|
// Get returns the hook function, or panics if it hasn't been set.
|
|
// Use IsSet to check if it's been set, or use GetOrNil if you're
|
|
// okay with a nil return value.
|
|
func (h *Hook[Func]) Get() Func {
|
|
if !h.ok {
|
|
panic("Get on unset feature hook, without IsSet")
|
|
}
|
|
return h.f
|
|
}
|
|
|
|
// GetOk returns the hook function and true if it has been set,
|
|
// otherwise its zero value and false.
|
|
func (h *Hook[Func]) GetOk() (f Func, ok bool) {
|
|
return h.f, h.ok
|
|
}
|
|
|
|
// GetOrNil returns the hook function or nil if it hasn't been set.
|
|
func (h *Hook[Func]) GetOrNil() Func {
|
|
return h.f
|
|
}
|
|
|
|
// Hooks is a slice of funcs.
|
|
//
|
|
// As opposed to a single Hook, this is meant to be used when
|
|
// multiple parties are able to install the same hook.
|
|
type Hooks[Func any] []Func
|
|
|
|
// Add adds a hook to the list of hooks.
|
|
//
|
|
// Add should only be called during early program
|
|
// startup before Tailscale has started.
|
|
// It is not safe for concurrent use.
|
|
func (h *Hooks[Func]) Add(f Func) {
|
|
if reflect.ValueOf(f).IsZero() {
|
|
panic("Add with zero value")
|
|
}
|
|
*h = append(*h, f)
|
|
}
|