Method TestTaskQueueRateLimit

tests/task_queue_test.go:59–67

Not using RunTestWithMatchingBehavior because I want to pass different expected drain times for different configurations

()

// Not using RunTestWithMatchingBehavior because I want to pass different expected drain times for different configurations
func (s *TaskQueueSuite) TestTaskQueueRateLimit() {
	s.runTaskQueueRateLimitTest(1, 1, 12*time.Second, true)  // ~0.75s avg
	s.runTaskQueueRateLimitTest(1, 1, 12*time.Second, false) // ~1.1s avg

	// Testing multiple partitions with insufficient pollers is too flaky, because token recycling
	// depends on a process being available to accept the token, so I'm not testing it
	s.runTaskQueueRateLimitTest(4, 8, 24*time.Second, true)  // ~1.6s avg
	s.runTaskQueueRateLimitTest(4, 8, 24*time.Second, false) // ~6s avg
}

func (s *TaskQueueSuite) runTaskQueueRateLimitTest(nPartitions, nWorkers int, timeToDrain time.Duration, useNewMatching bool) {
	s.Run(s.testTaskQueueRateLimitName(nPartitions, nWorkers, useNewMatching), func(s *TaskQueueSuite) {