51 lines
1.1 KiB
Go
51 lines
1.1 KiB
Go
|
package atomicx_test
|
||
|
|
|
||
|
|
import (
|
||
|
|
"testing"
|
||
|
|
"time"
|
||
|
|
|
||
|
|
"github.com/ooni/probe-cli/v3/internal/engine/atomicx"
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestInt64(t *testing.T) {
|
||
|
|
// TODO(bassosimone): how to write tests with race conditions
|
||
|
|
// and be confident that they're WAI? Here I hope this test is
|
||
|
|
// run with `-race` and I'm doing something that AFAICT will
|
||
|
|
// be flagged as race if we were not be using mutexes.
|
||
|
|
v := atomicx.NewInt64()
|
||
|
|
go func() {
|
||
|
|
v.Add(17)
|
||
|
|
}()
|
||
|
|
go func() {
|
||
|
|
v.Add(14)
|
||
|
|
}()
|
||
|
|
time.Sleep(1 * time.Second)
|
||
|
|
if v.Add(3) != 34 {
|
||
|
|
t.Fatal("unexpected result")
|
||
|
|
}
|
||
|
|
if v.Load() != 34 {
|
||
|
|
t.Fatal("unexpected result")
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestFloat64(t *testing.T) {
|
||
|
|
// TODO(bassosimone): how to write tests with race conditions
|
||
|
|
// and be confident that they're WAI? Here I hope this test is
|
||
|
|
// run with `-race` and I'm doing something that AFAICT will
|
||
|
|
// be flagged as race if we were not be using mutexes.
|
||
|
|
v := atomicx.NewFloat64()
|
||
|
|
go func() {
|
||
|
|
v.Add(17.0)
|
||
|
|
}()
|
||
|
|
go func() {
|
||
|
|
v.Add(14.0)
|
||
|
|
}()
|
||
|
|
time.Sleep(1 * time.Second)
|
||
|
|
if r := v.Add(3); r < 33.9 && r > 34.1 {
|
||
|
|
t.Fatal("unexpected result")
|
||
|
|
}
|
||
|
|
if v.Load() < 33.9 && v.Load() > 34.1 {
|
||
|
|
t.Fatal("unexpected result")
|
||
|
|
}
|
||
|
|
}
|