TestGoroutine

teacat · Sep 7, 2020 · 575e79b · 575e79b
1 parent c5cb54c
commit 575e79b
Showing 1 changed file with 34 additions and 0 deletions.
diff --git a/trillium_test.go b/trillium_test.go
@@ -29,6 +29,40 @@ func TestTimeout(t *testing.T) {
 	}
 }
 
+func TestGoroutine(t *testing.T) {
+	assert := assert.New(t)
+	tri := New(DefaultConfig())
+
+	done := make(chan bool, 2)
+	go func() {
+		var i int
+		for {
+			if i > 512 {
+				done <- true
+				break
+			}
+			i++
+
+			_, err := tri.Generate()
+			assert.NoError(err)
+		}
+	}()
+	go func() {
+		var i int
+		for {
+			if i > 512 {
+				done <- true
+				break
+			}
+			i++
+
+			_, err := tri.Generate()
+			assert.NoError(err)
+		}
+	}()
+	<-done
+}
+
 func BenchmarkUint64(b *testing.B) {
 	t := New(DefaultConfig())
 	b.ReportAllocs()