chore: test all routes

Co-authored-by: David Sharnoff <dsharnoff@singlestore.com>

pkg/op/context.go

@@ -7,16 +7,16 @@ import (
 
 type key int
 
-var (
-	issuer key = 0
+const (
+	issuerKey key = 0
 )
 
 type IssuerInterceptor struct {
 	issuerFromRequest IssuerFromRequest
 }
 
-//NewIssuerInterceptor will set the issuer into the context
-//by the provided IssuerFromRequest (e.g. returned from StaticIssuer or IssuerFromHost)
+// NewIssuerInterceptor will set the issuer into the context
+// by the provided IssuerFromRequest (e.g. returned from StaticIssuer or IssuerFromHost)
 func NewIssuerInterceptor(issuerFromRequest IssuerFromRequest) *IssuerInterceptor {
 	return &IssuerInterceptor{
 		issuerFromRequest: issuerFromRequest,
@@ -35,15 +35,19 @@ func (i *IssuerInterceptor) HandlerFunc(next http.HandlerFunc) http.HandlerFunc
 	}
 }
 
-//IssuerFromContext reads the issuer from the context (set by an IssuerInterceptor)
-//it will return an empty string if not found
+// IssuerFromContext reads the issuer from the context (set by an IssuerInterceptor)
+// it will return an empty string if not found
 func IssuerFromContext(ctx context.Context) string {
-	ctxIssuer, _ := ctx.Value(issuer).(string)
+	ctxIssuer, _ := ctx.Value(issuerKey).(string)
 	return ctxIssuer
 }
 
+// ContextWithIssuer returns a new context with issuer set to it.
+func ContextWithIssuer(ctx context.Context, issuer string) context.Context {
+	return context.WithValue(ctx, issuerKey, issuer)
+}
+
 func (i *IssuerInterceptor) setIssuerCtx(w http.ResponseWriter, r *http.Request, next http.Handler) {
-	ctx := context.WithValue(r.Context(), issuer, i.issuerFromRequest(r))
-	r = r.WithContext(ctx)
+	r = r.WithContext(ContextWithIssuer(r.Context(), i.issuerFromRequest(r)))
 	next.ServeHTTP(w, r)
 }