initial import

core/syncx/timeoutlimit_test.go

@@ -0,0 +1,33 @@
+package syncx
+
+import (
+	"sync"
+	"testing"
+	"time"
+
+	"github.com/stretchr/testify/assert"
+)
+
+func TestTimeoutLimit(t *testing.T) {
+	limit := NewTimeoutLimit(2)
+	assert.Nil(t, limit.Borrow(time.Millisecond*200))
+	assert.Nil(t, limit.Borrow(time.Millisecond*200))
+	var wait1, wait2, wait3 sync.WaitGroup
+	wait1.Add(1)
+	wait2.Add(1)
+	wait3.Add(1)
+	go func() {
+		wait1.Wait()
+		wait2.Done()
+		assert.Nil(t, limit.Return())
+		wait3.Done()
+	}()
+	wait1.Done()
+	wait2.Wait()
+	assert.Nil(t, limit.Borrow(time.Second))
+	wait3.Wait()
+	assert.Equal(t, ErrTimeout, limit.Borrow(time.Millisecond*100))
+	assert.Nil(t, limit.Return())
+	assert.Nil(t, limit.Return())
+	assert.Equal(t, ErrReturn, limit.Return())
+}