Fix test

Signed-off-by: Justin Jung <jungjust@amazon.com>
cortexproject · Jul 10, 2024 · 9eaf463 · 9eaf463
1 parent e58b288
commit 9eaf463
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 5 deletions.
diff --git a/pkg/scheduler/queue/user_queues.go b/pkg/scheduler/queue/user_queues.go
@@ -433,17 +433,13 @@ type MockLimits struct {
 	MaxOutstanding        int
 	MaxQueriersPerUserVal float64
 	QueryPriorityVal      validation.QueryPriority
-	RandomMaxOutstanding  bool
 }
 
 func (l MockLimits) MaxQueriersPerUser(_ string) float64 {
 	return l.MaxQueriersPerUserVal
 }
 
 func (l MockLimits) MaxOutstandingPerTenant(_ string) int {
-	if l.RandomMaxOutstanding {
-		return rand.Intn(100)
-	}
 	return l.MaxOutstanding
 }
 

diff --git a/pkg/scheduler/queue/user_queues_test.go b/pkg/scheduler/queue/user_queues_test.go
@@ -461,7 +461,7 @@ func TestGetOrAddQueueShouldUpdateProperties(t *testing.T) {
 func TestGetOrAddQueueConcurrency(t *testing.T) {
 	const numGoRoutines = 100
 	limits := MockLimits{
-		RandomMaxOutstanding: true,
+		MaxOutstanding: 50,
 	}
 	q := newUserQueues(0, 0, limits, nil)
 	q.addQuerierConnection("q-1")