@@ -14,7 +14,6 @@ import (
1414 "github.com/lavanet/lava/v5/protocol/common"
1515 "github.com/lavanet/lava/v5/protocol/lavaprotocol"
1616 "github.com/lavanet/lava/v5/protocol/lavasession"
17- "github.com/lavanet/lava/v5/protocol/qos"
1817 pairingtypes "github.com/lavanet/lava/v5/x/pairing/types"
1918 spectypes "github.com/lavanet/lava/v5/x/spec/types"
2019 "github.com/stretchr/testify/require"
@@ -95,7 +94,7 @@ func TestRelayProcessorHappyFlow(t *testing.T) {
9594 protocolMessage := chainlib .NewProtocolMessage (chainMsg , nil , nil , dappId , consumerIp )
9695 consistency := NewConsistency (specId )
9796 usedProviders := lavasession .NewUsedProviders (nil )
98- relayProcessor := NewRelayProcessor (ctx , common .DefaultCrossValidationParams , consistency , RelayProcessorMetrics , RelayProcessorMetrics , RelayRetriesManagerInstance , newMockRelayStateMachine (protocolMessage , usedProviders ), qos . NewQoSManager () )
97+ relayProcessor := NewRelayProcessor (ctx , common .DefaultCrossValidationParams , consistency , RelayProcessorMetrics , RelayProcessorMetrics , RelayRetriesManagerInstance , newMockRelayStateMachine (protocolMessage , usedProviders ))
9998
10099 ctx , cancel := context .WithTimeout (context .Background (), time .Millisecond * 10 )
101100 defer cancel ()
@@ -151,7 +150,7 @@ func TestRelayProcessorTimeout(t *testing.T) {
151150 require .NoError (t , err )
152151 protocolMessage := chainlib .NewProtocolMessage (chainMsg , nil , nil , "" , "" )
153152 usedProviders := lavasession .NewUsedProviders (nil )
154- relayProcessor := NewRelayProcessor (ctx , common .DefaultCrossValidationParams , nil , RelayProcessorMetrics , RelayProcessorMetrics , RelayRetriesManagerInstance , newMockRelayStateMachine (protocolMessage , usedProviders ), qos . NewQoSManager () )
153+ relayProcessor := NewRelayProcessor (ctx , common .DefaultCrossValidationParams , nil , RelayProcessorMetrics , RelayProcessorMetrics , RelayRetriesManagerInstance , newMockRelayStateMachine (protocolMessage , usedProviders ))
155154
156155 ctx , cancel := context .WithTimeout (context .Background (), time .Millisecond * 10 )
157156 defer cancel ()
@@ -204,7 +203,7 @@ func TestRelayProcessorRetry(t *testing.T) {
204203 require .NoError (t , err )
205204 protocolMessage := chainlib .NewProtocolMessage (chainMsg , nil , nil , "" , "" )
206205 usedProviders := lavasession .NewUsedProviders (nil )
207- relayProcessor := NewRelayProcessor (ctx , common .DefaultCrossValidationParams , nil , RelayProcessorMetrics , RelayProcessorMetrics , RelayRetriesManagerInstance , newMockRelayStateMachine (protocolMessage , usedProviders ), qos . NewQoSManager () )
206+ relayProcessor := NewRelayProcessor (ctx , common .DefaultCrossValidationParams , nil , RelayProcessorMetrics , RelayProcessorMetrics , RelayRetriesManagerInstance , newMockRelayStateMachine (protocolMessage , usedProviders ))
208207
209208 ctx , cancel := context .WithTimeout (context .Background (), time .Millisecond * 10 )
210209 defer cancel ()
@@ -249,7 +248,7 @@ func TestRelayProcessorRetryNodeError(t *testing.T) {
249248 require .NoError (t , err )
250249 protocolMessage := chainlib .NewProtocolMessage (chainMsg , nil , nil , "" , "" )
251250 usedProviders := lavasession .NewUsedProviders (nil )
252- relayProcessor := NewRelayProcessor (ctx , common .DefaultCrossValidationParams , nil , RelayProcessorMetrics , RelayProcessorMetrics , RelayRetriesManagerInstance , newMockRelayStateMachine (protocolMessage , usedProviders ), qos . NewQoSManager () )
251+ relayProcessor := NewRelayProcessor (ctx , common .DefaultCrossValidationParams , nil , RelayProcessorMetrics , RelayProcessorMetrics , RelayRetriesManagerInstance , newMockRelayStateMachine (protocolMessage , usedProviders ))
253252
254253 ctx , cancel := context .WithTimeout (context .Background (), time .Millisecond * 10 )
255254 defer cancel ()
@@ -297,7 +296,7 @@ func TestRelayProcessorStatefulApi(t *testing.T) {
297296 require .NoError (t , err )
298297 protocolMessage := chainlib .NewProtocolMessage (chainMsg , nil , nil , "" , "" )
299298 usedProviders := lavasession .NewUsedProviders (nil )
300- relayProcessor := NewRelayProcessor (ctx , common .DefaultCrossValidationParams , nil , RelayProcessorMetrics , RelayProcessorMetrics , RelayRetriesManagerInstance , newMockRelayStateMachine (protocolMessage , usedProviders ), qos . NewQoSManager () )
299+ relayProcessor := NewRelayProcessor (ctx , common .DefaultCrossValidationParams , nil , RelayProcessorMetrics , RelayProcessorMetrics , RelayRetriesManagerInstance , newMockRelayStateMachine (protocolMessage , usedProviders ))
301300 ctx , cancel := context .WithTimeout (context .Background (), time .Millisecond * 10 )
302301 defer cancel ()
303302 canUse := usedProviders .TryLockSelection (ctx )
@@ -352,7 +351,7 @@ func TestRelayProcessorStatefulApiErr(t *testing.T) {
352351 require .NoError (t , err )
353352 protocolMessage := chainlib .NewProtocolMessage (chainMsg , nil , nil , "" , "" )
354353 usedProviders := lavasession .NewUsedProviders (nil )
355- relayProcessor := NewRelayProcessor (ctx , common .DefaultCrossValidationParams , nil , RelayProcessorMetrics , RelayProcessorMetrics , RelayRetriesManagerInstance , newMockRelayStateMachine (protocolMessage , usedProviders ), qos . NewQoSManager () )
354+ relayProcessor := NewRelayProcessor (ctx , common .DefaultCrossValidationParams , nil , RelayProcessorMetrics , RelayProcessorMetrics , RelayRetriesManagerInstance , newMockRelayStateMachine (protocolMessage , usedProviders ))
356355 ctx , cancel := context .WithTimeout (context .Background (), time .Millisecond * 10 )
357356 defer cancel ()
358357 canUse := usedProviders .TryLockSelection (ctx )
@@ -400,7 +399,7 @@ func TestRelayProcessorLatest(t *testing.T) {
400399 require .NoError (t , err )
401400 protocolMessage := chainlib .NewProtocolMessage (chainMsg , nil , nil , "" , "" )
402401 usedProviders := lavasession .NewUsedProviders (nil )
403- relayProcessor := NewRelayProcessor (ctx , common .DefaultCrossValidationParams , nil , RelayProcessorMetrics , RelayProcessorMetrics , RelayRetriesManagerInstance , newMockRelayStateMachine (protocolMessage , usedProviders ), qos . NewQoSManager () )
402+ relayProcessor := NewRelayProcessor (ctx , common .DefaultCrossValidationParams , nil , RelayProcessorMetrics , RelayProcessorMetrics , RelayRetriesManagerInstance , newMockRelayStateMachine (protocolMessage , usedProviders ))
404403 ctx , cancel := context .WithTimeout (context .Background (), time .Millisecond * 10 )
405404 defer cancel ()
406405 canUse := usedProviders .TryLockSelection (ctx )
@@ -551,7 +550,6 @@ func TestHasRequiredNodeResultsCrossValidationScenarios(t *testing.T) {
551550 RelayProcessorMetrics ,
552551 RelayRetriesManagerInstance ,
553552 newMockRelayStateMachineWithSelection (protocolMessage , usedProviders , Stateless ),
554- qos .NewQoSManager (),
555553 )
556554
557555 // Set the batch size so WaitForResults knows how many responses to expect
@@ -637,7 +635,6 @@ func TestNodeErrorCrossValidationMet(t *testing.T) {
637635 RelayProcessorMetrics ,
638636 RelayRetriesManagerInstance ,
639637 newMockRelayStateMachineWithSelection (protocolMessage , usedProviders , Stateless ),
640- qos .NewQoSManager (),
641638 )
642639
643640 ctx , cancel := context .WithTimeout (context .Background (), time .Millisecond * 10 )
@@ -700,7 +697,6 @@ func TestNodeErrorCrossValidationNotMet(t *testing.T) {
700697 RelayProcessorMetrics ,
701698 RelayRetriesManagerInstance ,
702699 newMockRelayStateMachineWithSelection (protocolMessage , usedProviders , Stateless ),
703- qos .NewQoSManager (),
704700 )
705701
706702 ctx , cancel := context .WithTimeout (context .Background (), time .Millisecond * 10 )
@@ -763,7 +759,6 @@ func TestNodeErrorCrossValidationWithProtocolErrors(t *testing.T) {
763759 RelayProcessorMetrics ,
764760 RelayRetriesManagerInstance ,
765761 newMockRelayStateMachineWithSelection (protocolMessage , usedProviders , Stateless ),
766- qos .NewQoSManager (),
767762 )
768763
769764 ctx , cancel := context .WithTimeout (context .Background (), time .Millisecond * 10 )
@@ -829,7 +824,6 @@ func TestNodeErrorPrioritizedOverProtocolErrors(t *testing.T) {
829824 RelayProcessorMetrics ,
830825 RelayRetriesManagerInstance ,
831826 newMockRelayStateMachine (protocolMessage , usedProviders ),
832- qos .NewQoSManager (),
833827 )
834828
835829 ctx , cancel := context .WithTimeout (context .Background (), time .Millisecond * 10 )
@@ -897,7 +891,6 @@ func TestNodeErrorPrioritizedOverProtocolErrors(t *testing.T) {
897891 RelayProcessorMetrics ,
898892 RelayRetriesManagerInstance ,
899893 newMockRelayStateMachineWithSelection (protocolMessage , usedProviders , Stateless ),
900- qos .NewQoSManager (),
901894 )
902895
903896 ctx , cancel := context .WithTimeout (context .Background (), time .Millisecond * 10 )
@@ -973,7 +966,6 @@ func TestSuccessCrossValidationTakesPriorityOverNodeError(t *testing.T) {
973966 RelayProcessorMetrics ,
974967 RelayRetriesManagerInstance ,
975968 newMockRelayStateMachineWithSelection (protocolMessage , usedProviders , Stateless ),
976- qos .NewQoSManager (),
977969 )
978970
979971 ctx , cancel := context .WithTimeout (context .Background (), time .Millisecond * 10 )
@@ -1045,7 +1037,6 @@ func TestNodeErrorCrossValidationWhenSuccessInsufficient(t *testing.T) {
10451037 RelayProcessorMetrics ,
10461038 RelayRetriesManagerInstance ,
10471039 newMockRelayStateMachineWithSelection (protocolMessage , usedProviders , Stateless ),
1048- qos .NewQoSManager (),
10491040 )
10501041
10511042 ctx , cancel := context .WithTimeout (context .Background (), time .Millisecond * 10 )
@@ -1123,7 +1114,6 @@ func TestSuccessCrossValidationIgnoresNodeErrors(t *testing.T) {
11231114 RelayProcessorMetrics ,
11241115 RelayRetriesManagerInstance ,
11251116 newMockRelayStateMachineWithSelection (protocolMessage , usedProviders , Stateless ),
1126- qos .NewQoSManager (),
11271117 )
11281118
11291119 ctx , cancel := context .WithTimeout (context .Background (), time .Millisecond * 10 )
@@ -1202,7 +1192,6 @@ func TestSuccessCrossValidationFailsWhenCrossValidationDisabled(t *testing.T) {
12021192 RelayProcessorMetrics ,
12031193 RelayRetriesManagerInstance ,
12041194 newMockRelayStateMachineWithSelection (protocolMessage , usedProviders , Stateful ), // Stateful = cross-validation disabled
1205- qos .NewQoSManager (),
12061195 )
12071196
12081197 ctx , cancel := context .WithTimeout (context .Background (), time .Millisecond * 10 )
@@ -1502,7 +1491,6 @@ func TestNodeErrorsRecoveryMetricWithCrossValidation(t *testing.T) {
15021491 mockMetrics ,
15031492 RelayRetriesManagerInstance ,
15041493 newMockRelayStateMachineWithSelection (protocolMessage , usedProviders , Stateless ),
1505- qos .NewQoSManager (),
15061494 )
15071495
15081496 // Add providers
@@ -1605,7 +1593,6 @@ func TestProtocolErrorsRecoveryMetricWithCrossValidation(t *testing.T) {
16051593 mockMetrics ,
16061594 RelayRetriesManagerInstance ,
16071595 newMockRelayStateMachineWithSelection (protocolMessage , usedProviders , Stateless ),
1608- qos .NewQoSManager (),
16091596 )
16101597
16111598 // Add providers
@@ -1694,7 +1681,6 @@ func TestBothErrorTypesRecoveryMetricsWithCrossValidation(t *testing.T) {
16941681 mockMetrics ,
16951682 RelayRetriesManagerInstance ,
16961683 newMockRelayStateMachineWithSelection (protocolMessage , usedProviders , Stateless ),
1697- qos .NewQoSManager (),
16981684 )
16991685
17001686 // Add providers
@@ -1794,7 +1780,6 @@ func TestNoRecoveryMetricsWhenCrossValidationNotMet(t *testing.T) {
17941780 mockMetrics ,
17951781 RelayRetriesManagerInstance ,
17961782 newMockRelayStateMachineWithSelection (protocolMessage , usedProviders , Stateless ),
1797- qos .NewQoSManager (),
17981783 )
17991784
18001785 // Add providers
0 commit comments