tests: achive 100% coverage for go/atomic2

Signed-off-by: Manik Rana <[email protected]>
vitessio · Jan 17, 2024 · c648c7e · c648c7e
1 parent c1f9c80
commit c648c7e
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/go/atomic2/atomic128_test.go b/go/atomic2/atomic128_test.go
@@ -24,6 +24,17 @@ import (
 )
 
 func TestCompareAndSwap(t *testing.T) {
+	runCompareAndSwap(t)
+}
+
+func TestCompareAndSwapWriteBarrier(t *testing.T) {
+	writeBarrier.enabled = true
+	require.True(t, writeBarrier.enabled, "writeBarrier should be enabled")
+
+	runCompareAndSwap(t)
+}
+
+func runCompareAndSwap(t *testing.T) {
 	i1 := new(int)
 	i2 := new(int)
 	n := &PointerAndUint64[int]{p: unsafe.Pointer(i1), u: 12345}