From d116709dde3e71d6f6f0159e3774fc4ecbb62a7b Mon Sep 17 00:00:00 2001
From: lukechampine <luke.champine@gmail.com>
Date: Fri, 30 Sep 2016 02:13:33 -0400
Subject: broadcast in AtomicSet

---
 stm_test.go | 19 +++++++++++++++++++
 1 file changed, 19 insertions(+)

(limited to 'stm_test.go')

diff --git a/stm_test.go b/stm_test.go
index b6ecb16..1b1b737 100644
--- a/stm_test.go
+++ b/stm_test.go
@@ -199,6 +199,25 @@ func TestReadWritten(t *testing.T) {
 	})
 }
 
+func TestAtomicSetRetry(t *testing.T) {
+	// AtomicSet should cause waiting transactions to retry
+	x := NewVar(3)
+	done := make(chan struct{})
+	go func() {
+		Atomically(func(tx *Tx) {
+			tx.Assert(tx.Get(x).(int) == 5)
+		})
+		done <- struct{}{}
+	}()
+	time.Sleep(10 * time.Millisecond)
+	AtomicSet(x, 5)
+	select {
+	case <-done:
+	case <-time.After(time.Second):
+		t.Fatal("AtomicSet did not wake up a waiting transaction")
+	}
+}
+
 func BenchmarkAtomicGet(b *testing.B) {
 	x := NewVar(0)
 	for i := 0; i < b.N; i++ {
-- 
cgit v1.2.3