server: Fix selectOrchestrator test

livepeer · Nov 2, 2021 · fbcdccb · fbcdccb
1 parent 399714c
commit fbcdccb
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/server/mediaserver_test.go b/server/mediaserver_test.go
@@ -411,7 +411,7 @@ func TestSelectOrchestrator(t *testing.T) {
 	// Skip orchestrator if missing auth token
 	sd.infos[0].AuthToken = nil
 
-	sess, err = selectOrchestrator(s.LivepeerNode, sp, 4, newSuspender())
+	sess, err = selectOrchestrator(s.LivepeerNode, sp, 4, newSuspender(), func(float32) bool { return true })
 	require.Nil(err)
 
 	assert.Len(sess, 1)
@@ -421,7 +421,7 @@ func TestSelectOrchestrator(t *testing.T) {
 	sd.infos[0].AuthToken = &net.AuthToken{}
 	sd.infos[0].TicketParams = nil
 
-	sess, err = selectOrchestrator(s.LivepeerNode, sp, 4, newSuspender())
+	sess, err = selectOrchestrator(s.LivepeerNode, sp, 4, newSuspender(), func(float32) bool { return true })
 	require.Nil(err)
 
 	assert.Len(sess, 1)