fix race tests

benchmark_test.go

@@ -5,6 +5,7 @@ package konf_test
 
 import (
 	"context"
+	"sync/atomic"
 	"testing"
 
 	"github.com/ktong/konf"
@@ -55,15 +56,13 @@ func BenchmarkWatch(b *testing.B) {
 	}()
 	b.ResetTimer()
 
+	var cfg atomic.Value
+	konf.OnChange(func() {
+		cfg.Store(konf.Get[string]("config"))
+	})
 	for i := 0; i < b.N; i++ {
 		watcher.change(map[string]any{"config": "changed"})
 	}
 	b.StopTimer()
-
-	var cfg string
-	konf.OnChange(func() {
-		assert.NoError(b, konf.Unmarshal("config", &cfg))
-	})
-	watcher.change(map[string]any{"config": "changed"})
-	assert.Equal(b, "changed", cfg)
+	assert.Equal(b, "changed", cfg.Load())
 }

config_test.go

@@ -6,6 +6,7 @@ package konf_test
 import (
 	"context"
 	"errors"
+	"sync/atomic"
 	"testing"
 	"time"
 
@@ -167,11 +168,14 @@ func TestConfig_Watch(t *testing.T) {
 		assert.NoError(t, config.Watch(ctx))
 	}()
 
+	var newCfg atomic.Value
 	config.OnChange(func(unmarshaler konf.Unmarshaler) {
+		var cfg string
 		assert.NoError(t, config.Unmarshal("config", &cfg))
+		newCfg.Store(cfg)
 	})
 	watcher.change(map[string]any{"config": "changed"})
-	assert.Equal(t, "changed", cfg)
+	assert.Equal(t, "changed", newCfg.Load())
 }
 
 type mapWatcher chan map[string]any

global_test.go

@@ -7,6 +7,7 @@ import (
 	"bytes"
 	"context"
 	"log"
+	"sync/atomic"
 	"testing"
 
 	"github.com/ktong/konf"
@@ -64,10 +65,10 @@ func TestWatch(t *testing.T) {
 		assert.NoError(t, konf.Watch(ctx))
 	}()
 
-	var cfg string
+	var cfg atomic.Value
 	konf.OnChange(func() {
-		assert.NoError(t, konf.Unmarshal("config", &cfg))
+		cfg.Store(konf.Get[string]("config"))
 	})
 	watcher.change(map[string]any{"config": "changed"})
-	assert.Equal(t, "changed", cfg)
+	assert.Equal(t, "changed", cfg.Load())
 }