mirror of
https://github.com/lightninglabs/loop
synced 2024-11-13 13:10:30 +00:00
35 lines
485 B
Go
35 lines
485 B
Go
package test
|
|
|
|
import (
|
|
"os"
|
|
"runtime/pprof"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/fortytw2/leaktest"
|
|
)
|
|
|
|
// Guard implements a test level timeout.
|
|
func Guard(t *testing.T) func() {
|
|
done := make(chan struct{})
|
|
go func() {
|
|
select {
|
|
case <-time.After(5 * time.Second):
|
|
err := pprof.Lookup("goroutine").WriteTo(os.Stdout, 1)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
panic("test timeout")
|
|
case <-done:
|
|
}
|
|
}()
|
|
|
|
fn := leaktest.Check(t)
|
|
|
|
return func() {
|
|
close(done)
|
|
fn()
|
|
}
|
|
}
|