Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
1 change: 1 addition & 0 deletions .github/workflows/test.yml
Original file line number Diff line number Diff line change
Expand Up @@ -19,6 +19,7 @@ jobs:
- "1.22"
- "1.23"
- "1.24"
- "1.25"
- "1.x"
steps:
- uses: actions/checkout@v5
Expand Down
79 changes: 46 additions & 33 deletions future_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -2,19 +2,28 @@ package mo

import (
"fmt"
"sync"
"sync/atomic"
"testing"
"time"

"github.com/stretchr/testify/assert"
)

func assertAndIncrement(is *assert.Assertions, expected int, i *int32) {
got := atomic.LoadInt32(i)
type atomicInt32 struct {
v int32
mu sync.Mutex
}

is.Equal(int32(expected), got)
func (a *atomicInt32) Add(n int32) int32 {
a.mu.Lock()
defer a.mu.Unlock()
return atomic.AddInt32(&a.v, n)
}

atomic.AddInt32(i, 1)
func assertAndIncrement(t *testing.T, is *assert.Assertions, expected int, i *atomicInt32) {
got := i.Add(1)
is.Equal(int32(expected), got-1)
}

func TestFuture(t *testing.T) {
Expand Down Expand Up @@ -159,131 +168,135 @@ func TestFutureFinallyReject(t *testing.T) {
func TestFutureOrder(t *testing.T) {
is := assert.New(t)

var i int32 = 0
var i atomicInt32

_ = NewFuture[int](func(resolve func(int), reject func(error)) {
assertAndIncrement(is, 1, &i)
fut := NewFuture[int](func(resolve func(int), reject func(error)) {
assertAndIncrement(t, is, 1, &i)

resolve(42)
}).Then(func(value int) (int, error) {
assertAndIncrement(is, 2, &i)
assertAndIncrement(t, is, 2, &i)

return 21, assert.AnError
}).Catch(func(err error) (int, error) {
assertAndIncrement(is, 3, &i)
assertAndIncrement(t, is, 3, &i)

return 1, nil
}).Finally(func(value int, err error) (int, error) {
assertAndIncrement(is, 4, &i)
assertAndIncrement(t, is, 4, &i)

return 21, nil
})

assertAndIncrement(is, 0, &i)
assertAndIncrement(t, is, 0, &i)

_, _ = fut.Collect()

assertAndIncrement(t, is, 5, &i)
}

func TestFutureOrderCollect(t *testing.T) {
is := assert.New(t)

var i int32 = 0
var i atomicInt32

_, _ = NewFuture[int](func(resolve func(int), reject func(error)) {
assertAndIncrement(is, 0, &i)
assertAndIncrement(t, is, 0, &i)

resolve(42)
}).Then(func(value int) (int, error) {
assertAndIncrement(is, 1, &i)
assertAndIncrement(t, is, 1, &i)

return 21, assert.AnError
}).Catch(func(err error) (int, error) {
assertAndIncrement(is, 2, &i)
assertAndIncrement(t, is, 2, &i)

return 1, nil
}).Finally(func(value int, err error) (int, error) {
assertAndIncrement(is, 3, &i)
assertAndIncrement(t, is, 3, &i)

return 1, nil
}).Collect()

assertAndIncrement(is, 4, &i)
assertAndIncrement(t, is, 4, &i)
}

func TestFutureCancel(t *testing.T) {
is := assert.New(t)

var i int32 = 0
var i atomicInt32

future := NewFuture[int](func(resolve func(int), reject func(error)) {
assertAndIncrement(is, 0, &i)
assertAndIncrement(t, is, 0, &i)

time.Sleep(5 * time.Millisecond)

resolve(42)
}).Then(func(value int) (int, error) {
assertAndIncrement(is, 3, &i)
assertAndIncrement(t, is, 3, &i)
is.Fail("should not enter here")

return 21, assert.AnError
})

time.Sleep(1 * time.Millisecond)
assertAndIncrement(is, 1, &i)
assertAndIncrement(t, is, 1, &i)
future.Cancel()

time.Sleep(10 * time.Millisecond)
assertAndIncrement(is, 2, &i)
assertAndIncrement(t, is, 2, &i)
}

func TestFutureCancelDelayed(t *testing.T) {
is := assert.New(t)

var i int32 = 0
var i atomicInt32

future := NewFuture[int](func(resolve func(int), reject func(error)) {
time.Sleep(1 * time.Millisecond)
assertAndIncrement(is, 1, &i)
assertAndIncrement(t, is, 1, &i)

resolve(42)
}).Then(func(value int) (int, error) {
assertAndIncrement(is, 2, &i)
assertAndIncrement(t, is, 2, &i)

return 21, assert.AnError
})

assertAndIncrement(is, 0, &i)
assertAndIncrement(t, is, 0, &i)

time.Sleep(10 * time.Millisecond)

future.Cancel()

assertAndIncrement(is, 3, &i)
assertAndIncrement(t, is, 3, &i)
}

func TestFutureCancelTerminated(t *testing.T) {
is := assert.New(t)

var i int32 = 0
var i atomicInt32

future := NewFuture[int](func(resolve func(int), reject func(error)) {
time.Sleep(1 * time.Millisecond)
assertAndIncrement(is, 1, &i)
assertAndIncrement(t, is, 1, &i)

resolve(42)
}).Then(func(value int) (int, error) {
assertAndIncrement(is, 2, &i)
assertAndIncrement(t, is, 2, &i)

return 21, assert.AnError
})

assertAndIncrement(is, 0, &i)
assertAndIncrement(t, is, 0, &i)

_, _ = future.Collect()

assertAndIncrement(is, 3, &i)
assertAndIncrement(t, is, 3, &i)

future.Cancel()

assertAndIncrement(is, 4, &i)
assertAndIncrement(t, is, 4, &i)
}

func TestFutureResultResult(t *testing.T) {
Expand Down
26 changes: 24 additions & 2 deletions option_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -4,6 +4,8 @@ import (
"database/sql"
"encoding/json"
"fmt"
"runtime"
"strconv"
"strings"
"testing"

Expand Down Expand Up @@ -249,7 +251,7 @@ func TestOptionMarshalJSON(t *testing.T) {
is.Equal(`""`, string(value))

type testStruct struct {
Field Option[string]
Field Option[string]
OptionalField Option[string] `json:"optional_field,omitzero"`
}

Expand All @@ -259,7 +261,27 @@ func TestOptionMarshalJSON(t *testing.T) {
var marshalled []byte
marshalled, err = json.Marshal(optionInStruct)
is.NoError(err)
is.Equal(`{"Field":"foo"}`, string(marshalled))

major, minor, _ := decomposeGoVersion()
if major >= 1 && minor >= 24 {
is.Equal(`{"Field":"foo"}`, string(marshalled))
} else {
is.Equal(`{"Field":"foo","optional_field":null}`, string(marshalled))
}
}

func decomposeGoVersion() (int, int, int) {
version := runtime.Version() // e.g., "go1.21.0"

version = strings.TrimPrefix(version, "go")

// Split major, minor version, and patch
parts := strings.Split(version, ".")
major, _ := strconv.Atoi(parts[0])
minor, _ := strconv.Atoi(parts[1])
patch, _ := strconv.Atoi(parts[2])

return major, minor, patch
}

func TestOptionUnmarshalJSON(t *testing.T) {
Expand Down
Loading