hybridgroup.gobot/robot_work_test.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
}