add more test coverage

Author: muir

limit_test.go

@@ -17,13 +17,40 @@ const (
 	sleep       = time.Microsecond * 10
 )
 
-func TestLimit(t *testing.T) {
+func TestLimitSilent(t *testing.T) {
+	testLimit(t, false)
+}
+
+func TestLimitVerbose(t *testing.T) {
+	testLimit(t, true)
+}
+
+func testLimit(t *testing.T, withStuck bool) {
 	t.Parallel()
 
 	var running int32
 	var wg sync.WaitGroup
 	limit := simultaneous.New[any](max)
 
+	var stuckCalled atomic.Int32
+	var unstuckCalled atomic.Int32
+	someUnstuck := make(chan struct{})
+	noteStuckCalled := sync.OnceFunc(func() {
+		close(someUnstuck)
+	})
+
+	if withStuck {
+		limit = limit.SetForeverMessaging(time.Millisecond,
+			func() {
+				stuckCalled.Add(1)
+				noteStuckCalled()
+			},
+			func() {
+				unstuckCalled.Add(1)
+			},
+		)
+	}
+
 	for i := 0; i < threadCount; i++ {
 		i := i
 		wg.Add(1)
@@ -42,8 +69,17 @@ func TestLimit(t *testing.T) {
 			assert.LessOrEqual(t, atomic.AddInt32(&running, 1), int32(max))
 			time.Sleep(sleep)
 			assert.GreaterOrEqual(t, atomic.AddInt32(&running, -1), int32(0))
+			if withStuck {
+				<-someUnstuck
+			}
 			done.Done()
 		}()
 	}
 	wg.Wait()
+
+	if withStuck {
+		t.Logf("stuck called %d unstuck called %d", stuckCalled.Load(), unstuckCalled.Load())
+		assert.NotZero(t, stuckCalled.Load(), "stuck reported")
+		assert.Equal(t, stuckCalled.Load(), unstuckCalled.Load(), "stuck == unstuck")
+	}
 }