fix

trufflesecurity · Nov 7, 2024 · 627ece0 · 627ece0
1 parent 8740bc6
commit 627ece0
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 7 deletions.
diff --git a/pkg/sources/s3/progress_tracker.go b/pkg/sources/s3/progress_tracker.go
@@ -108,10 +108,6 @@ func (p *ProgressTracker) GetResumePoint(ctx context.Context) (ResumeInfo, error
 // Complete marks the entire scanning operation as finished and clears the resume state.
 // This should only be called once all scanning operations are complete.
 func (p *ProgressTracker) Complete(_ context.Context, message string) error {
-	if !p.enabled {
-		return nil
-	}
-
 	// Preserve existing progress counters while clearing resume state.
 	p.progress.SetProgressComplete(
 		int(p.progress.SectionsCompleted),

diff --git a/pkg/sources/s3/progress_tracker_test.go b/pkg/sources/s3/progress_tracker_test.go
@@ -577,7 +577,7 @@ func TestComplete(t *testing.T) {
 			}{
 				sectionsCompleted: 50,
 				sectionsRemaining: 100,
-				resumeInfo:        `{"CurrentBucket":"test-bucket","StartAfter":"some-key"}`,
+				resumeInfo:        "",
 				message:           "Should not change",
 			},
 			completeMessage: "Completed",
@@ -589,8 +589,8 @@ func TestComplete(t *testing.T) {
 			}{
 				sectionsCompleted: 50,
 				sectionsRemaining: 100,
-				resumeInfo:        `{"CurrentBucket":"test-bucket","StartAfter":"some-key"}`,
-				message:           "Should not change",
+				resumeInfo:        "",
+				message:           "Completed",
 			},
 		},
 		{