1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
|
package metrics
import (
"testing"
"time"
)
func TestNewSample(t *testing.T) {
tests := []struct {
name string
metric string
labels map[string]string
value float64
timestamp time.Time
wantNil bool
}{
{
name: "with labels",
metric: "test_metric",
labels: map[string]string{"env": "prod", "host": "server1"},
value: 42.5,
timestamp: time.Now(),
wantNil: false,
},
{
name: "nil labels initialized",
metric: "test_metric",
labels: nil,
value: 100,
timestamp: time.Now(),
wantNil: false,
},
{
name: "empty labels",
metric: "test_metric",
labels: map[string]string{},
value: 0,
timestamp: time.Now(),
wantNil: false,
},
}
for _, tt := range tests {
t.Run(tt.name, func(t *testing.T) {
sample := NewSample(tt.metric, tt.labels, tt.value, tt.timestamp)
if sample.MetricName != tt.metric {
t.Errorf("MetricName = %v, want %v", sample.MetricName, tt.metric)
}
if sample.Value != tt.value {
t.Errorf("Value = %v, want %v", sample.Value, tt.value)
}
if sample.Labels == nil {
t.Error("Labels should never be nil")
}
if !sample.Timestamp.Equal(tt.timestamp) {
t.Errorf("Timestamp = %v, want %v", sample.Timestamp, tt.timestamp)
}
})
}
}
func TestSample_Age(t *testing.T) {
tests := []struct {
name string
sample Sample
wantNear time.Duration
}{
{
name: "recent sample",
sample: Sample{
MetricName: "test",
Timestamp: time.Now().Add(-5 * time.Minute),
},
wantNear: 5 * time.Minute,
},
{
name: "old sample",
sample: Sample{
MetricName: "test",
Timestamp: time.Now().Add(-1 * time.Hour),
},
wantNear: 1 * time.Hour,
},
{
name: "very recent",
sample: Sample{
MetricName: "test",
Timestamp: time.Now().Add(-10 * time.Second),
},
wantNear: 10 * time.Second,
},
}
for _, tt := range tests {
t.Run(tt.name, func(t *testing.T) {
age := tt.sample.Age()
// Allow 1 second tolerance for test execution time
if age < tt.wantNear-time.Second || age > tt.wantNear+time.Second {
t.Errorf("Age() = %v, want near %v", age, tt.wantNear)
}
})
}
}
func TestSample_IsRecent(t *testing.T) {
threshold := 5 * time.Minute
tests := []struct {
name string
sample Sample
threshold time.Duration
want bool
}{
{
name: "within threshold",
sample: Sample{
MetricName: "test",
Timestamp: time.Now().Add(-2 * time.Minute),
},
threshold: threshold,
want: true,
},
{
name: "beyond threshold",
sample: Sample{
MetricName: "test",
Timestamp: time.Now().Add(-10 * time.Minute),
},
threshold: threshold,
want: false,
},
{
name: "exactly at threshold",
sample: Sample{
MetricName: "test",
Timestamp: time.Now().Add(-5 * time.Minute),
},
threshold: threshold,
want: false,
},
{
name: "very recent",
sample: Sample{
MetricName: "test",
Timestamp: time.Now().Add(-10 * time.Second),
},
threshold: threshold,
want: true,
},
}
for _, tt := range tests {
t.Run(tt.name, func(t *testing.T) {
if got := tt.sample.IsRecent(tt.threshold); got != tt.want {
t.Errorf("IsRecent() = %v, want %v (age: %v)", got, tt.want, tt.sample.Age())
}
})
}
}
|
