73 lines
2.4 KiB
Go
73 lines
2.4 KiB
Go
package lib_test
|
|
|
|
import (
|
|
"loadr/lib"
|
|
"reflect"
|
|
"testing"
|
|
"time"
|
|
)
|
|
|
|
func comparePerformanceMetrics(a, b *lib.PerformanceMetrics) bool {
|
|
return a.TotalRequests == b.TotalRequests &&
|
|
a.TotalResponses == b.TotalResponses &&
|
|
a.TotalLatency == b.TotalLatency &&
|
|
a.MaxLatency == b.MaxLatency &&
|
|
a.MinLatency == b.MinLatency &&
|
|
reflect.DeepEqual(a.ResponseCounters, b.ResponseCounters)
|
|
}
|
|
|
|
func TestCalculateAndPrintMetrics(t *testing.T) {
|
|
// Define test cases.
|
|
tests := []struct {
|
|
name string
|
|
startTime time.Time
|
|
requestsPerSecond float64
|
|
endpoint string
|
|
verb string
|
|
expectedMetrics lib.PerformanceMetrics
|
|
}{
|
|
{
|
|
name: "Test 1",
|
|
startTime: time.Now().Add(-1 * time.Second), // 1 second ago
|
|
requestsPerSecond: 1.0,
|
|
endpoint: "http://localhost",
|
|
verb: "GET",
|
|
expectedMetrics: lib.PerformanceMetrics{
|
|
TotalRequests: 1,
|
|
TotalResponses: 1,
|
|
TotalLatency: 1 * time.Second,
|
|
MaxLatency: 1 * time.Second,
|
|
MinLatency: 1 * time.Second,
|
|
ResponseCounters: map[int]int32{1: 1},
|
|
},
|
|
},
|
|
// Add more test cases as needed.
|
|
}
|
|
|
|
for i := range tests {
|
|
tt := &tests[i]
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
// Reset the metrics before each test
|
|
metrics := lib.PerformanceMetrics{}
|
|
|
|
// Mock the system behavior
|
|
metrics.TotalRequests++
|
|
metrics.ResponseCounters = make(map[int]int32)
|
|
metrics.TotalResponses++
|
|
metrics.TotalLatency += 1 * time.Second
|
|
metrics.MaxLatency = 1 * time.Second
|
|
metrics.MinLatency = 1 * time.Second
|
|
metrics.ResponseCounters[1]++
|
|
|
|
// Call the function
|
|
lib.CalculateAndPrintMetrics(tt.startTime, tt.requestsPerSecond, tt.endpoint, tt.verb)
|
|
|
|
// Check if the metrics are correct
|
|
if !comparePerformanceMetrics(&metrics, &tt.expectedMetrics) {
|
|
t.Errorf("CalculateAndPrintMetrics() = TotalRequests: %v, TotalResponses: %v, TotalLatency: %v, MaxLatency: %v, MinLatency: %v, ResponseCounters: %v, want TotalRequests: %v, TotalResponses: %v, TotalLatency: %v, MaxLatency: %v, MinLatency: %v, ResponseCounters: %v",
|
|
metrics.TotalRequests, metrics.TotalResponses, metrics.TotalLatency, metrics.MaxLatency, metrics.MinLatency, metrics.ResponseCounters,
|
|
tt.expectedMetrics.TotalRequests, tt.expectedMetrics.TotalResponses, tt.expectedMetrics.TotalLatency, tt.expectedMetrics.MaxLatency, tt.expectedMetrics.MinLatency, tt.expectedMetrics.ResponseCounters)
|
|
}
|
|
})
|
|
}
|
|
}
|