Additional test cases to raise test coverage

Signed-off-by: Ed Snible <[email protected]>
jaegertracing · Jul 5, 2022 · 4f449af · 4f449af
1 parent d7e2528
commit 4f449af
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 0 deletions.
diff --git a/pkg/config/tenancy/context_test.go b/pkg/config/tenancy/context_test.go
@@ -91,6 +91,11 @@ func TestMetadataAnnotator(t *testing.T) {
 		tenancyConfig  *TenancyConfig
 		requestHeaders map[string][]string
 	}{
+		{
+			name:           "missing tenant",
+			tenancyConfig:  NewTenancyConfig(&Options{Enabled: true}),
+			requestHeaders: map[string][]string{"x-tenant": {""}},
+		},
 		{
 			name:           "tenanted",
 			tenancyConfig:  NewTenancyConfig(&Options{Enabled: true}),

diff --git a/pkg/config/tenancy/interceptor_test.go b/pkg/config/tenancy/interceptor_test.go
@@ -69,6 +69,12 @@ func TestTenancyInterceptor(t *testing.T) {
 			ctx:           metadata.NewIncomingContext(context.Background(), map[string][]string{"x-tenant": {"acme"}}),
 			errMsg:        "",
 		},
+		{
+			name:          "extra tenant header",
+			tenancyConfig: NewTenancyConfig(&Options{Enabled: true, Tenants: []string{"acme"}}),
+			ctx:           metadata.NewIncomingContext(context.Background(), map[string][]string{"x-tenant": {"acme", "megacorp"}}),
+			errMsg:        "rpc error: code = PermissionDenied desc = extra tenant header",
+		},
 	}
 
 	for _, test := range tests {