Function setupMCPClient

e2e/e2e_test.go:146–250

(t *testing.T, options ...clientOption)

}

func setupMCPClient(t *testing.T, options ...clientOption) *mcp.ClientSession {
	// Check rate limit before setting up the client
	waitForRateLimit(t)

	// Get token and ensure Docker image is built
	token := getE2EToken(t)

	// Create and configure options with default to all toolsets
	opts := &clientOpts{
		enabledToolsets: []string{"all"},
	}

	// Apply all options to configure the opts struct
	for _, option := range options {
		option(opts)
	}

	ctx := context.Background()

	// By default, we run the tests including the Docker image, but with DEBUG
	// enabled, we run the server in-process, allowing for easier debugging.
	var session *mcp.ClientSession
	if os.Getenv("GITHUB_MCP_SERVER_E2E_DEBUG") == "" {
		ensureDockerImageBuilt(t)

		// Prepare Docker arguments
		args := []string{
			"run",
			"-i",
			"--rm",
			"-e",
			"GITHUB_PERSONAL_ACCESS_TOKEN", // Personal access token is all required
		}

		host := getE2EHost()
		if host != "" {
			args = append(args, "-e", "GITHUB_HOST")
		}

		// Add toolsets environment variable to the Docker arguments
		if len(opts.enabledToolsets) > 0 {
			args = append(args, "-e", "GITHUB_TOOLSETS")
		}

		// Add the image name
		args = append(args, "github/e2e-github-mcp-server")

		// Construct the env vars for the MCP Client to execute docker with
		// We need to include os.Environ() so docker can find its socket and config
		dockerEnvVars := append(os.Environ(),
			fmt.Sprintf("GITHUB_PERSONAL_ACCESS_TOKEN=%s", token),
			fmt.Sprintf("GITHUB_TOOLSETS=%s", strings.Join(opts.enabledToolsets, ",")),
		)

		if host != "" {
			dockerEnvVars = append(dockerEnvVars, fmt.Sprintf("GITHUB_HOST=%s", host))
		}