use new helper

GoogleChrome · Jan 24, 2025 · 29e8f21 · 29e8f21
1 parent 1665589
commit 29e8f21
Show file tree

Hide file tree

Showing 8 changed files with 15 additions and 0 deletions.
diff --git a/backend/pkg/httpserver/get_feature_metadata_test.go b/backend/pkg/httpserver/get_feature_metadata_test.go
@@ -71,6 +71,7 @@ func TestGetFeatureMetadata(t *testing.T) {
 			}
 			myServer := Server{wptMetricsStorer: mockStorer, metadataStorer: mockMetadataStorer}
 			assertTestServerRequest(t, &myServer, tc.request, tc.expectedResponse)
+			// TODO: Start tracking call count and assert call count.
 		})
 	}
 }
diff --git a/backend/pkg/httpserver/get_feature_test.go b/backend/pkg/httpserver/get_feature_test.go
@@ -194,6 +194,8 @@ func TestGetFeature(t *testing.T) {
 			}
 			myServer := Server{wptMetricsStorer: mockStorer, metadataStorer: nil}
 			assertTestServerRequest(t, &myServer, tc.request, tc.expectedResponse)
+			assertMockCallCount(t, tc.expectedCallCount, mockStorer.callCountGetFeature,
+				"GetFeature")
 		})
 	}
 }
diff --git a/backend/pkg/httpserver/get_features_test.go b/backend/pkg/httpserver/get_features_test.go
@@ -314,6 +314,7 @@ func TestListFeatures(t *testing.T) {
 			}
 			myServer := Server{wptMetricsStorer: mockStorer, metadataStorer: nil}
 			assertTestServerRequest(t, &myServer, tc.request, tc.expectedResponse)
+			assertMockCallCount(t, tc.expectedCallCount, mockStorer.callCountFeaturesSearch, "FeaturesSearch")
 		})
 	}
 }
diff --git a/backend/pkg/httpserver/list_aggregated_feature_support_test.go b/backend/pkg/httpserver/list_aggregated_feature_support_test.go
@@ -173,6 +173,9 @@ func TestListAggregatedFeatureSupport(t *testing.T) {
 			}
 			myServer := Server{wptMetricsStorer: mockStorer, metadataStorer: nil}
 			assertTestServerRequest(t, &myServer, tc.request, tc.expectedResponse)
+			assertMockCallCount(t, tc.expectedCallCount, mockStorer.callCountListBrowserFeatureCountMetric,
+				"ListBrowserFeatureCountMetric")
+
 		})
 	}
 }
diff --git a/backend/pkg/httpserver/list_aggregated_wpt_metrics_test.go b/backend/pkg/httpserver/list_aggregated_wpt_metrics_test.go
@@ -167,6 +167,8 @@ func TestListAggregatedWPTMetrics(t *testing.T) {
 			}
 			myServer := Server{wptMetricsStorer: mockStorer, metadataStorer: nil}
 			assertTestServerRequest(t, &myServer, tc.request, tc.expectedResponse)
+			assertMockCallCount(t, tc.expectedCallCount, mockStorer.callCountListMetricsOverTimeWithAggregatedTotals,
+				"ListMetricsOverTimeWithAggregatedTotals")
 		})
 	}
 }
diff --git a/backend/pkg/httpserver/list_chromium_usage_test.go b/backend/pkg/httpserver/list_chromium_usage_test.go
@@ -94,6 +94,8 @@ func TestListChromiumDailyUsageStats(t *testing.T) {
 			}
 			myServer := Server{wptMetricsStorer: mockStorer, metadataStorer: nil}
 			assertTestServerRequest(t, &myServer, tc.request, tc.expectedResponse)
+			assertMockCallCount(t, tc.expectedCallCount, mockStorer.callCountListChromiumDailyUsageStats,
+				"ListChromiumDailyUsageStats")
 		})
 	}
 }
diff --git a/backend/pkg/httpserver/list_feature_wpt_metrics_test.go b/backend/pkg/httpserver/list_feature_wpt_metrics_test.go
@@ -164,6 +164,8 @@ func TestListFeatureWPTMetrics(t *testing.T) {
 			}
 			myServer := Server{wptMetricsStorer: mockStorer, metadataStorer: nil}
 			assertTestServerRequest(t, &myServer, tc.request, tc.expectedResponse)
+			assertMockCallCount(t, tc.expectedCallCount, mockStorer.callCountListMetricsForFeatureIDBrowserAndChannel,
+				"ListMetricsForFeatureIDBrowserAndChannel")
 		})
 	}
 }
diff --git a/backend/pkg/httpserver/list_missing_one_implementation_counts_test.go b/backend/pkg/httpserver/list_missing_one_implementation_counts_test.go
@@ -182,6 +182,8 @@ func TestListMissingOneImplemenationCounts(t *testing.T) {
 			}
 			myServer := Server{wptMetricsStorer: mockStorer, metadataStorer: nil}
 			assertTestServerRequest(t, &myServer, tc.request, tc.expectedResponse)
+			assertMockCallCount(t, tc.expectedCallCount, mockStorer.callCountListMissingOneImplCounts,
+				"ListMissingOneImplCounts")
 		})
 	}
 }