updated SafeWrap tests

This commit is contained in:
Gani Georgiev
2026-07-16 08:42:17 +03:00
parent 6d6484c466
commit 794f1d5a8e
2 changed files with 44 additions and 16 deletions
+1 -1
View File
@@ -203,7 +203,7 @@ func (c *Cron) runDue(t time.Time) {
for _, j := range c.jobs {
if j.schedule.IsDue(moment) {
go j.run()
routine.FireAndForget(j.run)
}
}
}
+43 -15
View File
@@ -1,6 +1,7 @@
package routine_test
import (
"errors"
"strings"
"sync"
"testing"
@@ -28,24 +29,51 @@ func TestFireAndForget(t *testing.T) {
}
func TestSafeWrap(t *testing.T) {
called := false
t.Run("panic", func(t *testing.T) {
called := false
fn := func() error {
called = true
panic("test_recover")
}
fn := func() error {
called = true
panic("test_recover")
}
err := routine.SafeWrap(fn)()
err := routine.SafeWrap(fn)()
if !called {
t.Fatal("Expected fn to be called.")
}
if !called {
t.Fatal("Expected fn to be called.")
}
if err == nil {
t.Fatal("Expected fn panic to be converted to error")
}
if err == nil {
t.Fatal("Expected fn panic to be converted to error")
}
if !strings.Contains(err.Error(), "test_recover") {
t.Fatal("Expected the returned error to contain the recovered panic value")
}
if !strings.Contains(err.Error(), "test_recover") {
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)
}
})
}