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).
146 lines
2.6 KiB
Go
146 lines
2.6 KiB
Go
package collection_test
|
|
|
|
import (
|
|
"sync"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/hanzoai/tasks/common/collection"
|
|
)
|
|
|
|
// This isn't exhaustive but serves as a basic stress test to ensure our implementation is collection
|
|
func TestMap_MultiThreaded(t *testing.T) {
|
|
m := collection.NewSyncMap[int, int]()
|
|
var wg sync.WaitGroup
|
|
barrier := make(chan struct{})
|
|
wg.Add(5)
|
|
go func() {
|
|
defer wg.Done()
|
|
<-barrier
|
|
for i := range 1000 {
|
|
m.Set(i, i)
|
|
}
|
|
}()
|
|
go func() {
|
|
<-barrier
|
|
defer wg.Done()
|
|
for i := range 1000 {
|
|
m.Get(i)
|
|
}
|
|
}()
|
|
go func() {
|
|
<-barrier
|
|
defer wg.Done()
|
|
for i := range 1000 {
|
|
m.GetOrSet(i, i)
|
|
}
|
|
}()
|
|
go func() {
|
|
<-barrier
|
|
defer wg.Done()
|
|
for i := range 1000 {
|
|
m.Pop(i)
|
|
}
|
|
}()
|
|
go func() {
|
|
<-barrier
|
|
defer wg.Done()
|
|
for range 1000 {
|
|
m.PopAll()
|
|
}
|
|
}()
|
|
close(barrier)
|
|
wg.Wait()
|
|
}
|
|
|
|
func TestMap_Get(t *testing.T) {
|
|
m := collection.NewSyncMap[int, int]()
|
|
m.Set(1, 1)
|
|
v, ok := m.Get(1)
|
|
if !ok {
|
|
t.Error("Expected true, got false")
|
|
}
|
|
if v != 1 {
|
|
t.Errorf("Expected 1, got %v", v)
|
|
}
|
|
}
|
|
|
|
func TestMap_GetOrSet(t *testing.T) {
|
|
m := collection.NewSyncMap[int, int]()
|
|
m.Set(1, 1)
|
|
v, ok := m.GetOrSet(1, 2)
|
|
assert.True(t, ok, "expected exist key")
|
|
assert.Equal(t, 1, v, "expected the existing value")
|
|
|
|
v, ok = m.GetOrSet(2, 2)
|
|
assert.False(t, ok, "expected non exist key")
|
|
assert.Equal(t, 2, v, "expected the new set value")
|
|
|
|
v, ok = m.Get(2)
|
|
assert.True(t, ok, "expected exist key")
|
|
assert.Equal(t, 2, v, "expected the existing value")
|
|
}
|
|
|
|
func TestMap_Delete(t *testing.T) {
|
|
m := collection.NewSyncMap[int, int]()
|
|
m.Set(1, 1)
|
|
m.Set(2, 1)
|
|
m.Delete(1)
|
|
_, ok := m.Get(1)
|
|
if ok {
|
|
t.Error("Expected false, got true")
|
|
}
|
|
v, ok := m.Get(2)
|
|
if !ok {
|
|
t.Error("Expected true, got false")
|
|
}
|
|
if v != 1 {
|
|
t.Errorf("Expected 1, got %v", v)
|
|
}
|
|
}
|
|
|
|
func TestMap_Pop_ReturnsFalseWhenKeyDoesNotExist(t *testing.T) {
|
|
m := collection.NewSyncMap[int, int]()
|
|
_, ok := m.Pop(1)
|
|
if ok {
|
|
t.Error("Expected false, got true")
|
|
}
|
|
}
|
|
|
|
func TestMap_Pop_ReturnsTrueWhenKeyExists(t *testing.T) {
|
|
m := collection.NewSyncMap[int, int]()
|
|
m.Set(1, 1)
|
|
v, ok := m.Pop(1)
|
|
if !ok {
|
|
t.Error("Expected true, got false")
|
|
}
|
|
if v != 1 {
|
|
t.Errorf("Expected 1, got %v", v)
|
|
}
|
|
}
|
|
|
|
func TestMap_PopAll(t *testing.T) {
|
|
m := collection.NewSyncMap[int, int]()
|
|
values := m.PopAll()
|
|
assert.Equal(t, 0, len(values))
|
|
|
|
m.Set(1, 1)
|
|
m.Set(2, 2)
|
|
m.Set(3, 3)
|
|
m.Set(4, 4)
|
|
m.Pop(4)
|
|
|
|
mCopy := m
|
|
|
|
values = m.PopAll()
|
|
assert.Equal(t, 3, len(values))
|
|
sum := 0
|
|
for _, v := range values {
|
|
sum += v
|
|
}
|
|
assert.Equal(t, 6, sum)
|
|
|
|
_, ok := mCopy.Get(3)
|
|
assert.False(t, ok, "SyncMap is not correctly copyable")
|
|
}
|