with-timeout_test.go

// WithTimeout runs a function and fails if it didn't finish within the timeout
func (s *Suite) WithTimeout(timeout time.Duration, fn func()) {
	ch := make(chan struct{})
	go func() {
		fn()
		ch <- struct{}{}
	}()

	select {
	case <-time.After(timeout):
		s.Fail("The test stuck for more than expected timeout")
	case <-ch:
		return
	}
}