mirror of
https://github.com/hanzoai/tasks.git
synced 2026-11-03 03:56:04 +00:00
zeekay
5a601d081b
Full module path rebrand from Temporal to Hanzo Tasks: - go.mod module: github.com/hanzoai/tasks - All internal Go imports updated (1906 files) - cmd/server renamed to cmd/tasksd - Binary name: tasksd (was temporal-server) - Docker images: hanzoai/tasks (was temporalio/server) - README rewritten with Hanzo Tasks branding - LLM.md created with integration notes - Makefile, goreleaser, CI workflows updated - go.temporal.io/sdk and go.temporal.io/api imports preserved (separate repos) Fork of Temporal.io (MIT license).
70 lines
1.8 KiB
Go
70 lines
1.8 KiB
Go
package effect_test
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
"github.com/hanzoai/tasks/common/effect"
|
|
)
|
|
|
|
func TestBufferApplyOrder(t *testing.T) {
|
|
var buf effect.Buffer
|
|
state := make([]int, 0, 3)
|
|
|
|
buf.OnAfterCommit(func(context.Context) { state = append(state, 0) })
|
|
buf.OnAfterCommit(func(context.Context) { state = append(state, 1) })
|
|
buf.OnAfterCommit(func(context.Context) { state = append(state, 2) })
|
|
|
|
buf.Apply(context.TODO())
|
|
|
|
require.ElementsMatch(t, []int{0, 1, 2}, state)
|
|
}
|
|
|
|
func TestBufferRollbackOrder(t *testing.T) {
|
|
var buf effect.Buffer
|
|
state := make([]int, 0, 3)
|
|
|
|
buf.OnAfterRollback(func(context.Context) { state = append(state, 0) })
|
|
buf.OnAfterRollback(func(context.Context) { state = append(state, 1) })
|
|
buf.OnAfterRollback(func(context.Context) { state = append(state, 2) })
|
|
|
|
buf.Cancel(context.TODO())
|
|
|
|
require.ElementsMatch(t, []int{2, 1, 0}, state)
|
|
}
|
|
|
|
func TestBufferCancelAfterApply(t *testing.T) {
|
|
var buf effect.Buffer
|
|
var commit, rollback int
|
|
buf.OnAfterCommit(func(context.Context) { commit++ })
|
|
buf.OnAfterRollback(func(context.Context) { rollback++ })
|
|
|
|
buf.Apply(context.TODO())
|
|
buf.Cancel(context.TODO())
|
|
buf.Apply(context.TODO())
|
|
buf.Cancel(context.TODO())
|
|
buf.Apply(context.TODO())
|
|
buf.Cancel(context.TODO())
|
|
|
|
require.Equal(t, commit, 1)
|
|
require.Equal(t, rollback, 0)
|
|
}
|
|
|
|
func TestBufferApplyAfterCancel(t *testing.T) {
|
|
var buf effect.Buffer
|
|
var commit, rollback int
|
|
buf.OnAfterCommit(func(context.Context) { commit++ })
|
|
buf.OnAfterRollback(func(context.Context) { rollback++ })
|
|
|
|
buf.Cancel(context.TODO())
|
|
buf.Apply(context.TODO())
|
|
buf.Apply(context.TODO())
|
|
buf.Cancel(context.TODO())
|
|
buf.Apply(context.TODO())
|
|
buf.Cancel(context.TODO())
|
|
|
|
require.Equal(t, commit, 0)
|
|
require.Equal(t, rollback, 1)
|
|
}
|