updated SafeWrap tests

This commit is contained in:
Gani Georgiev committed 2026-07-16 08:42:17 +03:00
1 parent 6d6484c466
commit 794f1d5a8e
2 files changed
+44 -16

No files matched your search

+1 -1
View File
@@ -203,7 +203,7 @@ func (c *Cron) runDue(t time.Time) {
for _, j := range c.jobs { for _, j := range c.jobs {
if j.schedule.IsDue(moment) { if j.schedule.IsDue(moment) {
go j.run() routine.FireAndForget(j.run)
} }
} }
} }
+43 -15
View File
@@ -1,6 +1,7 @@
package routine_test package routine_test
import ( import (
"errors"
"strings" "strings"
"sync" "sync"
"testing" "testing"
@@ -28,24 +29,51 @@ func TestFireAndForget(t *testing.T) {
} }
func TestSafeWrap(t *testing.T) { func TestSafeWrap(t *testing.T) {
called := false t.Run("panic", func(t *testing.T) {
called := false
fn := func() error { fn := func() error {
called = true called = true
panic("test_recover") panic("test_recover")
} }
err := routine.SafeWrap(fn)() err := routine.SafeWrap(fn)()
if !called { if !called {
t.Fatal("Expected fn to be called.") t.Fatal("Expected fn to be called.")
} }
if err == nil { if err == nil {
t.Fatal("Expected fn panic to be converted to error") t.Fatal("Expected fn panic to be converted to error")
} }
if !strings.Contains(err.Error(), "test_recover") { if !strings.Contains(err.Error(), "test_recover") {
t.Fatal("Expected the returned error to contain the recovered panic value") t.Fatal("Expected the returned error to contain the recovered panic value")
} }
})
t.Run("regular error", func(t *testing.T) {
called := false
fn := func() error {
called = true
return errors.New("test_error")
}
err := routine.SafeWrap(fn)()
if !called {
t.Fatal("Expected fn to be called.")
}
if err == nil {
t.Fatal("Expected to return the wrapped fn error")
}
errStr := err.Error()
expected := "test_error"
if errStr != expected {
t.Fatalf("Expected %s error, got %s", expected, errStr)
}
})
} }