From ca4db7ebfc3a06234b0c1551afcca43a6188ce93 Mon Sep 17 00:00:00 2001 From: cs-308-2023 Date: Mon, 27 Jan 2025 00:42:56 +0530 Subject: [PATCH] Code coverage Signed-off-by: cs-308-2023 --- cmd/jaeger/internal/features/display_test.go | 21 ++++++++++++++++---- 1 file changed, 17 insertions(+), 4 deletions(-) diff --git a/cmd/jaeger/internal/features/display_test.go b/cmd/jaeger/internal/features/display_test.go index 48f89f39852..a010279ced1 100644 --- 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") }