mirror of
https://github.com/lukaszraczylo/traefikoidc.git
synced 2026-10-19 04:47:48 +00:00
fixup! fixup! fixup! fixup! Further pursue of perfection.
This commit is contained in:
+287
@@ -0,0 +1,287 @@
|
||||
package traefikoidc
|
||||
|
||||
import (
|
||||
"os"
|
||||
"strconv"
|
||||
"testing"
|
||||
"time"
|
||||
)
|
||||
|
||||
// TestConfig manages test execution configuration and performance settings
|
||||
type TestConfig struct {
|
||||
// Test execution modes
|
||||
ExtendedTests bool // Run extended/stress tests
|
||||
LongTests bool // Run long-running performance tests
|
||||
QuickMode bool // Quick smoke tests only
|
||||
|
||||
// Performance settings
|
||||
MaxConcurrency int // Maximum concurrent operations
|
||||
MaxIterations int // Maximum test iterations
|
||||
DefaultTimeout time.Duration // Default test timeout
|
||||
MemoryThreshold float64 // Memory growth threshold in MB
|
||||
GoroutineGrowth int // Acceptable goroutine growth
|
||||
|
||||
// Cache settings for tests
|
||||
CacheSize int // Default cache size for tests
|
||||
CleanupInterval time.Duration // Cleanup interval for tests
|
||||
|
||||
// Environment-specific overrides
|
||||
MemoryStressTest bool // Enable memory stress tests
|
||||
ConcurrencyTest bool // Enable high concurrency tests
|
||||
LeakDetection bool // Enable memory leak detection
|
||||
}
|
||||
|
||||
// NewTestConfig creates a test configuration based on flags and environment
|
||||
func NewTestConfig() *TestConfig {
|
||||
config := &TestConfig{
|
||||
// Default quick mode settings - very conservative for 30s target
|
||||
ExtendedTests: false,
|
||||
LongTests: false,
|
||||
QuickMode: true,
|
||||
MaxConcurrency: 2, // Reduced for quick mode
|
||||
MaxIterations: 1, // Minimal iterations for quick smoke tests
|
||||
DefaultTimeout: 5 * time.Second, // Shorter timeout
|
||||
MemoryThreshold: 1.0, // Strict memory limit
|
||||
GoroutineGrowth: 1, // Very strict goroutine limit
|
||||
CacheSize: 10, // Small cache size
|
||||
CleanupInterval: 50 * time.Millisecond, // Faster cleanup
|
||||
MemoryStressTest: false,
|
||||
ConcurrencyTest: false,
|
||||
LeakDetection: false, // Disable by default in quick mode for speed
|
||||
}
|
||||
|
||||
// Check for extended test flag
|
||||
if os.Getenv("RUN_EXTENDED_TESTS") == "1" || os.Getenv("RUN_EXTENDED_TESTS") == "true" {
|
||||
config.EnableExtendedTests()
|
||||
}
|
||||
|
||||
// Check for long test flag
|
||||
if os.Getenv("RUN_LONG_TESTS") == "1" || os.Getenv("RUN_LONG_TESTS") == "true" {
|
||||
config.EnableLongTests()
|
||||
}
|
||||
|
||||
// Check for stress tests
|
||||
if os.Getenv("RUN_STRESS_TESTS") == "1" || os.Getenv("RUN_STRESS_TESTS") == "true" {
|
||||
config.EnableStressTests()
|
||||
}
|
||||
|
||||
// Check for memory leak detection override
|
||||
if os.Getenv("DISABLE_LEAK_DETECTION") == "1" || os.Getenv("DISABLE_LEAK_DETECTION") == "true" {
|
||||
config.LeakDetection = false
|
||||
}
|
||||
|
||||
// Parse custom concurrency limit
|
||||
if concStr := os.Getenv("TEST_MAX_CONCURRENCY"); concStr != "" {
|
||||
if conc, err := strconv.Atoi(concStr); err == nil && conc > 0 {
|
||||
config.MaxConcurrency = conc
|
||||
}
|
||||
}
|
||||
|
||||
// Parse custom iteration limit
|
||||
if iterStr := os.Getenv("TEST_MAX_ITERATIONS"); iterStr != "" {
|
||||
if iter, err := strconv.Atoi(iterStr); err == nil && iter > 0 {
|
||||
config.MaxIterations = iter
|
||||
}
|
||||
}
|
||||
|
||||
// Parse memory threshold
|
||||
if memStr := os.Getenv("TEST_MEMORY_THRESHOLD_MB"); memStr != "" {
|
||||
if mem, err := strconv.ParseFloat(memStr, 64); err == nil && mem > 0 {
|
||||
config.MemoryThreshold = mem
|
||||
}
|
||||
}
|
||||
|
||||
return config
|
||||
}
|
||||
|
||||
// EnableExtendedTests switches to extended test mode
|
||||
func (c *TestConfig) EnableExtendedTests() {
|
||||
c.ExtendedTests = true
|
||||
c.QuickMode = false
|
||||
c.MaxConcurrency = 20
|
||||
c.MaxIterations = 10
|
||||
c.DefaultTimeout = 30 * time.Second
|
||||
c.MemoryThreshold = 10.0
|
||||
c.GoroutineGrowth = 5
|
||||
c.CacheSize = 200
|
||||
c.CleanupInterval = 50 * time.Millisecond
|
||||
c.ConcurrencyTest = true
|
||||
}
|
||||
|
||||
// EnableLongTests switches to long-running test mode
|
||||
func (c *TestConfig) EnableLongTests() {
|
||||
c.LongTests = true
|
||||
c.QuickMode = false
|
||||
c.MaxConcurrency = 50
|
||||
c.MaxIterations = 100
|
||||
c.DefaultTimeout = 60 * time.Second
|
||||
c.MemoryThreshold = 50.0
|
||||
c.GoroutineGrowth = 10
|
||||
c.CacheSize = 1000
|
||||
c.CleanupInterval = 10 * time.Millisecond
|
||||
c.ConcurrencyTest = true
|
||||
c.MemoryStressTest = true
|
||||
}
|
||||
|
||||
// EnableStressTests switches to stress test mode
|
||||
func (c *TestConfig) EnableStressTests() {
|
||||
c.ExtendedTests = true
|
||||
c.LongTests = true
|
||||
c.QuickMode = false
|
||||
c.MaxConcurrency = 100
|
||||
c.MaxIterations = 500
|
||||
c.DefaultTimeout = 120 * time.Second
|
||||
c.MemoryThreshold = 100.0
|
||||
c.GoroutineGrowth = 20
|
||||
c.CacheSize = 2000
|
||||
c.CleanupInterval = 5 * time.Millisecond
|
||||
c.ConcurrencyTest = true
|
||||
c.MemoryStressTest = true
|
||||
}
|
||||
|
||||
// ShouldSkipTest determines if a test should be skipped based on config
|
||||
func (c *TestConfig) ShouldSkipTest(t *testing.T, testType TestType) bool {
|
||||
// Always respect testing.Short() - skip everything except basic quick tests
|
||||
if testing.Short() {
|
||||
switch testType {
|
||||
case TestTypeQuick:
|
||||
return false // Allow quick tests
|
||||
case TestTypeExtended, TestTypeLong, TestTypeMemoryStress, TestTypeConcurrencyStress:
|
||||
t.Skip("Skipping extended test in short mode")
|
||||
return true
|
||||
case TestTypeLeakDetection:
|
||||
// Skip leak detection in short mode unless explicitly enabled
|
||||
if !c.LeakDetection {
|
||||
t.Skip("Skipping leak detection test in short mode (use RUN_EXTENDED_TESTS=1 to enable)")
|
||||
return true
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
// Check specific test type flags
|
||||
switch testType {
|
||||
case TestTypeExtended:
|
||||
if !c.ExtendedTests {
|
||||
t.Skip("Skipping extended test (use RUN_EXTENDED_TESTS=1 to enable)")
|
||||
return true
|
||||
}
|
||||
case TestTypeLong:
|
||||
if !c.LongTests {
|
||||
t.Skip("Skipping long test (use RUN_LONG_TESTS=1 to enable)")
|
||||
return true
|
||||
}
|
||||
case TestTypeMemoryStress:
|
||||
if !c.MemoryStressTest {
|
||||
t.Skip("Skipping memory stress test (use RUN_STRESS_TESTS=1 to enable)")
|
||||
return true
|
||||
}
|
||||
case TestTypeConcurrencyStress:
|
||||
if !c.ConcurrencyTest {
|
||||
t.Skip("Skipping concurrency stress test (use RUN_EXTENDED_TESTS=1 to enable)")
|
||||
return true
|
||||
}
|
||||
case TestTypeLeakDetection:
|
||||
if !c.LeakDetection {
|
||||
t.Skip("Skipping leak detection test (DISABLE_LEAK_DETECTION=1 set)")
|
||||
return true
|
||||
}
|
||||
}
|
||||
|
||||
return false
|
||||
}
|
||||
|
||||
// AdjustMemoryLeakTestCase adjusts a memory leak test case based on configuration
|
||||
func (c *TestConfig) AdjustMemoryLeakTestCase(testCase *MemoryLeakTestCase) {
|
||||
// Adjust iterations
|
||||
if testCase.Iterations > c.MaxIterations {
|
||||
testCase.Iterations = c.MaxIterations
|
||||
}
|
||||
|
||||
// Ensure minimum of 1 iteration
|
||||
if testCase.Iterations < 1 {
|
||||
testCase.Iterations = 1
|
||||
}
|
||||
|
||||
// Adjust memory threshold
|
||||
if testCase.MaxMemoryGrowthMB > c.MemoryThreshold && c.QuickMode {
|
||||
testCase.MaxMemoryGrowthMB = c.MemoryThreshold
|
||||
}
|
||||
|
||||
// Adjust goroutine growth
|
||||
if testCase.MaxGoroutineGrowth > c.GoroutineGrowth && c.QuickMode {
|
||||
testCase.MaxGoroutineGrowth = c.GoroutineGrowth
|
||||
}
|
||||
|
||||
// Adjust timeout
|
||||
if testCase.Timeout > c.DefaultTimeout && c.QuickMode {
|
||||
testCase.Timeout = c.DefaultTimeout
|
||||
} else if testCase.Timeout == 0 {
|
||||
testCase.Timeout = c.DefaultTimeout
|
||||
}
|
||||
}
|
||||
|
||||
// AdjustConcurrencyParams adjusts concurrency parameters for tests
|
||||
func (c *TestConfig) AdjustConcurrencyParams(requested int) int {
|
||||
if requested > c.MaxConcurrency {
|
||||
return c.MaxConcurrency
|
||||
}
|
||||
return requested
|
||||
}
|
||||
|
||||
// GetCacheSize returns appropriate cache size for tests
|
||||
func (c *TestConfig) GetCacheSize() int {
|
||||
return c.CacheSize
|
||||
}
|
||||
|
||||
// GetCleanupInterval returns appropriate cleanup interval for tests
|
||||
func (c *TestConfig) GetCleanupInterval() time.Duration {
|
||||
return c.CleanupInterval
|
||||
}
|
||||
|
||||
// TestType represents different categories of tests
|
||||
type TestType int
|
||||
|
||||
const (
|
||||
TestTypeQuick TestType = iota
|
||||
TestTypeExtended
|
||||
TestTypeLong
|
||||
TestTypeMemoryStress
|
||||
TestTypeConcurrencyStress
|
||||
TestTypeLeakDetection
|
||||
)
|
||||
|
||||
// String returns string representation of test type
|
||||
func (tt TestType) String() string {
|
||||
switch tt {
|
||||
case TestTypeQuick:
|
||||
return "quick"
|
||||
case TestTypeExtended:
|
||||
return "extended"
|
||||
case TestTypeLong:
|
||||
return "long"
|
||||
case TestTypeMemoryStress:
|
||||
return "memory-stress"
|
||||
case TestTypeConcurrencyStress:
|
||||
return "concurrency-stress"
|
||||
case TestTypeLeakDetection:
|
||||
return "leak-detection"
|
||||
default:
|
||||
return "unknown"
|
||||
}
|
||||
}
|
||||
|
||||
// Global test configuration instance
|
||||
var globalTestConfig *TestConfig
|
||||
|
||||
// GetTestConfig returns the global test configuration
|
||||
func GetTestConfig() *TestConfig {
|
||||
if globalTestConfig == nil {
|
||||
globalTestConfig = NewTestConfig()
|
||||
}
|
||||
return globalTestConfig
|
||||
}
|
||||
|
||||
// SetTestConfig sets the global test configuration (useful for testing)
|
||||
func SetTestConfig(config *TestConfig) {
|
||||
globalTestConfig = config
|
||||
}
|
||||
Reference in New Issue
Block a user