@@ -34,7 +34,10 @@ import (
34
34
discovery "k8s.io/api/discovery/v1"
35
35
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
36
36
"k8s.io/apimachinery/pkg/types"
37
+ "k8s.io/component-base/metrics/legacyregistry"
38
+ "k8s.io/component-base/metrics/testutil"
37
39
"k8s.io/kubernetes/pkg/proxy"
40
+ "k8s.io/kubernetes/pkg/proxy/metrics"
38
41
netutils "k8s.io/utils/net"
39
42
"k8s.io/utils/ptr"
40
43
)
@@ -306,9 +309,15 @@ func TestCleanStaleEntries(t *testing.T) {
306
309
t .Logf ("entries before cleanup %d after cleanup %d" , len (entriesBeforeCleanup ), len (entriesAfterCleanup ))
307
310
fake := NewFake ()
308
311
fake .entries = entriesBeforeCleanup
309
- CleanStaleEntries (fake , testIPFamily , svcPortMap , endpointsMap )
310
312
313
+ legacyregistry .MustRegister (metrics .ReconcileConntrackFlowsDeletedEntriesTotal )
314
+ CleanStaleEntries (fake , testIPFamily , svcPortMap , endpointsMap )
311
315
actualEntries , _ := fake .ListEntries (ipFamilyMap [testIPFamily ])
316
+
317
+ metricCount , err := testutil .GetCounterMetricValue (metrics .ReconcileConntrackFlowsDeletedEntriesTotal .WithLabelValues (string (testIPFamily )))
318
+ require .NoError (t , err )
319
+ require .Equal (t , int (metricCount ), len (entriesBeforeCleanup )- len (entriesAfterCleanup ))
320
+
312
321
require .Equal (t , len (entriesAfterCleanup ), len (actualEntries ))
313
322
314
323
// sort the actual flows before comparison
0 commit comments