package gobot import ( "fmt" "testing" "time" "github.com/hybridgroup/gobot/gobottest" ) func TestEvery(t *testing.T) { i := 0 begin := time.Now().UnixNano() sem := make(chan int64, 1) Every(2*time.Millisecond, func() { i++ if i == 2 { sem <- time.Now().UnixNano() } }) end := <-sem if end-begin < 4000000 { t.Error("Test should have taken at least 4 milliseconds") } } func TestEveryWhenDone(t *testing.T) { i := 0 done := Every(20*time.Millisecond, func() { i++ }) <-time.After(10 * time.Millisecond) done <- true <-time.After(50 * time.Millisecond) if i > 1 { t.Error("Test should have stopped after 20ms") } } func TestAfter(t *testing.T) { i := 0 After(1*time.Millisecond, func() { i++ }) <-time.After(2 * time.Millisecond) gobottest.Assert(t, i, 1) } func TestPublish(t *testing.T) { c := make(chan interface{}, 1) cb := callback{ f: func(val interface{}) { c <- val }, } e := &Event{Callbacks: []callback{cb}} Publish(e, 1) <-time.After(10 * time.Millisecond) Publish(e, 2) Publish(e, 3) Publish(e, 4) i := <-c gobottest.Assert(t, i, 1) var e1 = (*Event)(nil) gobottest.Assert(t, Publish(e1, 4), ErrUnknownEvent) } func TestOn(t *testing.T) { var i int e := NewEvent() On(e, func(data interface{}) { i = data.(int) }) Publish(e, 10) <-time.After(1 * time.Millisecond) gobottest.Assert(t, i, 10) var e1 = (*Event)(nil) err := On(e1, func(data interface{}) { i = data.(int) }) gobottest.Assert(t, err, ErrUnknownEvent) } func TestOnce(t *testing.T) { i := 0 e := NewEvent() Once(e, func(data interface{}) { i += data.(int) }) On(e, func(data interface{}) { i += data.(int) }) Publish(e, 10) <-time.After(1 * time.Millisecond) Publish(e, 10) <-time.After(1 * time.Millisecond) gobottest.Assert(t, i, 30) var e1 = (*Event)(nil) err := Once(e1, func(data interface{}) { i = data.(int) }) gobottest.Assert(t, err, ErrUnknownEvent) } func TestFromScale(t *testing.T) { gobottest.Assert(t, FromScale(5, 0, 10), 0.5) } func TestToScale(t *testing.T) { gobottest.Assert(t, ToScale(500, 0, 10), 10.0) gobottest.Assert(t, ToScale(-1, 0, 10), 0.0) gobottest.Assert(t, ToScale(0.5, 0, 10), 5.0) } func TestRand(t *testing.T) { a := Rand(1000) b := Rand(1000) if a == b { t.Error(fmt.Sprintf("%v should not equal %v", a, b)) } }