Code coverage

Signed-off-by: cs-308-2023 <[email protected]>
jaegertracing · Jan 26, 2025 · ca4db7e · ca4db7e
1 parent bfc6cac
commit ca4db7e
Showing 1 changed file with 17 additions and 4 deletions.
diff --git a/cmd/jaeger/internal/features/display_test.go b/cmd/jaeger/internal/features/display_test.go
@@ -16,13 +16,26 @@ func TestMain(m *testing.M) {
 	testutils.VerifyGoLeaks(m)
 }
 
-func TestDisplay(t *testing.T) {
+func TestDisplayStageAlpha(t *testing.T) {
 	featuregate.GlobalRegistry().MustRegister(
-		"jaeger.featuresdisplay.testgate",
+		"jaeger.featuresdisplay.testgate1",
+		featuregate.StageAlpha,
+		featuregate.WithRegisterDescription("testdescription"),
+		featuregate.WithRegisterReferenceURL("https://testurl.com"),
+	)
+	out := DisplayFeatures()
+	require.Contains(t, out, "jaeger.featuresdisplay.testgate")
+	require.Contains(t, out, "Off")
+}
+
+func TestDisplayStageBeta(t *testing.T) {
+	featuregate.GlobalRegistry().MustRegister(
+		"jaeger.featuresdisplay.testgate2",
 		featuregate.StageBeta,
-		featuregate.WithRegisterDescription("test-description"),
-		featuregate.WithRegisterReferenceURL("https://test-url.com"),
+		featuregate.WithRegisterDescription("testdescription"),
+		featuregate.WithRegisterReferenceURL("https://testurl.com"),
 	)
 	out := DisplayFeatures()
 	require.Contains(t, out, "jaeger.featuresdisplay.testgate")
+	require.Contains(t, out, "On")
 }