mirror of
https://github.com/tailscale/tailscale.git
synced 2024-11-25 19:15:34 +00:00
71029cea2d
This updates all source files to use a new standard header for copyright and license declaration. Notably, copyright no longer includes a date, and we now use the standard SPDX-License-Identifier header. This commit was done almost entirely mechanically with perl, and then some minimal manual fixes. Updates #6865 Signed-off-by: Will Norris <will@tailscale.com>
79 lines
2.1 KiB
Go
79 lines
2.1 KiB
Go
// Copyright (c) Tailscale Inc & AUTHORS
|
|
// SPDX-License-Identifier: BSD-3-Clause
|
|
|
|
package syncs
|
|
|
|
import (
|
|
"context"
|
|
"sync"
|
|
"testing"
|
|
"time"
|
|
|
|
"tailscale.com/util/cibuild"
|
|
)
|
|
|
|
// Time-based tests are fundamentally flaky.
|
|
// We use exaggerated durations in the hopes of minimizing such issues.
|
|
|
|
func TestWatchUncontended(t *testing.T) {
|
|
mu := new(sync.Mutex)
|
|
ctx, cancel := context.WithCancel(context.Background())
|
|
defer cancel()
|
|
// Once an hour, and now, check whether we can lock mu in under an hour.
|
|
tick := time.Hour
|
|
max := time.Hour
|
|
c := Watch(ctx, mu, tick, max)
|
|
d := <-c
|
|
if d == max {
|
|
t.Errorf("uncontended mutex did not lock in under %v", max)
|
|
}
|
|
}
|
|
|
|
func TestWatchContended(t *testing.T) {
|
|
mu := new(sync.Mutex)
|
|
ctx, cancel := context.WithCancel(context.Background())
|
|
defer cancel()
|
|
// Every hour, and now, check whether we can lock mu in under a millisecond,
|
|
// which is enough time for an uncontended mutex by several orders of magnitude.
|
|
tick := time.Hour
|
|
max := time.Millisecond
|
|
mu.Lock()
|
|
defer mu.Unlock()
|
|
c := Watch(ctx, mu, tick, max)
|
|
d := <-c
|
|
if d != max {
|
|
t.Errorf("contended mutex locked in under %v", max)
|
|
}
|
|
}
|
|
|
|
func TestWatchMultipleValues(t *testing.T) {
|
|
if cibuild.On() {
|
|
// On the CI machine, it sometimes takes 500ms to start a new goroutine.
|
|
// When this happens, we don't get enough events quickly enough.
|
|
// Nothing's wrong, and it's not worth working around. Just skip the test.
|
|
t.Skip("flaky on CI")
|
|
}
|
|
mu := new(sync.Mutex)
|
|
ctx, cancel := context.WithCancel(context.Background())
|
|
defer cancel() // not necessary, but keep vet happy
|
|
// Check the mutex every millisecond.
|
|
// The goal is to see that we get a sufficient number of values out of the channel.
|
|
tick := time.Millisecond
|
|
max := time.Millisecond
|
|
c := Watch(ctx, mu, tick, max)
|
|
start := time.Now()
|
|
n := 0
|
|
for d := range c {
|
|
n++
|
|
if d == max {
|
|
t.Errorf("uncontended mutex did not lock in under %v", max)
|
|
}
|
|
if n == 10 {
|
|
cancel()
|
|
}
|
|
}
|
|
if elapsed := time.Since(start); elapsed > 100*time.Millisecond {
|
|
t.Errorf("expected 1 event per millisecond, got only %v events in %v", n, elapsed)
|
|
}
|
|
}
|