@@ -77,15 +77,6 @@ func (m *mockSender) Send(batch *logEventBatch) {
7777 m .Called (batch )
7878}
7979
80- func (m * mockSender ) SetRetryDuration (d time.Duration ) {
81- m .Called (d )
82- }
83-
84- func (m * mockSender ) RetryDuration () time.Duration {
85- args := m .Called ()
86- return args .Get (0 ).(time.Duration )
87- }
88-
8980func (m * mockSender ) Stop () {
9081 m .Called ()
9182}
@@ -123,7 +114,7 @@ func TestAddSingleEvent_WithAccountId(t *testing.T) {
123114 }
124115
125116 ep := newMockEntityProvider (expectedEntity )
126- q , sender := testPreparation (t , - 1 , & s , 1 * time .Hour , 2 * time . Hour , ep , & wg )
117+ q , sender := testPreparation (t , - 1 , & s , 1 * time .Hour , ep , & wg )
127118 q .AddEvent (newStubLogEvent ("MSG" , time .Now ()))
128119 require .False (t , called .Load (), "PutLogEvents has been called too fast, it should wait until FlushTimeout." )
129120
@@ -160,7 +151,7 @@ func TestAddSingleEvent_WithoutAccountId(t *testing.T) {
160151 }
161152
162153 ep := newMockEntityProvider (nil )
163- q , sender := testPreparation (t , - 1 , & s , 1 * time .Hour , 2 * time . Hour , ep , & wg )
154+ q , sender := testPreparation (t , - 1 , & s , 1 * time .Hour , ep , & wg )
164155 q .AddEvent (newStubLogEvent ("MSG" , time .Now ()))
165156 require .False (t , called .Load (), "PutLogEvents has been called too fast, it should wait until FlushTimeout." )
166157
@@ -190,7 +181,7 @@ func TestStopQueueWouldDoFinalSend(t *testing.T) {
190181 return & cloudwatchlogs.PutLogEventsOutput {}, nil
191182 }
192183
193- q , sender := testPreparation (t , - 1 , & s , 1 * time .Hour , 2 * time . Hour , nil , & wg )
184+ q , sender := testPreparation (t , - 1 , & s , 1 * time .Hour , nil , & wg )
194185 q .AddEvent (newStubLogEvent ("MSG" , time .Now ()))
195186
196187 time .Sleep (10 * time .Millisecond )
@@ -214,7 +205,7 @@ func TestStopPusherWouldStopRetries(t *testing.T) {
214205 }
215206
216207 logSink := testutil .NewLogSink ()
217- q , sender := testPreparationWithLogger (t , logSink , - 1 , & s , 1 * time .Hour , 2 * time . Hour , nil , & wg )
208+ q , sender := testPreparationWithLogger (t , logSink , - 1 , & s , 1 * time .Hour , nil , & wg )
218209 q .AddEvent (newStubLogEvent ("MSG" , time .Now ()))
219210 time .Sleep (10 * time .Millisecond )
220211
@@ -256,7 +247,7 @@ func TestLongMessageHandling(t *testing.T) {
256247 return & cloudwatchlogs.PutLogEventsOutput {}, nil
257248 }
258249
259- q , sender := testPreparation (t , - 1 , & s , 1 * time .Hour , 2 * time . Hour , nil , & wg )
250+ q , sender := testPreparation (t , - 1 , & s , 1 * time .Hour , nil , & wg )
260251 q .AddEvent (newStubLogEvent (longMsg , time .Now ()))
261252
262253 triggerSend (t , q )
@@ -285,7 +276,7 @@ func TestRequestIsLessThan1MB(t *testing.T) {
285276 return & cloudwatchlogs.PutLogEventsOutput {}, nil
286277 }
287278
288- q , sender := testPreparation (t , - 1 , & s , 1 * time .Hour , 2 * time . Hour , nil , & wg )
279+ q , sender := testPreparation (t , - 1 , & s , 1 * time .Hour , nil , & wg )
289280 for i := 0 ; i < 8 ; i ++ {
290281 q .AddEvent (newStubLogEvent (longMsg , time .Now ()))
291282 }
@@ -311,7 +302,7 @@ func TestRequestIsLessThan10kEvents(t *testing.T) {
311302 return & cloudwatchlogs.PutLogEventsOutput {}, nil
312303 }
313304
314- q , sender := testPreparation (t , - 1 , & s , 1 * time .Hour , 2 * time . Hour , nil , & wg )
305+ q , sender := testPreparation (t , - 1 , & s , 1 * time .Hour , nil , & wg )
315306 for i := 0 ; i < 30000 ; i ++ {
316307 q .AddEvent (newStubLogEvent (msg , time .Now ()))
317308 }
@@ -337,7 +328,7 @@ func TestTimestampPopulation(t *testing.T) {
337328 return & cloudwatchlogs.PutLogEventsOutput {}, nil
338329 }
339330
340- q , sender := testPreparation (t , - 1 , & s , 1 * time .Hour , 2 * time . Hour , nil , & wg )
331+ q , sender := testPreparation (t , - 1 , & s , 1 * time .Hour , nil , & wg )
341332 for i := 0 ; i < 3 ; i ++ {
342333 q .AddEvent (newStubLogEvent ("msg" , time.Time {}))
343334 }
@@ -361,7 +352,7 @@ func TestIgnoreOutOfTimeRangeEvent(t *testing.T) {
361352 }
362353
363354 logSink := testutil .NewLogSink ()
364- q , sender := testPreparationWithLogger (t , logSink , - 1 , & s , 10 * time .Millisecond , 2 * time . Hour , nil , & wg )
355+ q , sender := testPreparationWithLogger (t , logSink , - 1 , & s , 10 * time .Millisecond , nil , & wg )
365356 q .AddEvent (newStubLogEvent ("MSG" , time .Now ().Add (- 15 * 24 * time .Hour )))
366357 q .AddEventNonBlocking (newStubLogEvent ("MSG" , time .Now ().Add (2 * time .Hour + 1 * time .Minute )))
367358
@@ -414,7 +405,7 @@ func TestAddMultipleEvents(t *testing.T) {
414405 ))
415406 }
416407 evts [10 ], evts [90 ] = evts [90 ], evts [10 ] // make events out of order
417- q , sender := testPreparation (t , - 1 , & s , 1 * time .Hour , 2 * time . Hour , nil , & wg )
408+ q , sender := testPreparation (t , - 1 , & s , 1 * time .Hour , nil , & wg )
418409 for _ , e := range evts {
419410 q .AddEvent (e )
420411 }
@@ -466,7 +457,7 @@ func TestSendReqWhenEventsSpanMoreThan24Hrs(t *testing.T) {
466457 return nil , nil
467458 }
468459
469- q , sender := testPreparation (t , - 1 , & s , 1 * time .Hour , 2 * time . Hour , nil , & wg )
460+ q , sender := testPreparation (t , - 1 , & s , 1 * time .Hour , nil , & wg )
470461 q .AddEvent (newStubLogEvent ("MSG 25hrs ago" , time .Now ().Add (- 25 * time .Hour )))
471462 q .AddEvent (newStubLogEvent ("MSG 24hrs ago" , time .Now ().Add (- 24 * time .Hour )))
472463 q .AddEvent (newStubLogEvent ("MSG 23hrs ago" , time .Now ().Add (- 23 * time .Hour )))
@@ -496,7 +487,7 @@ func TestUnhandledErrorWouldNotResend(t *testing.T) {
496487 }
497488
498489 logSink := testutil .NewLogSink ()
499- q , sender := testPreparationWithLogger (t , logSink , - 1 , & s , 10 * time .Millisecond , 2 * time . Hour , nil , & wg )
490+ q , sender := testPreparationWithLogger (t , logSink , - 1 , & s , 10 * time .Millisecond , nil , & wg )
500491 q .AddEvent (newStubLogEvent ("msg" , time .Now ()))
501492 time .Sleep (2 * time .Second )
502493
@@ -542,7 +533,7 @@ func TestCreateLogGroupAndLogStreamWhenNotFound(t *testing.T) {
542533 }
543534
544535 logSink := testutil .NewLogSink ()
545- q , sender := testPreparationWithLogger (t , logSink , - 1 , & s , 1 * time .Hour , 2 * time . Hour , nil , & wg )
536+ q , sender := testPreparationWithLogger (t , logSink , - 1 , & s , 1 * time .Hour , nil , & wg )
546537 var eventWG sync.WaitGroup
547538 eventWG .Add (1 )
548539 q .AddEvent (& stubLogEvent {message : "msg" , timestamp : time .Now (), done : eventWG .Done })
@@ -580,7 +571,7 @@ func TestLogRejectedLogEntryInfo(t *testing.T) {
580571 }
581572
582573 logSink := testutil .NewLogSink ()
583- q , sender := testPreparationWithLogger (t , logSink , - 1 , & s , 1 * time .Hour , 2 * time . Hour , nil , & wg )
574+ q , sender := testPreparationWithLogger (t , logSink , - 1 , & s , 1 * time .Hour , nil , & wg )
584575 var eventWG sync.WaitGroup
585576 eventWG .Add (1 )
586577 q .AddEvent (& stubLogEvent {message : "msg" , timestamp : time .Now (), done : eventWG .Done })
@@ -630,7 +621,7 @@ func TestAddEventNonBlocking(t *testing.T) {
630621 start .Add (time .Duration (i )* time .Millisecond ),
631622 ))
632623 }
633- q , sender := testPreparation (t , - 1 , & s , 1 * time .Hour , 2 * time . Hour , nil , & wg )
624+ q , sender := testPreparation (t , - 1 , & s , 1 * time .Hour , nil , & wg )
634625 time .Sleep (200 * time .Millisecond ) // Wait until pusher started, merge channel is blocked
635626
636627 for _ , e := range evts {
@@ -646,32 +637,6 @@ func TestAddEventNonBlocking(t *testing.T) {
646637 wg .Wait ()
647638}
648639
649- func TestResendWouldStopAfterExhaustedRetries (t * testing.T ) {
650- t .Parallel ()
651- var wg sync.WaitGroup
652- var s stubLogsService
653- var cnt atomic.Int32
654-
655- s .ple = func (* cloudwatchlogs.PutLogEventsInput ) (* cloudwatchlogs.PutLogEventsOutput , error ) {
656- cnt .Add (1 )
657- return nil , & cloudwatchlogs.ServiceUnavailableException {}
658- }
659-
660- logSink := testutil .NewLogSink ()
661- q , sender := testPreparationWithLogger (t , logSink , - 1 , & s , 10 * time .Millisecond , time .Second , nil , & wg )
662- q .AddEvent (newStubLogEvent ("msg" , time .Now ()))
663- time .Sleep (2 * time .Second )
664-
665- logLines := logSink .Lines ()
666- lastLine := logLines [len (logLines )- 1 ]
667- expected := fmt .Sprintf ("All %v retries to G/S failed for PutLogEvents, request dropped." , cnt .Load ()- 1 )
668- require .True (t , strings .HasSuffix (lastLine , expected ), fmt .Sprintf ("Expecting error log to end with request dropped, but received '%s' in the log" , logSink .String ()))
669-
670- q .Stop ()
671- sender .Stop ()
672- wg .Wait ()
673- }
674-
675640// Cannot call q.send() directly as it would cause a race condition. Reset last sent time and trigger flush.
676641func triggerSend (t * testing.T , q * queue ) {
677642 t .Helper ()
@@ -684,7 +649,6 @@ func testPreparation(
684649 retention int ,
685650 service cloudWatchLogsService ,
686651 flushTimeout time.Duration ,
687- retryDuration time.Duration ,
688652 entityProvider logs.LogEntityProvider ,
689653 wg * sync.WaitGroup ,
690654) (* queue , Sender ) {
@@ -694,7 +658,6 @@ func testPreparation(
694658 retention ,
695659 service ,
696660 flushTimeout ,
697- retryDuration ,
698661 entityProvider ,
699662 wg ,
700663 )
@@ -706,13 +669,12 @@ func testPreparationWithLogger(
706669 retention int ,
707670 service cloudWatchLogsService ,
708671 flushTimeout time.Duration ,
709- retryDuration time.Duration ,
710672 entityProvider logs.LogEntityProvider ,
711673 wg * sync.WaitGroup ,
712674) (* queue , Sender ) {
713675 t .Helper ()
714676 tm := NewTargetManager (logger , service )
715- s := newSender (logger , service , tm , retryDuration , nil )
677+ s := newSender (logger , service , tm , nil )
716678 q := newQueue (
717679 logger ,
718680 Target {"G" , "S" , util .StandardLogGroupClass , retention },
0 commit comments