influxdb/timer_test.go

package raft

import (
	"testing"
	"time"
)

//------------------------------------------------------------------------------
//
// Tests
//
//------------------------------------------------------------------------------

// Ensure that we can start an election timer and it will go off in the specified duration.
func TestTimer(t *testing.T) {

	timer := NewTimer(5*time.Millisecond, 10*time.Millisecond)

	// test timer start
	for i := 0; i < 10; i++ {

		start := time.Now()

		timer.Start()

		duration := time.Now().Sub(start)
		if duration > 12*time.Millisecond || duration < 5*time.Millisecond {
			t.Fatal("Duration Error! ", duration)
		}

	}

	// test timer stop
	for i := 0; i < 100; i++ {

		start := time.Now()

		go stop(timer)

		timer.Start()

		duration := time.Now().Sub(start)
		if duration > 3*time.Millisecond {
			t.Fatal("Duration Error! ", duration)
		}

		// ready the timer after stop it
		timer.Ready()

	}

	// test timer fire
	for i := 0; i < 100; i++ {

		start := time.Now()

		go fire(timer)

		timer.Start()

		duration := time.Now().Sub(start)
		if duration > 3*time.Millisecond {
			t.Fatal("Fire Duration Error! ", duration)
		}

	}

	resp := make(chan bool)
	// play with start and stop
	// make sure we can stop timer
	// in all the possible seq of start and stop
	for i := 0; i < 100; i++ {
		go stop(timer)
		go start(timer, resp)
		ret := <-resp
		if ret != false {
			t.Fatal("cannot stop timer!")
		}
		timer.Ready()
	}

}

func stop(t *Timer) {
	// sync
	time.Sleep(time.Millisecond)
	t.Stop()
}

func start(t *Timer, resp chan bool) {
	// sync
	time.Sleep(time.Millisecond)
	resp <- t.Start()
}

func fire(t *Timer) {
	// sync
	time.Sleep(time.Millisecond)
	t.Fire()
}