mirror of
https://github.com/lukaszraczylo/traefikoidc.git
synced 2026-06-05 22:44:17 +00:00
Complete rebuild of the plugin
* Fix bug affecting Azure OIDC authentication ( and most likely others ) * Fixes issue #51 * Ensure that appended roles are unique. Update the documentation. * Improvements targetting possible memory usage spikes. * Additional fixes and cleanup * Refactoring code to fix the issues identified by the users. * Modernize run * Fieldalignment * Multiple changes to improve performance and reduce complexity. - Optimise the errors and recovery. - Deduplicate code in metadata cache. - Remove unused performance monitoring code. - Simplify session management and settings handling. * Fix claims issue. * Add ability to overwrite the default scopes in the settings file * Well.. that escalated quickly. Completely forgot that Traefik uses outdated Yaegi and requires compatibility with 1.20 ( pre-generic Go code ). * Bugfix #51: Ensures that user provided scopes overrides work. * fixup! Bugfix #51: Ensures that user provided scopes overrides work. * fixup! fixup! Bugfix #51: Ensures that user provided scopes overrides work. * Abstract the provider logic into a separate package. * Additional micro fixes and cleanups. * Simplify all the things. * fixup! Simplify all the things. * fixup! fixup! Simplify all the things. * fixup! fixup! fixup! Simplify all the things. * fixup! fixup! fixup! fixup! Simplify all the things. * ... * Cleanup tests. * fixup! Cleanup tests. * fixup! fixup! fixup! Cleanup tests. * fixup! fixup! fixup! fixup! Cleanup tests. * fixup! fixup! fixup! fixup! fixup! Cleanup tests. * Issue #53: Fix CSRF token handling in reverse proxy 1. ✅ HTTPS Detection Fixed (session.go:723) - Now uses X-Forwarded-Proto header instead of r.URL.Scheme - Properly detects HTTPS in reverse proxy environments 2. ✅ SameSite Cookie Attribute Fixed - Removed automatic SameSiteStrictMode for HTTPS (would break OAuth) - Keeps SameSiteLaxMode to allow OAuth callbacks from external domains - Only uses Strict for AJAX requests which don't involve OAuth redirects 3. ✅ Cookie Domain Handling Fixed - Now respects X-Forwarded-Host header for cookie domain - Ensures cookies are set for the public domain, not internal proxy domain 4. ✅ EnhanceSessionSecurity Properly Integrated - Function is now actually called during session save - Applies security enhancements without breaking OAuth flow Why Issue #53 Failed Before: 1. Cookies were not marked Secure in HTTPS environments (browser wouldn't send them back) 2. If they had been Secure with SameSite=Strict, Azure callbacks would still fail 3. Cookie domain might have been wrong (internal vs public domain) Why It Works Now: 1. Cookies are properly marked Secure for HTTPS 2. Uses SameSite=Lax to allow OAuth provider callbacks 3. Cookie domain uses public domain from X-Forwarded-Host 4. CSRF token persists through the entire OAuth flow * Next set of enhancements together with memory usage improvements. * Memory leak fixes and optimisations. * CSRF and Cookie Domain fixes * fixup! CSRF and Cookie Domain fixes * Metadata cache leak fix + profiling * fixup! Metadata cache leak fix + profiling * Memory leaks hunting, part 1337. * Further pursue of perfection. * fixup! Further pursue of perfection. * fixup! fixup! Further pursue of perfection. * fixup! fixup! fixup! Further pursue of perfection. * fixup! fixup! fixup! fixup! Further pursue of perfection. * fixup! fixup! fixup! fixup! fixup! Further pursue of perfection. * fixup! fixup! fixup! fixup! fixup! fixup! Further pursue of perfection. * fixup! fixup! fixup! fixup! fixup! fixup! fixup! Further pursue of perfection. * fixup! fixup! fixup! fixup! fixup! fixup! fixup! fixup! Further pursue of perfection. * fixup! fixup! fixup! fixup! fixup! fixup! fixup! fixup! fixup! Further pursue of perfection. * Clear race conditions * fixup! Clear race conditions * Weekend fun with memory leaks * Splitting code into multiple files with reasonable testing coverage. ``` ok github.com/lukaszraczylo/traefikoidc 117.017s coverage: 72.6% of statements ok github.com/lukaszraczylo/traefikoidc/auth 0.505s coverage: 87.1% of statements ok github.com/lukaszraczylo/traefikoidc/circuit_breaker 0.283s coverage: 99.0% of statements github.com/lukaszraczylo/traefikoidc/config coverage: 0.0% of statements ok github.com/lukaszraczylo/traefikoidc/handlers 0.349s coverage: 98.2% of statements ok github.com/lukaszraczylo/traefikoidc/internal/providers (cached) coverage: 94.3% of statements ok github.com/lukaszraczylo/traefikoidc/middleware 0.808s coverage: 78.0% of statements ok github.com/lukaszraczylo/traefikoidc/recovery 0.653s coverage: 100.0% of statements ok github.com/lukaszraczylo/traefikoidc/session/chunking (cached) coverage: 87.8% of statements ok github.com/lukaszraczylo/traefikoidc/session/core (cached) coverage: 85.6% of statements ok github.com/lukaszraczylo/traefikoidc/session/crypto (cached) coverage: 81.8% of statements ok github.com/lukaszraczylo/traefikoidc/session/storage (cached) coverage: 93.5% of statements ok github.com/lukaszraczylo/traefikoidc/session/validators (cached) coverage: 98.8% of statements ```` * fixup! Splitting code into multiple files with reasonable testing coverage. * fixup! fixup! Splitting code into multiple files with reasonable testing coverage. * Weekend fun with further optimisations. * fixup! Weekend fun with further optimisations. * fixup! fixup! Weekend fun with further optimisations. * fixup! fixup! fixup! Weekend fun with further optimisations. * fixup! fixup! fixup! fixup! Weekend fun with further optimisations. * fixup! fixup! fixup! fixup! fixup! Weekend fun with further optimisations. * Pre-release cleanup. * Enhance test coverage. * fixup! Enhance test coverage. * fixup! fixup! Enhance test coverage. * fixup! fixup! fixup! Enhance test coverage.
This commit is contained in:
GitHub
@@ -0,0 +1,498 @@
|
||||
package traefikoidc
|
||||
|
||||
import (
|
||||
"context"
|
||||
"fmt"
|
||||
"runtime"
|
||||
"sync"
|
||||
"sync/atomic"
|
||||
"testing"
|
||||
"time"
|
||||
)
|
||||
|
||||
// TestSingletonResourceManager tests the singleton resource manager implementation
|
||||
func TestSingletonResourceManager(t *testing.T) {
|
||||
t.Run("SingletonInstance", func(t *testing.T) {
|
||||
// Test that GetResourceManager returns the same instance
|
||||
rm1 := GetResourceManager()
|
||||
rm2 := GetResourceManager()
|
||||
|
||||
if rm1 != rm2 {
|
||||
t.Error("GetResourceManager did not return singleton instance")
|
||||
}
|
||||
})
|
||||
|
||||
t.Run("ThreadSafeInitialization", func(t *testing.T) {
|
||||
// Reset singleton for test
|
||||
resetResourceManagerForTesting()
|
||||
|
||||
const numGoroutines = 100
|
||||
instances := make([]*ResourceManager, numGoroutines)
|
||||
var wg sync.WaitGroup
|
||||
|
||||
for i := 0; i < numGoroutines; i++ {
|
||||
wg.Add(1)
|
||||
go func(idx int) {
|
||||
defer wg.Done()
|
||||
instances[idx] = GetResourceManager()
|
||||
}(i)
|
||||
}
|
||||
|
||||
wg.Wait()
|
||||
|
||||
// Verify all instances are the same
|
||||
first := instances[0]
|
||||
for i := 1; i < numGoroutines; i++ {
|
||||
if instances[i] != first {
|
||||
t.Errorf("Instance %d differs from first instance", i)
|
||||
}
|
||||
}
|
||||
})
|
||||
|
||||
t.Run("SharedHTTPClient", func(t *testing.T) {
|
||||
rm := GetResourceManager()
|
||||
|
||||
client1 := rm.GetHTTPClient("test-client-1")
|
||||
client2 := rm.GetHTTPClient("test-client-1")
|
||||
|
||||
if client1 != client2 {
|
||||
t.Error("GetHTTPClient did not return same client for same key")
|
||||
}
|
||||
|
||||
client3 := rm.GetHTTPClient("test-client-2")
|
||||
if client1 == client3 {
|
||||
t.Error("GetHTTPClient returned same client for different keys")
|
||||
}
|
||||
})
|
||||
|
||||
t.Run("SharedCache", func(t *testing.T) {
|
||||
rm := GetResourceManager()
|
||||
|
||||
cache1 := rm.GetCache("test-cache-1")
|
||||
cache2 := rm.GetCache("test-cache-1")
|
||||
|
||||
if cache1 != cache2 {
|
||||
t.Error("GetCache did not return same cache for same key")
|
||||
}
|
||||
})
|
||||
|
||||
t.Run("SingletonTaskRegistry", func(t *testing.T) {
|
||||
rm := GetResourceManager()
|
||||
|
||||
err := rm.RegisterBackgroundTask("test-task", 1*time.Second, func() {
|
||||
// Test task
|
||||
})
|
||||
|
||||
if err != nil {
|
||||
t.Errorf("Failed to register task: %v", err)
|
||||
}
|
||||
|
||||
// Try to register same task again - should return existing
|
||||
err = rm.RegisterBackgroundTask("test-task", 1*time.Second, func() {
|
||||
// Duplicate task
|
||||
})
|
||||
|
||||
if err != nil {
|
||||
t.Errorf("Failed to handle duplicate task registration: %v", err)
|
||||
}
|
||||
})
|
||||
|
||||
t.Run("ReferenceCountingCleanup", func(t *testing.T) {
|
||||
rm := GetResourceManager()
|
||||
|
||||
// Add reference
|
||||
rm.AddReference("test-instance-1")
|
||||
|
||||
// Get reference count
|
||||
if rm.GetReferenceCount("test-instance-1") != 1 {
|
||||
t.Error("Reference count should be 1")
|
||||
}
|
||||
|
||||
// Add another reference
|
||||
rm.AddReference("test-instance-1")
|
||||
if rm.GetReferenceCount("test-instance-1") != 2 {
|
||||
t.Error("Reference count should be 2")
|
||||
}
|
||||
|
||||
// Remove reference
|
||||
rm.RemoveReference("test-instance-1")
|
||||
if rm.GetReferenceCount("test-instance-1") != 1 {
|
||||
t.Error("Reference count should be 1 after removal")
|
||||
}
|
||||
|
||||
// Remove last reference
|
||||
rm.RemoveReference("test-instance-1")
|
||||
if rm.GetReferenceCount("test-instance-1") != 0 {
|
||||
t.Error("Reference count should be 0 after removing all references")
|
||||
}
|
||||
})
|
||||
|
||||
t.Run("GracefulShutdown", func(t *testing.T) {
|
||||
rm := GetResourceManager()
|
||||
|
||||
// Register a task with atomic variable to avoid race condition
|
||||
var taskExecuted int32
|
||||
err := rm.RegisterBackgroundTask("shutdown-test-task", 100*time.Millisecond, func() {
|
||||
atomic.StoreInt32(&taskExecuted, 1)
|
||||
})
|
||||
|
||||
if err != nil {
|
||||
t.Errorf("Failed to register task: %v", err)
|
||||
}
|
||||
|
||||
// Start the task
|
||||
rm.StartBackgroundTask("shutdown-test-task")
|
||||
|
||||
// Wait for task to execute at least once
|
||||
time.Sleep(150 * time.Millisecond)
|
||||
|
||||
if atomic.LoadInt32(&taskExecuted) == 0 {
|
||||
t.Error("Task was not executed")
|
||||
}
|
||||
|
||||
// Shutdown
|
||||
ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
|
||||
defer cancel()
|
||||
|
||||
err = rm.Shutdown(ctx)
|
||||
if err != nil {
|
||||
t.Errorf("Shutdown failed: %v", err)
|
||||
}
|
||||
|
||||
// Verify task is stopped
|
||||
if rm.IsTaskRunning("shutdown-test-task") {
|
||||
t.Error("Task should be stopped after shutdown")
|
||||
}
|
||||
})
|
||||
}
|
||||
|
||||
// TestContextAwareGoroutineManagement tests context-aware goroutine management
|
||||
func TestContextAwareGoroutineManagement(t *testing.T) {
|
||||
t.Run("GoroutineCleanupOnContextCancel", func(t *testing.T) {
|
||||
// Reset singletons to ensure clean state
|
||||
resetResourceManagerForTesting()
|
||||
ResetUniversalCacheManagerForTesting()
|
||||
defer ResetUniversalCacheManagerForTesting()
|
||||
|
||||
initialGoroutines := runtime.NumGoroutine()
|
||||
|
||||
ctx, cancel := context.WithCancel(context.Background())
|
||||
|
||||
// Create a TraefikOidc instance with context
|
||||
config := &Config{
|
||||
ProviderURL: "https://example.com",
|
||||
ClientID: "test-client",
|
||||
ClientSecret: "test-secret",
|
||||
}
|
||||
|
||||
plugin, err := NewWithContext(ctx, config, nil, "test")
|
||||
if err != nil {
|
||||
t.Fatalf("Failed to create plugin: %v", err)
|
||||
}
|
||||
|
||||
// Wait for goroutines to start
|
||||
time.Sleep(100 * time.Millisecond)
|
||||
|
||||
midGoroutines := runtime.NumGoroutine()
|
||||
if midGoroutines <= initialGoroutines {
|
||||
t.Error("No goroutines were created")
|
||||
}
|
||||
|
||||
// Cancel context
|
||||
cancel()
|
||||
|
||||
// Close the plugin to trigger cleanup
|
||||
plugin.Close()
|
||||
|
||||
// Wait for cleanup
|
||||
time.Sleep(500 * time.Millisecond)
|
||||
|
||||
finalGoroutines := runtime.NumGoroutine()
|
||||
|
||||
// Allow for some singleton background goroutines (caches, pools, etc.)
|
||||
// These are shared across all instances and persist for the test duration
|
||||
tolerance := 10
|
||||
if finalGoroutines > initialGoroutines+tolerance {
|
||||
t.Errorf("Goroutine leak detected: initial=%d, final=%d", initialGoroutines, finalGoroutines)
|
||||
}
|
||||
})
|
||||
|
||||
t.Run("NoGoroutineLeakOnMultipleInstances", func(t *testing.T) {
|
||||
// Reset singletons to ensure clean state
|
||||
resetResourceManagerForTesting()
|
||||
ResetUniversalCacheManagerForTesting()
|
||||
defer ResetUniversalCacheManagerForTesting()
|
||||
|
||||
initialGoroutines := runtime.NumGoroutine()
|
||||
|
||||
configs := []Config{
|
||||
{ProviderURL: "https://example1.com", ClientID: "client1", ClientSecret: "secret1"},
|
||||
{ProviderURL: "https://example2.com", ClientID: "client2", ClientSecret: "secret2"},
|
||||
{ProviderURL: "https://example3.com", ClientID: "client3", ClientSecret: "secret3"},
|
||||
}
|
||||
|
||||
var plugins []*TraefikOidc
|
||||
var cancels []context.CancelFunc
|
||||
|
||||
// Create multiple instances
|
||||
for i, config := range configs {
|
||||
ctx, cancel := context.WithCancel(context.Background())
|
||||
cancels = append(cancels, cancel)
|
||||
|
||||
plugin, err := NewWithContext(ctx, &config, nil, fmt.Sprintf("test-%d", i))
|
||||
if err != nil {
|
||||
t.Fatalf("Failed to create plugin %d: %v", i, err)
|
||||
}
|
||||
plugins = append(plugins, plugin)
|
||||
}
|
||||
|
||||
// Wait for all goroutines to start
|
||||
time.Sleep(200 * time.Millisecond)
|
||||
|
||||
midGoroutines := runtime.NumGoroutine()
|
||||
|
||||
// Cancel all contexts
|
||||
for _, cancel := range cancels {
|
||||
cancel()
|
||||
}
|
||||
|
||||
// Close all plugins
|
||||
for _, plugin := range plugins {
|
||||
plugin.Close()
|
||||
}
|
||||
|
||||
// Wait for cleanup
|
||||
time.Sleep(500 * time.Millisecond)
|
||||
|
||||
finalGoroutines := runtime.NumGoroutine()
|
||||
|
||||
// Check for leaks
|
||||
tolerance := 5
|
||||
if finalGoroutines > initialGoroutines+tolerance {
|
||||
t.Errorf("Goroutine leak with multiple instances: initial=%d, mid=%d, final=%d",
|
||||
initialGoroutines, midGoroutines, finalGoroutines)
|
||||
}
|
||||
})
|
||||
|
||||
t.Run("SingletonTasksAcrossInstances", func(t *testing.T) {
|
||||
// Reset singletons to ensure clean state
|
||||
resetResourceManagerForTesting()
|
||||
ResetUniversalCacheManagerForTesting()
|
||||
defer ResetUniversalCacheManagerForTesting()
|
||||
|
||||
rm := GetResourceManager()
|
||||
|
||||
// Register singleton cleanup task
|
||||
var cleanupCount int32
|
||||
err := rm.RegisterBackgroundTask("singleton-cleanup", 100*time.Millisecond, func() {
|
||||
atomic.AddInt32(&cleanupCount, 1)
|
||||
})
|
||||
|
||||
if err != nil {
|
||||
t.Fatalf("Failed to register singleton task: %v", err)
|
||||
}
|
||||
|
||||
// Start the task
|
||||
rm.StartBackgroundTask("singleton-cleanup")
|
||||
|
||||
// Create multiple plugin instances
|
||||
var plugins []*TraefikOidc
|
||||
for i := 0; i < 3; i++ {
|
||||
ctx := context.Background()
|
||||
config := &Config{
|
||||
ProviderURL: fmt.Sprintf("https://example%d.com", i),
|
||||
ClientID: fmt.Sprintf("client%d", i),
|
||||
ClientSecret: fmt.Sprintf("secret%d", i),
|
||||
}
|
||||
|
||||
plugin, err := NewWithContext(ctx, config, nil, fmt.Sprintf("test-%d", i))
|
||||
if err != nil {
|
||||
t.Fatalf("Failed to create plugin %d: %v", i, err)
|
||||
}
|
||||
plugins = append(plugins, plugin)
|
||||
}
|
||||
|
||||
// Wait for cleanup to run multiple times
|
||||
time.Sleep(350 * time.Millisecond)
|
||||
|
||||
// Check that cleanup ran but not excessively (should be singleton)
|
||||
count := atomic.LoadInt32(&cleanupCount)
|
||||
if count < 2 || count > 5 {
|
||||
t.Errorf("Unexpected cleanup count: %d (expected 2-5 for singleton)", count)
|
||||
}
|
||||
|
||||
// Cleanup
|
||||
for _, plugin := range plugins {
|
||||
plugin.Close()
|
||||
}
|
||||
|
||||
rm.StopBackgroundTask("singleton-cleanup")
|
||||
})
|
||||
}
|
||||
|
||||
// TestResourcePooling tests resource pooling implementation
|
||||
func TestResourcePooling(t *testing.T) {
|
||||
t.Run("GoroutinePoolLimiting", func(t *testing.T) {
|
||||
rm := GetResourceManager()
|
||||
|
||||
// Configure pool with max workers
|
||||
pool := rm.GetGoroutinePool("test-pool", 5) // Max 5 workers
|
||||
|
||||
if pool == nil {
|
||||
t.Fatal("Failed to get goroutine pool")
|
||||
}
|
||||
|
||||
// Submit more tasks than pool size
|
||||
var taskCount int32
|
||||
var runningCount int32
|
||||
maxRunning := int32(0)
|
||||
|
||||
for i := 0; i < 20; i++ {
|
||||
err := pool.Submit(func() {
|
||||
atomic.AddInt32(&taskCount, 1)
|
||||
current := atomic.AddInt32(&runningCount, 1)
|
||||
|
||||
// Track max concurrent tasks
|
||||
for {
|
||||
oldMax := atomic.LoadInt32(&maxRunning)
|
||||
if current <= oldMax || atomic.CompareAndSwapInt32(&maxRunning, oldMax, current) {
|
||||
break
|
||||
}
|
||||
}
|
||||
|
||||
time.Sleep(50 * time.Millisecond)
|
||||
atomic.AddInt32(&runningCount, -1)
|
||||
})
|
||||
|
||||
if err != nil {
|
||||
t.Errorf("Failed to submit task %d: %v", i, err)
|
||||
}
|
||||
}
|
||||
|
||||
// Wait for all tasks to complete
|
||||
pool.Wait()
|
||||
|
||||
// Verify all tasks executed
|
||||
if atomic.LoadInt32(&taskCount) != 20 {
|
||||
t.Errorf("Expected 20 tasks to execute, got %d", taskCount)
|
||||
}
|
||||
|
||||
// Verify concurrency was limited
|
||||
if atomic.LoadInt32(&maxRunning) > 5 {
|
||||
t.Errorf("Max concurrent tasks exceeded pool size: %d > 5", maxRunning)
|
||||
}
|
||||
})
|
||||
|
||||
t.Run("PoolShutdown", func(t *testing.T) {
|
||||
rm := GetResourceManager()
|
||||
pool := rm.GetGoroutinePool("shutdown-pool", 3)
|
||||
|
||||
// Submit tasks
|
||||
var completed int32
|
||||
for i := 0; i < 10; i++ {
|
||||
pool.Submit(func() {
|
||||
time.Sleep(10 * time.Millisecond)
|
||||
atomic.AddInt32(&completed, 1)
|
||||
})
|
||||
}
|
||||
|
||||
// Shutdown pool
|
||||
ctx, cancel := context.WithTimeout(context.Background(), 2*time.Second)
|
||||
defer cancel()
|
||||
|
||||
err := pool.Shutdown(ctx)
|
||||
if err != nil {
|
||||
t.Errorf("Pool shutdown failed: %v", err)
|
||||
}
|
||||
|
||||
// Try to submit after shutdown - should fail
|
||||
err = pool.Submit(func() {
|
||||
t.Error("Task should not execute after shutdown")
|
||||
})
|
||||
|
||||
if err == nil {
|
||||
t.Error("Expected error when submitting to shutdown pool")
|
||||
}
|
||||
})
|
||||
|
||||
t.Run("ResourceReuse", func(t *testing.T) {
|
||||
rm := GetResourceManager()
|
||||
|
||||
// Get same pool multiple times
|
||||
pool1 := rm.GetGoroutinePool("reuse-pool", 3)
|
||||
pool2 := rm.GetGoroutinePool("reuse-pool", 3)
|
||||
|
||||
if pool1 != pool2 {
|
||||
t.Error("Expected same pool instance for same key")
|
||||
}
|
||||
|
||||
// Get HTTP client multiple times
|
||||
client1 := rm.GetHTTPClient("reuse-client")
|
||||
client2 := rm.GetHTTPClient("reuse-client")
|
||||
|
||||
if client1 != client2 {
|
||||
t.Error("Expected same HTTP client instance for same key")
|
||||
}
|
||||
})
|
||||
}
|
||||
|
||||
// TestBackwardCompatibility verifies the changes maintain backward compatibility
|
||||
func TestBackwardCompatibility(t *testing.T) {
|
||||
t.Run("LegacyNewFunction", func(t *testing.T) {
|
||||
// Test that the original New function still works
|
||||
config := &Config{
|
||||
ProviderURL: "https://example.com",
|
||||
ClientID: "test-client",
|
||||
ClientSecret: "test-secret",
|
||||
}
|
||||
|
||||
handler, err := New(context.Background(), nil, config, "test")
|
||||
if err != nil {
|
||||
t.Fatalf("Legacy New function failed: %v", err)
|
||||
}
|
||||
|
||||
if handler == nil {
|
||||
t.Fatal("Handler should not be nil")
|
||||
}
|
||||
|
||||
// Cleanup - cast to TraefikOidc if needed
|
||||
if plugin, ok := handler.(*TraefikOidc); ok {
|
||||
plugin.Close()
|
||||
}
|
||||
})
|
||||
|
||||
t.Run("ExistingAPICompatibility", func(t *testing.T) {
|
||||
config := &Config{
|
||||
ProviderURL: "https://example.com",
|
||||
ClientID: "test-client",
|
||||
ClientSecret: "test-secret",
|
||||
}
|
||||
|
||||
handler, _ := New(context.Background(), nil, config, "test")
|
||||
|
||||
// Test that the handler works
|
||||
if handler == nil {
|
||||
t.Error("Handler should not be nil")
|
||||
}
|
||||
|
||||
// Cleanup - cast to TraefikOidc if needed
|
||||
if plugin, ok := handler.(*TraefikOidc); ok {
|
||||
plugin.Close()
|
||||
}
|
||||
})
|
||||
}
|
||||
|
||||
// Helper function to reset singleton for testing
|
||||
func resetResourceManagerForTesting() {
|
||||
resourceManagerMutex.Lock()
|
||||
defer resourceManagerMutex.Unlock()
|
||||
|
||||
if globalResourceManager != nil {
|
||||
ctx, cancel := context.WithTimeout(context.Background(), 1*time.Second)
|
||||
defer cancel()
|
||||
globalResourceManager.Shutdown(ctx)
|
||||
}
|
||||
|
||||
resourceManagerOnce = sync.Once{}
|
||||
globalResourceManager = nil
|
||||
}
|
||||
Reference in New Issue
Block a user