Files
tasks/service/history/queues/util_test.go
zeekay 5a601d081b rebrand: go.temporal.io/server -> github.com/hanzoai/tasks
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).
2026-03-19 10:36:40 -07:00

52 lines
1.5 KiB
Go

package queues
import (
"math/rand"
"testing"
"github.com/stretchr/testify/assert"
"github.com/hanzoai/tasks/common/definition"
"github.com/hanzoai/tasks/service/history/tasks"
"github.com/hanzoai/tasks/service/history/tests"
)
func TestIsTaskAcked(t *testing.T) {
scopes := NewRandomScopes(5)
exclusiveReaderHighWatermark := scopes[len(scopes)-1].Range.ExclusiveMax.Next()
persistenceQueueState := ToPersistenceQueueState(&queueState{
readerScopes: map[int64][]Scope{
DefaultReaderId: scopes,
},
exclusiveReaderHighWatermark: exclusiveReaderHighWatermark,
})
workflowKey := definition.NewWorkflowKey(tests.NamespaceID.String(), tests.WorkflowID, tests.RunID)
testKey := exclusiveReaderHighWatermark
testTask := tasks.NewFakeTask(
workflowKey,
tasks.CategoryTimer,
testKey.FireTime,
)
testTask.SetTaskID(testKey.TaskID)
assert.False(t, IsTaskAcked(testTask, persistenceQueueState))
testKey = NewRandomKeyInRange(scopes[rand.Intn(len(scopes))].Range)
testTask.SetVisibilityTime(testKey.FireTime)
testTask.SetTaskID(testKey.TaskID)
assert.False(t, IsTaskAcked(testTask, persistenceQueueState))
testKey = NewRandomKeyInRange(NewRange(
scopes[2].Range.ExclusiveMax,
scopes[3].Range.InclusiveMin,
))
testTask.SetVisibilityTime(testKey.FireTime)
testTask.SetTaskID(testKey.TaskID)
assert.True(t, IsTaskAcked(testTask, persistenceQueueState))
testKey = scopes[0].Range.InclusiveMin.Prev()
testTask.SetVisibilityTime(testKey.FireTime)
testTask.SetTaskID(testKey.TaskID)
assert.True(t, IsTaskAcked(testTask, persistenceQueueState))
}