wgengine: prevent log after exit in watchdog test

Signed-off-by: Dmytro Shynkevych <dmytro@tailscale.com>
This commit is contained in:
Dmytro Shynkevych 2020-07-03 03:00:45 -04:00 committed by Dave Anderson
parent 16b2bbbbbb
commit e9643ae724

View File

@ -24,6 +24,8 @@ func TestWatchdog(t *testing.T) {
e = NewWatchdog(e)
e.(*watchdogEngine).maxWait = 150 * time.Millisecond
e.(*watchdogEngine).logf = t.Logf
e.(*watchdogEngine).fatalf = t.Fatalf
e.RequestStatus()
e.RequestStatus()
@ -65,5 +67,9 @@ func TestWatchdog(t *testing.T) {
case <-time.After(3 * time.Second):
t.Fatalf("watchdog failed to fire")
}
usEngine.wgLock.Unlock()
wdEngine.fatalf = t.Fatalf
wdEngine.Close()
})
}