diff --git a/pkg/agent/agent_test.go b/pkg/agent/agent_test.go index 670af6213..079cf8e02 100644 --- a/pkg/agent/agent_test.go +++ b/pkg/agent/agent_test.go @@ -106,7 +106,10 @@ func TestAgentLoadBalancing(t *testing.T) { agents := make([]*testutils.AgentWorker, totalWorkers) for i := 0; i < totalWorkers; i++ { - agents[i] = server.SimulateAgentWorker(testutils.WithLabel(fmt.Sprintf("agent-%d", i))) + agents[i] = server.SimulateAgentWorker( + testutils.WithLabel(fmt.Sprintf("agent-%d", i)), + testutils.WithJobLoad(testutils.NewStableJobLoad(0.01)), + ) agents[i].Register("test", livekit.JobType_JT_ROOM) } diff --git a/pkg/agent/testutils/server.go b/pkg/agent/testutils/server.go index a8d48e3c5..4178e0100 100644 --- a/pkg/agent/testutils/server.go +++ b/pkg/agent/testutils/server.go @@ -21,6 +21,7 @@ import ( "github.com/livekit/protocol/utils" "github.com/livekit/protocol/utils/guid" "github.com/livekit/protocol/utils/must" + "github.com/livekit/protocol/utils/options" "github.com/livekit/psrpc" ) @@ -103,9 +104,7 @@ func (h *TestServer) SimulateAgentWorker(opts ...SimulatedWorkerOption) *AgentWo HandleAvailability: func(r AgentJobRequest) { r.Accept() }, HandleAssignment: func(j *livekit.Job) JobLoad { return nil }, } - for _, opt := range opts { - opt(o) - } + options.Apply(o, opts) w := &AgentWorker{ workerMessages: make(chan *livekit.WorkerMessage, 1),