126 lines
2.3 KiB
Go
126 lines
2.3 KiB
Go
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))
|
|
}
|
|
}
|