94f2e54abe
Action, Query, QueryAll, Perform → Result QueryHandler, TaskHandler → func returning Result RegisterAction/RegisterActions → handler returns Result ServiceStartup, ServiceShutdown → Result LogError, LogWarn → Result ACTION, QUERY, QUERYALL, PERFORM aliases → Result Tests updated to match new signatures. Co-Authored-By: Virgil <virgil@lethean.io>
66 lines
1.2 KiB
Go
66 lines
1.2 KiB
Go
package core_test
|
|
|
|
import (
|
|
"sync"
|
|
"testing"
|
|
"time"
|
|
|
|
. "forge.lthn.ai/core/go/pkg/core"
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
// --- PerformAsync ---
|
|
|
|
func TestPerformAsync_Good(t *testing.T) {
|
|
c := New()
|
|
var mu sync.Mutex
|
|
var result string
|
|
|
|
c.RegisterTask(func(_ *Core, task Task) Result {
|
|
mu.Lock()
|
|
result = "done"
|
|
mu.Unlock()
|
|
return Result{Value: "completed", OK: true}
|
|
})
|
|
|
|
taskID := c.PerformAsync("work")
|
|
assert.NotEmpty(t, taskID)
|
|
|
|
time.Sleep(100 * time.Millisecond)
|
|
|
|
mu.Lock()
|
|
assert.Equal(t, "done", result)
|
|
mu.Unlock()
|
|
}
|
|
|
|
func TestPerformAsync_Progress_Good(t *testing.T) {
|
|
c := New()
|
|
c.RegisterTask(func(_ *Core, task Task) Result {
|
|
return Result{OK: true}
|
|
})
|
|
|
|
taskID := c.PerformAsync("work")
|
|
c.Progress(taskID, 0.5, "halfway", "work")
|
|
}
|
|
|
|
// --- RegisterAction + RegisterActions ---
|
|
|
|
func TestRegisterAction_Good(t *testing.T) {
|
|
c := New()
|
|
called := false
|
|
c.RegisterAction(func(_ *Core, _ Message) Result {
|
|
called = true
|
|
return Result{OK: true}
|
|
})
|
|
c.Action(nil)
|
|
assert.True(t, called)
|
|
}
|
|
|
|
func TestRegisterActions_Good(t *testing.T) {
|
|
c := New()
|
|
count := 0
|
|
h := func(_ *Core, _ Message) Result { count++; return Result{OK: true} }
|
|
c.RegisterActions(h, h)
|
|
c.Action(nil)
|
|
assert.Equal(t, 2, count)
|
|
}
|