Merge branch 'v2/main' into v2/feat/2234-skills-extension
This commit is contained in:
commit
5af54301a6
2 changed files with 193 additions and 28 deletions
|
|
@ -1,12 +1,14 @@
|
|||
import { describe, it, expect, vi } from "vitest";
|
||||
import userEvent from "@testing-library/user-event";
|
||||
import {
|
||||
act,
|
||||
renderWithMantine,
|
||||
screen,
|
||||
fireEvent,
|
||||
waitFor,
|
||||
} from "../../../test/renderWithMantine";
|
||||
import { setAceText } from "../../../test/aceEditor";
|
||||
import { VALIDATE_DEBOUNCE_MS } from "../../../hooks/useServerJsonImport";
|
||||
import { ServerImportJsonModal } from "./ServerImportJsonModal";
|
||||
|
||||
const npmJson = JSON.stringify({
|
||||
|
|
@ -221,28 +223,41 @@ describe("ServerImportJsonModal", () => {
|
|||
expect(screen.getByRole("button", { name: "Add Server" })).toBeDisabled();
|
||||
});
|
||||
|
||||
// The window this exercises is the one between an edit and the debounce that
|
||||
// re-disables the button, so the whole test runs on fake timers: the pending
|
||||
// re-validation then cannot land unless this test advances it, and the window
|
||||
// stops depending on how long the machine takes to get from the paste to the
|
||||
// click. On real timers a loaded box could spend more than
|
||||
// VALIDATE_DEBOUNCE_MS there, re-disable the button, and turn the click into a
|
||||
// no-op that sets no submit error at all (#2250).
|
||||
it("guards against a live edit made before the debounce re-validates", async () => {
|
||||
const onAddServer = vi.fn();
|
||||
renderWithMantine(
|
||||
<ServerImportJsonModal
|
||||
opened
|
||||
existingIds={[]}
|
||||
onClose={vi.fn()}
|
||||
onAddServer={onAddServer}
|
||||
/>,
|
||||
);
|
||||
await pasteJson(npmJson);
|
||||
await waitFor(() =>
|
||||
expect(screen.getByRole("button", { name: "Add Server" })).toBeEnabled(),
|
||||
);
|
||||
// Replace with invalid content; the button hasn't re-disabled yet (the
|
||||
// debounce is still pending), so clicking exercises the submit-time guard.
|
||||
await pasteJson("{not json");
|
||||
fireEvent.click(screen.getByRole("button", { name: "Add Server" }));
|
||||
expect(onAddServer).not.toHaveBeenCalled();
|
||||
expect(
|
||||
await screen.findByText(/Fix the validation errors/),
|
||||
).toBeInTheDocument();
|
||||
vi.useFakeTimers();
|
||||
try {
|
||||
const onAddServer = vi.fn();
|
||||
renderWithMantine(
|
||||
<ServerImportJsonModal
|
||||
opened
|
||||
existingIds={[]}
|
||||
onClose={vi.fn()}
|
||||
onAddServer={onAddServer}
|
||||
/>,
|
||||
);
|
||||
await pasteJson(npmJson);
|
||||
// Let the first validation land, so the button is enabled to click.
|
||||
await act(async () => {
|
||||
await vi.advanceTimersByTimeAsync(VALIDATE_DEBOUNCE_MS);
|
||||
});
|
||||
expect(screen.getByRole("button", { name: "Add Server" })).toBeEnabled();
|
||||
// Replace with invalid content and do *not* advance: the debounce stays
|
||||
// pending, the button stays enabled, and clicking exercises the
|
||||
// submit-time guard that re-parses the live text.
|
||||
await pasteJson("{not json");
|
||||
fireEvent.click(screen.getByRole("button", { name: "Add Server" }));
|
||||
expect(onAddServer).not.toHaveBeenCalled();
|
||||
expect(screen.getByText(/Fix the validation errors/)).toBeInTheDocument();
|
||||
} finally {
|
||||
vi.useRealTimers();
|
||||
}
|
||||
});
|
||||
|
||||
it("loads server.json from a chosen file", async () => {
|
||||
|
|
|
|||
|
|
@ -3,6 +3,7 @@ import type { OAuthMetadata } from "@modelcontextprotocol/client";
|
|||
import { BrowserOAuthStorage } from "@inspector/core/auth/browser/storage.js";
|
||||
import {
|
||||
DEFAULT_REVOCATION_TIMEOUT_MS,
|
||||
MIN_REVOCATION_REQUEST_BUDGET_MS,
|
||||
aggregateOutcomes,
|
||||
buildRevocationRequest,
|
||||
revocationAuthMethods,
|
||||
|
|
@ -431,6 +432,27 @@ describe("revokeToken", () => {
|
|||
expect(seen?.signal).toBeInstanceOf(AbortSignal);
|
||||
expect(DEFAULT_REVOCATION_TIMEOUT_MS).toBeGreaterThan(0);
|
||||
});
|
||||
|
||||
// What is left of a shared deadline is measured with a sub-millisecond clock,
|
||||
// so the budget handed down here is routinely fractional — and Node's
|
||||
// `AbortSignal.timeout` throws `ERR_OUT_OF_RANGE` on a non-integer delay,
|
||||
// before the fetch, turning a perfectly good request into a revocation failure
|
||||
// that never left the process (#2252).
|
||||
it("accepts a fractional budget and still sends the request", async () => {
|
||||
const fetchFn = vi.fn<typeof fetch>(
|
||||
async () => new Response(null, { status: 200 }),
|
||||
);
|
||||
const outcome = await revokeToken({
|
||||
endpoint: REVOKE_URL,
|
||||
token: "r",
|
||||
tokenTypeHint: "refresh_token",
|
||||
supportedAuthMethods: [],
|
||||
fetchFn,
|
||||
timeoutMs: 19.996,
|
||||
});
|
||||
expect(fetchFn).toHaveBeenCalledTimes(1);
|
||||
expect(outcome).toMatchObject({ status: "revoked" });
|
||||
});
|
||||
});
|
||||
|
||||
/**
|
||||
|
|
@ -838,6 +860,58 @@ describe("revokeStoredOAuthTokens (plan + execute)", () => {
|
|||
expect(fetchFn).toHaveBeenCalledTimes(1);
|
||||
});
|
||||
|
||||
// The boundary the shared budget used to be decided on was `remainingMs > 0`,
|
||||
// which timer resolution can land either side of: a grant that finished a
|
||||
// hair before the deadline left a fractional budget behind, and the next grant
|
||||
// spent it on a request that could not possibly complete (#2252). The floor
|
||||
// makes that decision the same on every run.
|
||||
it("does not issue a request with less than the minimum budget left", async () => {
|
||||
const grant = (n: string) => ({
|
||||
issuer: "https://as.example.com",
|
||||
token: `r-${n}`,
|
||||
tokenTypeHint: "refresh_token" as const,
|
||||
});
|
||||
const timeoutMs = 40;
|
||||
// The clock is stubbed rather than slept against. A real sleep makes this a
|
||||
// ONE-SIDED detector: under contention the first request overruns, the
|
||||
// remainder goes negative, and the unfixed `remainingMs <= 0` bound skips
|
||||
// the second grant for the wrong reason — so the test passes on an
|
||||
// implementation that has no floor at all. Advancing a stub inside the
|
||||
// fetch puts the second grant's remainder at exactly
|
||||
// `MIN_REVOCATION_REQUEST_BUDGET_MS - 1` on every machine: positive, and
|
||||
// under the floor, which is the only state that tells the two apart.
|
||||
let now = 1_000;
|
||||
const nowSpy = vi.spyOn(performance, "now").mockImplementation(() => now);
|
||||
const fetchFn = vi.fn<typeof fetch>(async () => {
|
||||
now += timeoutMs - MIN_REVOCATION_REQUEST_BUDGET_MS + 1;
|
||||
return new Response(null, { status: 200 });
|
||||
});
|
||||
|
||||
try {
|
||||
const outcome = await executeOAuthRevocation(
|
||||
{
|
||||
serverUrl: SERVER_URL,
|
||||
grants: [grant("a"), grant("b")],
|
||||
failures: [],
|
||||
endpoint: REVOKE_URL,
|
||||
supportedAuthMethods: [],
|
||||
metadataIssuer: "https://as.example.com",
|
||||
},
|
||||
{ fetchFn, timeoutMs },
|
||||
);
|
||||
|
||||
expect(fetchFn).toHaveBeenCalledTimes(1);
|
||||
// The unattempted grant outranks the first one's success, so the caller
|
||||
// hears that a grant may still be live rather than that all was well.
|
||||
expect(outcome).toMatchObject({ status: "failed" });
|
||||
expect(outcome.status === "failed" ? outcome.detail : "").toContain(
|
||||
"budget was exhausted",
|
||||
);
|
||||
} finally {
|
||||
nowSpy.mockRestore();
|
||||
}
|
||||
});
|
||||
|
||||
// A grant bound to an issuer the cached metadata does not describe cannot be
|
||||
// revoked — that endpoint belongs to a different authorization server, and
|
||||
// sending it another AS's token would hand a credential to a server that
|
||||
|
|
|
|||
Loading…
Reference in a new issue