Function TestScheduler_ConcurrentStepNotify

pkg/core/scheduler_test.go:408–434

TestScheduler_ConcurrentStepNotify 测试并发步骤通知

(t *testing.T)

// TestScheduler_ConcurrentStepNotify 测试并发步骤通知
func TestScheduler_ConcurrentStepNotify(t *testing.T) {
	scheduler := NewScheduler(nil)
	defer scheduler.Shutdown()

	var callCount int32

	if _, err := scheduler.EverySteps(1, func(ctx context.Context, stepCount int) error {
		atomic.AddInt32(&callCount, 1)
		return nil
	}); err != nil {
		t.Fatalf("Failed to create step task: %v", err)
	}

	// 顺序通知 (避免 LastTriggered 的竞态问题)
	// 并发通知会导致某些步骤被跳过
	for i := 1; i <= 10; i++ {
		scheduler.NotifyStep(i)
	}

	time.Sleep(200 * time.Millisecond)

	// 验证所有步骤都被处理
	count := atomic.LoadInt32(&callCount)
	if count != 10 {
		t.Errorf("Expected 10 calls, got %d", count)
	}
}