103 lines
2.2 KiB
Go
103 lines
2.2 KiB
Go
package gobot
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/gofrs/uuid"
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func TestRobotWork(t *testing.T) {
|
|
id, _ := uuid.NewV4()
|
|
|
|
rw := &RobotWork{
|
|
id: id,
|
|
kind: EveryWorkKind,
|
|
function: func() {},
|
|
}
|
|
|
|
duration := time.Second * 1
|
|
ctx, cancelFunc := context.WithCancel(context.Background())
|
|
|
|
rw.ctx = ctx
|
|
rw.cancelFunc = cancelFunc
|
|
rw.duration = duration
|
|
|
|
t.Run("ID()", func(t *testing.T) {
|
|
assert.Equal(t, rw.ID(), id)
|
|
})
|
|
|
|
t.Run("Ticker()", func(t *testing.T) {
|
|
t.Skip()
|
|
})
|
|
|
|
t.Run("Duration()", func(t *testing.T) {
|
|
assert.Equal(t, rw.duration, duration)
|
|
})
|
|
}
|
|
|
|
func TestRobotWorkRegistry(t *testing.T) {
|
|
robot := NewRobot("testbot")
|
|
|
|
rw := robot.Every(context.Background(), time.Millisecond*250, func() {
|
|
_ = 1 + 1
|
|
})
|
|
|
|
t.Run("Get retrieves", func(t *testing.T) {
|
|
assert.Equal(t, robot.workRegistry.Get(rw.id), rw)
|
|
})
|
|
|
|
t.Run("delete deletes", func(t *testing.T) {
|
|
robot.workRegistry.delete(rw.id)
|
|
postDeleteKeys := collectStringKeysFromWorkRegistry(robot.workRegistry)
|
|
assert.NotContains(t, postDeleteKeys, rw.id.String())
|
|
})
|
|
}
|
|
|
|
func TestRobotAutomationFunctions(t *testing.T) {
|
|
t.Run("Every with cancel", func(t *testing.T) {
|
|
robot := NewRobot("testbot")
|
|
counter := 0
|
|
|
|
rw := robot.Every(context.Background(), time.Millisecond*100, func() {
|
|
counter++
|
|
})
|
|
|
|
time.Sleep(time.Millisecond * 225)
|
|
rw.CallCancelFunc()
|
|
|
|
robot.WorkEveryWaitGroup.Wait()
|
|
|
|
assert.Equal(t, 2, counter)
|
|
postDeleteKeys := collectStringKeysFromWorkRegistry(robot.workRegistry)
|
|
assert.NotContains(t, postDeleteKeys, rw.id.String())
|
|
})
|
|
|
|
t.Run("After with cancel", func(t *testing.T) {
|
|
robot := NewRobot("testbot")
|
|
|
|
rw := robot.After(context.Background(), time.Millisecond*10, func() {
|
|
_ = 1 + 1 // perform mindless computation!
|
|
})
|
|
|
|
rw.CallCancelFunc()
|
|
|
|
robot.WorkAfterWaitGroup.Wait()
|
|
|
|
postDeleteKeys := collectStringKeysFromWorkRegistry(robot.workRegistry)
|
|
assert.NotContains(t, postDeleteKeys, rw.id.String())
|
|
})
|
|
}
|
|
|
|
func collectStringKeysFromWorkRegistry(rwr *RobotWorkRegistry) []string {
|
|
keys := make([]string, len(rwr.r))
|
|
var idx int
|
|
for key := range rwr.r {
|
|
keys[idx] = key
|
|
idx++
|
|
}
|
|
return keys
|
|
}
|