Export WithGraphQLFeatures for remote server tests

Co-authored-by: SamMorrowDrums <4811358+SamMorrowDrums@users.noreply.github.com>

Author: Copilot

pkg/github/issues.go

@@ -1789,7 +1789,7 @@ func AssignCopilotToIssue(t translations.TranslationHelperFunc) inventory.Server
 
 			// Add the GraphQL-Features header for the agent assignment API
 			// The header will be read by the HTTP transport if it's configured to do so
-			ctxWithFeatures := withGraphQLFeatures(ctx, "issues_copilot_assignment_api_support")
+			ctxWithFeatures := WithGraphQLFeatures(ctx, "issues_copilot_assignment_api_support")
 
 			if err := client.Mutate(
 				ctxWithFeatures,

pkg/github/transport.go

@@ -10,10 +10,11 @@ import (
 // These flags enable preview or experimental GitHub API features that are not yet GA.
 type graphQLFeaturesKey struct{}
 
-// withGraphQLFeatures adds GraphQL feature flags to the context.
+// WithGraphQLFeatures adds GraphQL feature flags to the context.
 // The flags are read by GraphQLFeaturesTransport and sent as the GraphQL-Features header.
-// This is used internally by tool handlers that require experimental GitHub API features.
-func withGraphQLFeatures(ctx context.Context, features ...string) context.Context {
+// This is used by tool handlers that require experimental GitHub API features.
+// Remote servers can also use this function in tests to simulate feature flag contexts.
+func WithGraphQLFeatures(ctx context.Context, features ...string) context.Context {
 	return context.WithValue(ctx, graphQLFeaturesKey{}, features)
 }
 
@@ -30,7 +31,7 @@ func GetGraphQLFeatures(ctx context.Context) []string {
 }
 
 // GraphQLFeaturesTransport is an http.RoundTripper that adds GraphQL-Features
-// header based on context values set by withGraphQLFeatures.
+// header based on context values set by WithGraphQLFeatures.
 //
 // This transport should be used in the HTTP client chain for githubv4.Client
 // to ensure GraphQL feature flags are properly sent to the GitHub API.
@@ -59,7 +60,7 @@ func GetGraphQLFeatures(ctx context.Context) []string {
 //	gqlClient := githubv4.NewClient(httpClient)
 //
 // The transport reads feature flags from request context using GetGraphQLFeatures.
-// Feature flags are added to context by the tool handler via withGraphQLFeatures.
+// Feature flags are added to context by the tool handler via WithGraphQLFeatures.
 type GraphQLFeaturesTransport struct {
 	// Transport is the underlying http.RoundTripper. If nil, http.DefaultTransport is used.
 	Transport http.RoundTripper

pkg/github/transport_test.go

@@ -61,7 +61,7 @@ func TestGraphQLFeaturesTransport(t *testing.T) {
 			// Create request with or without features in context
 			ctx := context.Background()
 			if tt.features != nil {
-				ctx = withGraphQLFeatures(ctx, tt.features...)
+				ctx = WithGraphQLFeatures(ctx, tt.features...)
 			}
 
 			req, err := http.NewRequestWithContext(ctx, "GET", server.URL, nil)
@@ -97,7 +97,7 @@ func TestGraphQLFeaturesTransport_NilTransport(t *testing.T) {
 		},
 	}
 
-	ctx := withGraphQLFeatures(context.Background(), "test_feature")
+	ctx := WithGraphQLFeatures(context.Background(), "test_feature")
 	req, err := http.NewRequestWithContext(ctx, "GET", server.URL, nil)
 	require.NoError(t, err)
 
@@ -123,7 +123,7 @@ func TestGraphQLFeaturesTransport_PreservesOtherHeaders(t *testing.T) {
 		},
 	}
 
-	ctx := withGraphQLFeatures(context.Background(), "feature1")
+	ctx := WithGraphQLFeatures(context.Background(), "feature1")
 	req, err := http.NewRequestWithContext(ctx, "GET", server.URL, nil)
 	require.NoError(t, err)