diff --git a/__fixtures__/core.ts b/__fixtures__/core.ts index efe7ebe..becbebd 100644 --- a/__fixtures__/core.ts +++ b/__fixtures__/core.ts @@ -9,3 +9,4 @@ export const getBooleanInput = vi.fn() export const setOutput = vi.fn() export const setFailed = vi.fn() export const warning = vi.fn() +export const setSecret = vi.fn() diff --git a/__tests__/main.test.ts b/__tests__/main.test.ts index 578bf98..387e064 100644 --- a/__tests__/main.test.ts +++ b/__tests__/main.test.ts @@ -136,6 +136,7 @@ describe('main.ts', () => { await run() expect(core.setOutput).toHaveBeenCalled() + expect(core.setSecret).toHaveBeenCalledWith('fake-token') verifyStandardResponse() expect(mockProcessExit).toHaveBeenCalledWith(0) }) @@ -199,6 +200,7 @@ describe('main.ts', () => { await run() + expect(core.setSecret).toHaveBeenCalledWith('fake-token') expect(mockConnectToGitHubMCP).toHaveBeenCalledWith('fake-token', '') expect(mockMcpInference).toHaveBeenCalledWith( expect.objectContaining({ diff --git a/src/main.ts b/src/main.ts index 18febc1..73c768a 100644 --- a/src/main.ts +++ b/src/main.ts @@ -61,9 +61,12 @@ export async function run(): Promise { if (token === undefined) { throw new Error('GITHUB_TOKEN is not set') } + core.setSecret(token) // Get GitHub MCP token (use dedicated token if provided, otherwise fall back to main token) const githubMcpToken = core.getInput('github-mcp-token') || token + core.setSecret(githubMcpToken) + const githubMcpToolsets = core.getInput('github-mcp-toolsets') const endpoint = core.getInput('endpoint')