status-go/server/timeout_test.go

package server

import (
	"crypto/rand"
	"math/big"
	"testing"
	"time"
)

func TestTimeoutManager(t *testing.T) {
	tm := newTimeoutManager()

	// test 0 timeout means timeout does not occur
	tm.SetTimeout(0)

	// test fuzzing - 0 timeout - multiple sequential calls to random init and stop funcs
	for i := 0; i < 30; i++ {
		b, err := rand.Int(rand.Reader, big.NewInt(2))
		if err != nil {
			t.Error(err)
		}

		if b.Int64() == 1 {
			tm.StartTimeout(t.FailNow)
		} else {
			tm.StopTimeout()
		}
	}

	// test fuzzing - random timeout - multiple sequential calls to random init and stop funcs
	for i := 0; i < 30; i++ {
		b, err := rand.Int(rand.Reader, big.NewInt(2))
		if err != nil {
			t.Error(err)
		}
		to, err := rand.Int(rand.Reader, big.NewInt(11))
		if err != nil {
			t.Error(err)
		}

		tm.SetTimeout(uint(to.Int64() * 20))

		if b.Int64() == 1 {
			tm.StartTimeout(t.FailNow)
		} else {
			tm.StopTimeout()
		}
		tm.StopTimeout()
	}

	// test StopTimeout() prevents termination func
	tm.SetTimeout(20)
	tm.StartTimeout(t.FailNow)
	time.Sleep(10 * time.Millisecond)
	tm.StopTimeout()

	// test StartTimeout() executes termination func on timeout
	ok := false
	tm.SetTimeout(10)
	tm.StartTimeout(func() {
		ok = true
	})
	time.Sleep(20 * time.Millisecond)
	if !ok {
		t.FailNow()
	}

}