Increase azure fetcher timeout (#2628)

* increase azure fetcher timeout

* add function

(cherry picked from commit b24534b)

internal/flavors/benchmark/azure.go

@@ -21,6 +21,7 @@ import (
 	"context"
 	"errors"
 	"fmt"
+	"math"
 	"time"
 
 	"github.com/elastic/elastic-agent-libs/logp"
@@ -50,7 +51,7 @@ func (a *Azure) NewBenchmark(ctx context.Context, log *logp.Logger, cfg *config.
 
 	return builder.New(
 		builder.WithBenchmarkDataProvider(bdp),
-		builder.WithManagerTimeout(20*time.Minute),
+		builder.WithManagerTimeout(calculateFetcherTimeout(cfg.Period)),
 	).Build(ctx, log, cfg, resourceCh, reg)
 }
 
@@ -95,3 +96,16 @@ func (a *Azure) checkDependencies() error {
 	}
 	return nil
 }
+
+// calculateFetcherTimeout calculates the timeout for each fetcher based on period as ~70% of the period duration. If less than 3 hours, it returns 3 hours.
+func calculateFetcherTimeout(period time.Duration) time.Duration {
+	roundedHours := math.Round(period.Hours() * 0.7)
+	to := time.Duration(roundedHours) * time.Hour
+
+	const min = 3 * time.Hour
+	if to < min {
+		return min
+	}
+
+	return to
+}

internal/flavors/benchmark/azure_test.go

@@ -20,8 +20,10 @@ package benchmark
 import (
 	"errors"
 	"testing"
+	"time"
 
 	"github.com/stretchr/testify/mock"
+	"github.com/stretchr/testify/require"
 
 	"github.com/elastic/cloudbeat/internal/config"
 	"github.com/elastic/cloudbeat/internal/resources/providers/azurelib"
@@ -119,3 +121,46 @@ func mockAzureInventoryInitializerService(err error) azurelib.ProviderInitialize
 	}
 	return initializer
 }
+
+func TestCalculateFetcherTimeout(t *testing.T) {
+	tests := map[string]struct {
+		inputPeriod time.Duration
+		expected    time.Duration
+	}{
+		"48h": {
+			inputPeriod: 48 * time.Hour,
+			expected:    34 * time.Hour,
+		},
+		"24h": {
+			inputPeriod: 24 * time.Hour,
+			expected:    17 * time.Hour,
+		},
+		"3h": {
+			inputPeriod: 3 * time.Hour,
+			expected:    3 * time.Hour,
+		},
+		"30m": {
+			inputPeriod: 30 * time.Minute,
+			expected:    3 * time.Hour,
+		},
+		"0": {
+			inputPeriod: 0,
+			expected:    3 * time.Hour,
+		},
+		"-30m": {
+			inputPeriod: -30 * time.Minute,
+			expected:    3 * time.Hour,
+		},
+		"-3h": {
+			inputPeriod: -3 * time.Hour,
+			expected:    3 * time.Hour,
+		},
+	}
+
+	for name, tc := range tests {
+		t.Run(name, func(t *testing.T) {
+			got := calculateFetcherTimeout(tc.inputPeriod)
+			require.Equal(t, tc.expected, got)
+		})
+	}
+}