fixup! fixup! fixup! fixup! Further pursue of perfection.

2026-10-18 04:47:28 +00:00 · 2025-09-05 16:17:02 +01:00
parent e267cb3a6f
commit b182951f22
18 changed files with 6256 additions and 1113 deletions
@@ -0,0 +1,789 @@
+package traefikoidc
+
+import (
+	"crypto/rand"
+	"encoding/hex"
+	"fmt"
+	"net/http"
+	"runtime"
+	"sync"
+	"sync/atomic"
+	"testing"
+	"time"
+)
+
+// UnifiedMockSession provides a comprehensive mock for the Session interface
+type UnifiedMockSession struct {
+	mu           sync.RWMutex
+	data         map[string]interface{}
+	callCounts   map[string]int64
+	errors       map[string]error
+	delays       map[string]time.Duration
+	destroyed    bool
+	destroyCount int64
+}
+
+// NewUnifiedMockSession creates a new mock session with default behavior
+func NewUnifiedMockSession() *UnifiedMockSession {
+	return &UnifiedMockSession{
+		data:       make(map[string]interface{}),
+		callCounts: make(map[string]int64),
+		errors:     make(map[string]error),
+		delays:     make(map[string]time.Duration),
+	}
+}
+
+// SetError configures the mock to return an error for specific method calls
+func (m *UnifiedMockSession) SetError(method string, err error) {
+	m.mu.Lock()
+	defer m.mu.Unlock()
+	m.errors[method] = err
+}
+
+// SetDelay configures the mock to add delay for specific method calls
+func (m *UnifiedMockSession) SetDelay(method string, delay time.Duration) {
+	m.mu.Lock()
+	defer m.mu.Unlock()
+	m.delays[method] = delay
+}
+
+// GetCallCount returns the number of times a method was called
+func (m *UnifiedMockSession) GetCallCount(method string) int64 {
+	m.mu.RLock()
+	defer m.mu.RUnlock()
+	return m.callCounts[method]
+}
+
+func (m *UnifiedMockSession) recordCall(method string) {
+	m.mu.Lock()
+	m.callCounts[method]++
+	m.mu.Unlock()
+}
+
+func (m *UnifiedMockSession) checkError(method string) error {
+	m.mu.RLock()
+	defer m.mu.RUnlock()
+	if err, exists := m.errors[method]; exists {
+		return err
+	}
+	return nil
+}
+
+func (m *UnifiedMockSession) applyDelay(method string) {
+	m.mu.RLock()
+	delay, exists := m.delays[method]
+	m.mu.RUnlock()
+	if exists && delay > 0 {
+		time.Sleep(delay)
+	}
+}
+
+// Session interface implementation
+func (m *UnifiedMockSession) Get(key string) (interface{}, bool) {
+	m.recordCall("Get")
+	if err := m.checkError("Get"); err != nil {
+		return nil, false
+	}
+	m.applyDelay("Get")
+
+	m.mu.RLock()
+	defer m.mu.RUnlock()
+	val, exists := m.data[key]
+	return val, exists
+}
+
+func (m *UnifiedMockSession) Set(key string, value interface{}) {
+	m.recordCall("Set")
+	if err := m.checkError("Set"); err != nil {
+		return
+	}
+	m.applyDelay("Set")
+
+	m.mu.Lock()
+	defer m.mu.Unlock()
+	m.data[key] = value
+}
+
+func (m *UnifiedMockSession) Delete(key string) {
+	m.recordCall("Delete")
+	if err := m.checkError("Delete"); err != nil {
+		return
+	}
+	m.applyDelay("Delete")
+
+	m.mu.Lock()
+	defer m.mu.Unlock()
+	delete(m.data, key)
+}
+
+func (m *UnifiedMockSession) Destroy() error {
+	m.recordCall("Destroy")
+	if err := m.checkError("Destroy"); err != nil {
+		return err
+	}
+	m.applyDelay("Destroy")
+
+	m.mu.Lock()
+	defer m.mu.Unlock()
+
+	if m.destroyed {
+		return fmt.Errorf("session already destroyed")
+	}
+
+	m.destroyed = true
+	atomic.AddInt64(&m.destroyCount, 1)
+
+	// Clear data to help with memory leak detection
+	for k := range m.data {
+		delete(m.data, k)
+	}
+
+	return nil
+}
+
+func (m *UnifiedMockSession) IsDestroyed() bool {
+	m.mu.RLock()
+	defer m.mu.RUnlock()
+	return m.destroyed
+}
+
+func (m *UnifiedMockSession) GetDestroyCount() int64 {
+	return atomic.LoadInt64(&m.destroyCount)
+}
+
+// UnifiedMockTokenVerifier provides a comprehensive mock for token verification
+type UnifiedMockTokenVerifier struct {
+	mu               sync.RWMutex
+	validTokens      map[string]bool
+	tokenMetadata    map[string]map[string]interface{}
+	callCounts       map[string]int64
+	errors           map[string]error
+	delays           map[string]time.Duration
+	verificationFunc func(string) error
+}
+
+// NewUnifiedMockTokenVerifier creates a new mock token verifier
+func NewUnifiedMockTokenVerifier() *UnifiedMockTokenVerifier {
+	return &UnifiedMockTokenVerifier{
+		validTokens:   make(map[string]bool),
+		tokenMetadata: make(map[string]map[string]interface{}),
+		callCounts:    make(map[string]int64),
+		errors:        make(map[string]error),
+		delays:        make(map[string]time.Duration),
+	}
+}
+
+// SetTokenValid configures whether a token should be considered valid
+func (m *UnifiedMockTokenVerifier) SetTokenValid(token string, valid bool) {
+	m.mu.Lock()
+	defer m.mu.Unlock()
+	m.validTokens[token] = valid
+}
+
+// SetTokenMetadata configures metadata for a token
+func (m *UnifiedMockTokenVerifier) SetTokenMetadata(token string, metadata map[string]interface{}) {
+	m.mu.Lock()
+	defer m.mu.Unlock()
+	m.tokenMetadata[token] = metadata
+}
+
+// SetVerificationFunc allows custom verification logic
+func (m *UnifiedMockTokenVerifier) SetVerificationFunc(fn func(string) error) {
+	m.mu.Lock()
+	defer m.mu.Unlock()
+	m.verificationFunc = fn
+}
+
+// SetError configures the mock to return an error for specific method calls
+func (m *UnifiedMockTokenVerifier) SetError(method string, err error) {
+	m.mu.Lock()
+	defer m.mu.Unlock()
+	m.errors[method] = err
+}
+
+// GetCallCount returns the number of times a method was called
+func (m *UnifiedMockTokenVerifier) GetCallCount(method string) int64 {
+	m.mu.RLock()
+	defer m.mu.RUnlock()
+	return m.callCounts[method]
+}
+
+func (m *UnifiedMockTokenVerifier) recordCall(method string) {
+	m.mu.Lock()
+	m.callCounts[method]++
+	m.mu.Unlock()
+}
+
+func (m *UnifiedMockTokenVerifier) VerifyToken(token string) error {
+	m.recordCall("VerifyToken")
+
+	if err := m.errors["VerifyToken"]; err != nil {
+		return err
+	}
+
+	if delay := m.delays["VerifyToken"]; delay > 0 {
+		time.Sleep(delay)
+	}
+
+	m.mu.RLock()
+	defer m.mu.RUnlock()
+
+	if m.verificationFunc != nil {
+		return m.verificationFunc(token)
+	}
+
+	if valid, exists := m.validTokens[token]; exists && valid {
+		return nil
+	}
+
+	return fmt.Errorf("invalid token")
+}
+
+// UnifiedMockTokenCache provides a comprehensive mock for token caching
+type UnifiedMockTokenCache struct {
+	mu         sync.RWMutex
+	cache      map[string]TestCacheEntry
+	callCounts map[string]int64
+	errors     map[string]error
+	delays     map[string]time.Duration
+	hitRate    float64
+}
+
+// TestCacheEntry represents a cached token entry for testing
+type TestCacheEntry struct {
+	Token     string
+	ExpiresAt time.Time
+	Metadata  map[string]interface{}
+}
+
+// NewUnifiedMockTokenCache creates a new mock token cache
+func NewUnifiedMockTokenCache() *UnifiedMockTokenCache {
+	return &UnifiedMockTokenCache{
+		cache:      make(map[string]TestCacheEntry),
+		callCounts: make(map[string]int64),
+		errors:     make(map[string]error),
+		delays:     make(map[string]time.Duration),
+		hitRate:    1.0, // Default to 100% hit rate
+	}
+}
+
+// SetError configures the mock to return an error for specific method calls
+func (m *UnifiedMockTokenCache) SetError(method string, err error) {
+	m.mu.Lock()
+	defer m.mu.Unlock()
+	m.errors[method] = err
+}
+
+// SetHitRate configures the cache hit rate (0.0 to 1.0)
+func (m *UnifiedMockTokenCache) SetHitRate(rate float64) {
+	m.mu.Lock()
+	defer m.mu.Unlock()
+	m.hitRate = rate
+}
+
+// GetCallCount returns the number of times a method was called
+func (m *UnifiedMockTokenCache) GetCallCount(method string) int64 {
+	m.mu.RLock()
+	defer m.mu.RUnlock()
+	return m.callCounts[method]
+}
+
+func (m *UnifiedMockTokenCache) recordCall(method string) {
+	m.mu.Lock()
+	m.callCounts[method]++
+	m.mu.Unlock()
+}
+
+func (m *UnifiedMockTokenCache) Get(key string) (string, bool) {
+	m.recordCall("Get")
+
+	if err := m.errors["Get"]; err != nil {
+		return "", false
+	}
+
+	if delay := m.delays["Get"]; delay > 0 {
+		time.Sleep(delay)
+	}
+
+	m.mu.RLock()
+	defer m.mu.RUnlock()
+
+	// Simulate cache miss based on hit rate
+	if m.hitRate < 1.0 {
+		// Simple random check (in real tests, you might want deterministic behavior)
+		if float64(len(key)%100)/100.0 > m.hitRate {
+			return "", false
+		}
+	}
+
+	entry, exists := m.cache[key]
+	if !exists {
+		return "", false
+	}
+
+	if time.Now().After(entry.ExpiresAt) {
+		return "", false
+	}
+
+	return entry.Token, true
+}
+
+func (m *UnifiedMockTokenCache) Set(key, token string, expiry time.Time) {
+	m.recordCall("Set")
+
+	if delay := m.delays["Set"]; delay > 0 {
+		time.Sleep(delay)
+	}
+
+	m.mu.Lock()
+	defer m.mu.Unlock()
+
+	m.cache[key] = TestCacheEntry{
+		Token:     token,
+		ExpiresAt: expiry,
+		Metadata:  make(map[string]interface{}),
+	}
+}
+
+func (m *UnifiedMockTokenCache) Delete(key string) {
+	m.recordCall("Delete")
+
+	m.mu.Lock()
+	defer m.mu.Unlock()
+	delete(m.cache, key)
+}
+
+func (m *UnifiedMockTokenCache) Clear() {
+	m.recordCall("Clear")
+
+	m.mu.Lock()
+	defer m.mu.Unlock()
+
+	for k := range m.cache {
+		delete(m.cache, k)
+	}
+}
+
+// TableTestCase represents a standardized test case structure
+type TableTestCase struct {
+	Name          string
+	Description   string
+	Input         interface{}
+	Expected      interface{}
+	ExpectedError error
+	Setup         func(*testing.T) error
+	Teardown      func(*testing.T) error
+	Timeout       time.Duration
+	SkipReason    string
+	Tags          []string
+	Parallel      bool
+}
+
+// MemoryLeakTestCase represents a test case specifically for memory leak detection
+type MemoryLeakTestCase struct {
+	Name               string
+	Description        string
+	Operation          func() error
+	Iterations         int
+	MaxGoroutineGrowth int
+	MaxMemoryGrowthMB  float64
+	Setup              func() error
+	Teardown           func() error
+	GCBetweenRuns      bool
+	Timeout            time.Duration
+}
+
+// TestSuiteRunner provides utilities for running table-driven tests
+type TestSuiteRunner struct {
+	parallelTests bool
+	timeout       time.Duration
+	beforeEach    func(*testing.T)
+	afterEach     func(*testing.T)
+}
+
+// NewTestSuiteRunner creates a new test suite runner
+func NewTestSuiteRunner() *TestSuiteRunner {
+	return &TestSuiteRunner{
+		timeout: 30 * time.Second,
+	}
+}
+
+// SetParallel enables or disables parallel test execution
+func (r *TestSuiteRunner) SetParallel(parallel bool) {
+	r.parallelTests = parallel
+}
+
+// SetTimeout sets the default timeout for tests
+func (r *TestSuiteRunner) SetTimeout(timeout time.Duration) {
+	r.timeout = timeout
+}
+
+// SetBeforeEach sets a function to run before each test
+func (r *TestSuiteRunner) SetBeforeEach(fn func(*testing.T)) {
+	r.beforeEach = fn
+}
+
+// SetAfterEach sets a function to run after each test
+func (r *TestSuiteRunner) SetAfterEach(fn func(*testing.T)) {
+	r.afterEach = fn
+}
+
+// RunTests executes a table of test cases
+func (r *TestSuiteRunner) RunTests(t *testing.T, tests []TableTestCase) {
+	for _, test := range tests {
+		test := test // Capture loop variable
+
+		if test.SkipReason != "" {
+			t.Skip(test.SkipReason)
+			continue
+		}
+
+		testFunc := func(t *testing.T) {
+			if r.beforeEach != nil {
+				r.beforeEach(t)
+			}
+
+			if r.afterEach != nil {
+				defer r.afterEach(t)
+			}
+
+			timeout := test.Timeout
+			if timeout == 0 {
+				timeout = r.timeout
+			}
+
+			done := make(chan bool, 1)
+			var testErr error
+
+			go func() {
+				defer func() {
+					if r := recover(); r != nil {
+						testErr = fmt.Errorf("test panicked: %v", r)
+					}
+					done <- true
+				}()
+
+				if test.Setup != nil {
+					if err := test.Setup(t); err != nil {
+						testErr = fmt.Errorf("setup failed: %w", err)
+						return
+					}
+				}
+
+				if test.Teardown != nil {
+					defer func() {
+						if err := test.Teardown(t); err != nil {
+							t.Errorf("teardown failed: %v", err)
+						}
+					}()
+				}
+
+				// Execute the actual test logic here
+				// This would be filled in by specific test implementations
+			}()
+
+			select {
+			case <-done:
+				if testErr != nil {
+					t.Error(testErr)
+				}
+			case <-time.After(timeout):
+				t.Errorf("test timed out after %v", timeout)
+			}
+		}
+
+		if test.Parallel || r.parallelTests {
+			t.Run(test.Name, func(t *testing.T) {
+				t.Parallel()
+				testFunc(t)
+			})
+		} else {
+			t.Run(test.Name, testFunc)
+		}
+	}
+}
+
+// RunMemoryLeakTests executes memory leak test cases
+func (r *TestSuiteRunner) RunMemoryLeakTests(t *testing.T, tests []MemoryLeakTestCase) {
+	for _, test := range tests {
+		test := test // Capture loop variable
+
+		t.Run(test.Name, func(t *testing.T) {
+			if test.Setup != nil {
+				if err := test.Setup(); err != nil {
+					t.Fatalf("setup failed: %v", err)
+				}
+			}
+
+			if test.Teardown != nil {
+				defer func() {
+					if err := test.Teardown(); err != nil {
+						t.Errorf("teardown failed: %v", err)
+					}
+				}()
+			}
+
+			// Record initial state
+			runtime.GC()
+			initialGoroutines := runtime.NumGoroutine()
+
+			var initialMem runtime.MemStats
+			runtime.ReadMemStats(&initialMem)
+
+			// Run the operation multiple times
+			for i := 0; i < test.Iterations; i++ {
+				if test.Operation != nil {
+					if err := test.Operation(); err != nil {
+						t.Errorf("iteration %d failed: %v", i, err)
+						return
+					}
+				}
+
+				if test.GCBetweenRuns {
+					runtime.GC()
+				}
+			}
+
+			// Force garbage collection and check final state
+			runtime.GC()
+			runtime.GC() // Double GC to ensure cleanup
+
+			finalGoroutines := runtime.NumGoroutine()
+
+			var finalMem runtime.MemStats
+			runtime.ReadMemStats(&finalMem)
+
+			// Check goroutine growth
+			goroutineGrowth := finalGoroutines - initialGoroutines
+			if test.MaxGoroutineGrowth >= 0 && goroutineGrowth > test.MaxGoroutineGrowth {
+				t.Errorf("goroutine leak detected: started with %d, ended with %d (growth: %d, max allowed: %d)",
+					initialGoroutines, finalGoroutines, goroutineGrowth, test.MaxGoroutineGrowth)
+			}
+
+			// Check memory growth
+			memoryGrowthBytes := int64(finalMem.Alloc) - int64(initialMem.Alloc)
+			memoryGrowthMB := float64(memoryGrowthBytes) / (1024 * 1024)
+
+			if test.MaxMemoryGrowthMB >= 0 && memoryGrowthMB > test.MaxMemoryGrowthMB {
+				t.Errorf("memory leak detected: memory grew by %.2f MB (max allowed: %.2f MB)",
+					memoryGrowthMB, test.MaxMemoryGrowthMB)
+			}
+
+			t.Logf("Memory test completed: goroutines %d->%d (Δ%d), memory %.2f MB growth",
+				initialGoroutines, finalGoroutines, goroutineGrowth, memoryGrowthMB)
+		})
+	}
+}
+
+// TestDataFactory provides utilities for generating test data
+type TestDataFactory struct{}
+
+// NewTestDataFactory creates a new test data factory
+func NewTestDataFactory() *TestDataFactory {
+	return &TestDataFactory{}
+}
+
+// GenerateRandomString generates a random string of specified length
+func (f *TestDataFactory) GenerateRandomString(length int) string {
+	if length <= 0 {
+		return ""
+	}
+	if length == 1 {
+		return "a" // Return a simple character for length 1
+	}
+	bytes := make([]byte, (length+1)/2) // Ensure we have enough bytes
+	if _, err := rand.Read(bytes); err != nil {
+		return fmt.Sprintf("test-string-%d", time.Now().UnixNano())[:length]
+	}
+	encoded := hex.EncodeToString(bytes)
+	if len(encoded) >= length {
+		return encoded[:length]
+	}
+	return encoded
+}
+
+// GenerateTestToken generates a test JWT-like token
+func (f *TestDataFactory) GenerateTestToken() string {
+	header := "eyJhbGciOiJIUzI1NiIsInR5cCI6IkpXVCJ9"
+	payload := "eyJzdWIiOiIxMjM0NTY3ODkwIiwibmFtZSI6IkpvaG4gRG9lIiwiaWF0IjoxNTE2MjM5MDIyfQ"
+	signature := f.GenerateRandomString(32)
+	return fmt.Sprintf("%s.%s.%s", header, payload, signature)
+}
+
+// GenerateTestHTTPRequest generates a test HTTP request
+func (f *TestDataFactory) GenerateTestHTTPRequest() *http.Request {
+	req, _ := http.NewRequest("GET", "http://example.com/test", nil)
+	req.Header.Set("User-Agent", "test-agent")
+	req.Header.Set("Authorization", "Bearer "+f.GenerateTestToken())
+	return req
+}
+
+// GenerateTestSession generates a test session with random data
+func (f *TestDataFactory) GenerateTestSession() *UnifiedMockSession {
+	session := NewUnifiedMockSession()
+	session.Set("user_id", f.GenerateRandomString(16))
+	session.Set("email", fmt.Sprintf("user%[email protected]", f.GenerateRandomString(8)))
+	session.Set("created_at", time.Now())
+	return session
+}
+
+// EdgeCaseGenerator provides utilities for generating comprehensive edge cases
+type EdgeCaseGenerator struct {
+	factory *TestDataFactory
+}
+
+// NewEdgeCaseGenerator creates a new edge case generator
+func NewEdgeCaseGenerator() *EdgeCaseGenerator {
+	return &EdgeCaseGenerator{
+		factory: NewTestDataFactory(),
+	}
+}
+
+// GenerateStringEdgeCases generates edge cases for string inputs
+func (g *EdgeCaseGenerator) GenerateStringEdgeCases() []string {
+	return []string{
+		"",                                    // Empty string
+		" ",                                   // Single space
+		"  ",                                  // Multiple spaces
+		"\t",                                  // Tab
+		"\n",                                  // Newline
+		"\r\n",                                // Windows newline
+		"a",                                   // Single character
+		g.factory.GenerateRandomString(1),     // Random single char
+		g.factory.GenerateRandomString(1000),  // Long string
+		g.factory.GenerateRandomString(10000), // Very long string
+		"特殊字符",                                // Unicode characters
+		"🚀🎯📊",                                 // Emojis
+		"'DROP TABLE users;",                  // SQL injection attempt
+		"<script>alert('xss')</script>",       // XSS attempt
+		"../../etc/passwd",                    // Path traversal attempt
+		string([]byte{0, 1, 2, 255}),          // Binary data
+	}
+}
+
+// GenerateIntegerEdgeCases generates edge cases for integer inputs
+func (g *EdgeCaseGenerator) GenerateIntegerEdgeCases() []int {
+	return []int{
+		0,
+		1,
+		-1,
+		42,
+		-42,
+		2147483647,  // max int32
+		-2147483648, // min int32
+		1000000,
+		-1000000,
+	}
+}
+
+// GenerateTimeEdgeCases generates edge cases for time inputs
+func (g *EdgeCaseGenerator) GenerateTimeEdgeCases() []time.Time {
+	now := time.Now()
+	return []time.Time{
+		time.Time{},                     // Zero time
+		now,                             // Current time
+		now.Add(-time.Hour),             // One hour ago
+		now.Add(time.Hour),              // One hour from now
+		now.Add(-24 * time.Hour),        // One day ago
+		now.Add(24 * time.Hour),         // One day from now
+		now.Add(-365 * 24 * time.Hour),  // One year ago
+		now.Add(365 * 24 * time.Hour),   // One year from now
+		time.Unix(0, 0),                 // Unix epoch
+		time.Unix(1<<63-62135596801, 0), // Max time
+	}
+}
+
+// GenerateHTTPRequestEdgeCases generates edge cases for HTTP requests
+func (g *EdgeCaseGenerator) GenerateHTTPRequestEdgeCases() []*http.Request {
+	cases := make([]*http.Request, 0)
+
+	// Basic cases
+	req1, _ := http.NewRequest("GET", "http://example.com", nil)
+	cases = append(cases, req1)
+
+	// Request with headers
+	req2, _ := http.NewRequest("POST", "https://api.example.com/endpoint", nil)
+	req2.Header.Set("Content-Type", "application/json")
+	req2.Header.Set("Authorization", "Bearer "+g.factory.GenerateTestToken())
+	cases = append(cases, req2)
+
+	// Request with query parameters
+	req3, _ := http.NewRequest("GET", "http://example.com/search?q=test&limit=10", nil)
+	cases = append(cases, req3)
+
+	// Request with unusual headers
+	req4, _ := http.NewRequest("GET", "http://example.com", nil)
+	req4.Header.Set("X-Custom-Header", g.factory.GenerateRandomString(1000))
+	req4.Header.Set("User-Agent", "")
+	cases = append(cases, req4)
+
+	return cases
+}
+
+// PerformanceTestHelper provides utilities for performance testing
+type PerformanceTestHelper struct {
+	samples []time.Duration
+	mu      sync.Mutex
+}
+
+// NewPerformanceTestHelper creates a new performance test helper
+func NewPerformanceTestHelper() *PerformanceTestHelper {
+	return &PerformanceTestHelper{
+		samples: make([]time.Duration, 0),
+	}
+}
+
+// Measure measures the execution time of a function
+func (h *PerformanceTestHelper) Measure(fn func()) time.Duration {
+	start := time.Now()
+	fn()
+	duration := time.Since(start)
+
+	h.mu.Lock()
+	h.samples = append(h.samples, duration)
+	h.mu.Unlock()
+
+	return duration
+}
+
+// GetAverageTime returns the average execution time
+func (h *PerformanceTestHelper) GetAverageTime() time.Duration {
+	h.mu.Lock()
+	defer h.mu.Unlock()
+
+	if len(h.samples) == 0 {
+		return 0
+	}
+
+	var total time.Duration
+	for _, sample := range h.samples {
+		total += sample
+	}
+
+	return total / time.Duration(len(h.samples))
+}
+
+// GetPercentile returns the nth percentile of execution times
+func (h *PerformanceTestHelper) GetPercentile(percentile float64) time.Duration {
+	h.mu.Lock()
+	defer h.mu.Unlock()
+
+	if len(h.samples) == 0 {
+		return 0
+	}
+
+	// Simple percentile calculation (could be improved with sorting)
+	index := int(float64(len(h.samples)) * percentile / 100.0)
+	if index >= len(h.samples) {
+		index = len(h.samples) - 1
+	}
+
+	return h.samples[index]
+}
+
+// Reset clears all performance samples
+func (h *PerformanceTestHelper) Reset() {
+	h.mu.Lock()
+	defer h.mu.Unlock()
+	h.samples = h.samples[:0]
+}