mirror of
https://github.com/ivuorinen/gibidify.git
synced 2026-01-26 03:24:05 +00:00
e9bd694685
Add overflow checks before converting uint64 memory values to int64 to prevent potential integer overflow issues identified by gosec (G115). - Add math.MaxInt64 checks in fileproc/backpressure.go - Add math.MaxInt64 checks in fileproc/resource_monitor_validation.go - Add math.MaxInt64 checks in fileproc/resource_monitor_metrics.go - Add math.MaxInt64 check in benchmark/benchmark.go with nosec annotation Co-authored-by: ivuorinen <11024+ivuorinen@users.noreply.github.com>
41 lines
961 B
Go
41 lines
961 B
Go
package fileproc
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/spf13/viper"
|
|
|
|
"github.com/ivuorinen/gibidify/testutil"
|
|
)
|
|
|
|
func TestResourceMonitor_RateLimiting(t *testing.T) {
|
|
testutil.ResetViperConfig(t, "")
|
|
|
|
// Enable rate limiting with a low rate for testing
|
|
viper.Set("resourceLimits.enabled", true)
|
|
viper.Set("resourceLimits.rateLimitFilesPerSec", 5) // 5 files per second
|
|
|
|
rm := NewResourceMonitor()
|
|
defer rm.Close()
|
|
|
|
ctx, cancel := context.WithTimeout(context.Background(), 1*time.Second)
|
|
defer cancel()
|
|
|
|
// First few requests should succeed quickly
|
|
start := time.Now()
|
|
for i := 0; i < 3; i++ {
|
|
err := rm.WaitForRateLimit(ctx)
|
|
if err != nil {
|
|
t.Errorf("Expected no error for rate limit wait %d, got %v", i, err)
|
|
}
|
|
}
|
|
|
|
// Should have taken some time due to rate limiting
|
|
duration := time.Since(start)
|
|
if duration < 200*time.Millisecond {
|
|
t.Logf("Rate limiting may not be working as expected, took only %v", duration)
|
|
}
|
|
}
|