mirror of
https://github.com/paperclipai/paperclip.git
synced 2026-11-08 12:33:05 +01:00
## Thinking Path > - Paperclip is the open source app people use to manage AI agents for work > - The server has about 1,100 test and helper files in `server/src/__tests__` > - `server/tsconfig.json` excludes that directory, Vitest does not check types, and no script runs the tsconfig in that directory > - So a test can call a function with the wrong arguments and still pass, and no tool reports it > - This gap hid 2,168 real type errors, and some of them hid tests that did not test what they claim > - This pull request fixes all of these errors and adds the directory to the server layers from paperclipai/paperclip#18737 > - The benefit is that a type error in a server test now fails the typecheck, with no baseline and no backlog ## Linked Issues or Issue Description Refs paperclipai/paperclip#18737, which is merged and added the server layers. Refs paperclipai/paperclip#18738, which is merged and checks the test projects outside `pnpm -r`, so it is not a duplicate. Refs paperclipai/paperclip#18137, which records that this directory has no check. **What happened?** No tool checked the types of 1,074 of the 1,101 files in `server/src/__tests__`. I searched `package.json`, the workflows, and all tsconfig files. 27 files are in the program of another tsconfig, and these 27 have 0 errors. 11 of the 27 are in `helpers/maintenance/tsconfig.json`. No script ran that project when I measured this. paperclipai/paperclip#18738 now runs it. **Expected behavior** The server typecheck checks these files, and a type error fails the check. **Steps to reproduce** 1. On the base commit, run `tsc -p server/src/__tests__/tsconfig.json --noEmit`. It reports 6,536 errors and needs 12.3 GiB. 2. Add a type error to any file in `server/src/__tests__`. 3. Run `pnpm --filter @paperclipai/server typecheck`. It passes. **Paperclip version or commit** I measured the error breakdown below on commit `8dfd632452`, with TypeScript 7.0.2. The branch now includes master up to paperclipai/paperclip#18865. **Deployment mode** Not applicable. This issue applies to development and CI. No production source file in the server changes. ### Error breakdown Each row adds one change to the row above. | Step | Total errors | Change | Cause | | --- | --- | --- | --- | | Config as committed | 6,536 | | | | Add `src/types/*.d.ts` | 4,447 | -2,089 | Config. The Express request types were missing. | | Set `rootDir` to the repository root | 2,436 | -2,011 | Config. | | Set `allowImportingTsExtensions` | 2,181 | -255 | Config. Vitest accepts `.ts` imports. | | Add `ES2024.Promise` to `lib` | 2,138 | -43 | Config. Node 24 has `Promise.withResolvers`. | | Set `allowJs` for `.mjs` imports | 2,148 | +10 | Config. It removes 8 `TS7016` errors and shows new real errors. | | Set `declaration` to false | 2,142 | -6 | Config. Tests emit nothing. | | Count only files in `server/src` | 2,058 | -84 | Other packages: 53 Daytona, 29 Runner test, 2 UI. | So 4,478 of the 6,536 errors (69%) came from config or scope, and 2,058 are real type errors in 314 files. The 29 Runner test errors are also real, which gives 2,087 errors. Tests that master added after this measurement had 81 more errors, so this pull request fixes 2,168 errors. This table classifies the first 2,058 errors. | Kind of real error | Errors | | --- | --- | | Wrong argument or assignment type | 1,023 | | Stale or partial fixture objects | 525 | | Property that the type does not have | 317 | | Missing narrowing of null or unknown | 101 | | Implicit any | 60 | | Missing name or module | 16 | | Other | 16 | ### What the type errors hid Most errors were stale fixtures and loose mocks. These errors hid real test problems, and this pull request fixes them: - **A test that never ran.** `instrumentation.test.ts` required `sdk-trace-base`, which is not installed. So 3 span tests were skipped. They now run and pass. - **Assertions that could not fail.** - `issue-thread-interactions-service` used a company ID as an issue ID. - `status-card-evaluation-work` compared `[]` with `[]` in a rollback check. - `slack-conversation-lifecycle` read `item.issueId`, which does not exist. - `email-channels` used a permission key that does not exist. - `heartbeat-project-env` imported `LOW_TRUST_REVIEW_PRESET`, which is not exported. The value was undefined. - **Postgres processes that did not stop.** 11 test files called `stop` on the test database helper. That method does not exist, and the helper method is `cleanup`. - **Arguments that the service ignored.** - `methodKey` is now `connectionMethodKey` in the Daytona, Unikraft, 1Password, and Railway tests. - 108 calls passed `"owner-user"` to a parameter that expects an options object. - 9 chat wizard fixtures passed `managementContext` as `source`, so they ran `existing_app` and not `manifest`. - 2 tests passed a `{ userId }` actor that the function ignores. - Inserts into `environments` passed a `companyId` column that the table does not have. - **Problems in tests that master added during this work.** - `sandbox-creation-operations` read `database.connectionString` when an external test database was set. In that mode `database` is undefined, so the fixture would stop with a TypeError. It now uses the external URL. - `idle-completed-history` set a Dot operation to `dispatched`. That is a Muse status, and the Dot type does not have it. The fixture now uses `admitted`, which is a real in-flight Dot status. ### Possible product problems, not changed These may be real problems in production code or types. I did not change them, because this pull request does not change production behavior: - `decisions-service` `service.create` does not validate the lenient `cancel_issue_tree` input. - `createConnection` ignores `input.connectionPurpose`. - The `createConnection` route schema default stores `{}` for `transportConfig`. - `prepareManagedAiRuntime` accepted the invalid value `mode: "connection"`. - `createBinding` accepts invalid `projectionClass` strings outside class 3. - `ensureManagedProjectWorkspace` forwards an untyped `repoRef` by spread. - `routineVariableSchema.label` is optional, but `RoutineVariable.label` is required. - `ToolRunContext.projectId` may be too strict. - `withdrawInteraction` reads `issue.status`, which its type omits. - `issueService.list` and `reconcileNativeFinalizations` omit fields that they return at runtime. - `createGitHubExternalObjectProvider` handles `tokenProvider: null`, but its type forbids this value. ## What Changed - Fixed the 2,168 type errors. The diff changes 322 test and helper files. - Corrected `server/src/__tests__/tsconfig.json`. Editors and the typecheck now read the same options and report the same errors. - Added three `test-tree` layers to `server/scripts/typecheck-layers.mjs`. They read the server layers as declarations, and any error fails them. A hash of the path selects the shard, so a new file does not move other files. - Added `server/scripts/typecheck-stubs/daytona-sdk.d.ts`. Server tests reach Daytona plugin source, and the root install does not have `@daytonaio/sdk`. - Changed two imports in `ui/src/lib/issue-execution-policy.ts` to use `.js` extensions, because server tests import this file. - Made `declarationsConfig` and `--declarations-only` skip the `test-tree` layers, because these layers emit nothing. Other projects read only the source layers. Added a test for this. - Updated `doc/DEVELOPING.md`. ## Verification Run these commands: ```sh pnpm --filter @paperclipai/server typecheck pnpm exec vitest run server/scripts/typecheck-layers.test.mjs ``` Results on macOS (arm64), Node 24, 16 cores. Each check ran after the merge of paperclipai/paperclip#18738 unless the row says otherwise: | Check | Commit | Result | | --- | --- | --- | | Server typecheck, `PAPERCLIP_TYPECHECK_LOW_MEMORY=1`, all 7 layers | `6e45a08573` | Exit 0, 0 errors, 62 seconds, peak 3.52 GiB | | `node scripts/typecheck-test-projects.mjs --low-memory` | `a3306bd13e` | Exit 0, 0 errors, 42 seconds, peak 3.25 GiB | | The same script with `--reuse-declarations` | `1b878aabc0` | Exit 0, 0 errors | | `typecheck-layers.test.mjs` and `live-recovery-work.test.ts` | `01db5634e4` | 90 passed. These include real compiler runs for a pass, a type error, and an invalid tsconfig. | | `tsconfig.declarations.json` after a full server typecheck | `0fb1e4fb0a` | It references only `below`, `cycle`, and `above`. No generated config names a `tsconfig.layers.json` file. | | Owned runtime test in `idle-completed-history` | `e8f2f448ac` | Passed, with the fix that master later replaced | | `idle-incoming-wake`, `plugin-idle-manager`, and the changed `chat-channels` test | `6e45a08573` | 30 passed | | Runner `native-execution.test.ts` | `eab4dcc1f4` | 114 passed | | The 19 server test files that changed since `eab4dcc1f4`, and `typecheck-layers.test.mjs` | `259737b700` | 4,275 of 4,277 passed. The 2 failures are in `chat-channels` and are in the list of 7 below. | The peak is the largest single process, from `/usr/bin/time -l`. Peaks for the server typecheck were between 3.42 GiB and 3.56 GiB across 8 runs. Peaks for the test projects were 3.25 GiB and 3.37 GiB across 2 runs. I did not measure a cgroup. **Runtime check of the fixes.** Types do not prove that a test still works. So I ran all 309 changed server test files, 11,929 tests, before the rebase onto paperclipai/paperclip#18737: | Result | Tests | Detail | | --- | --- | --- | | Passed | 11,913 | Includes the 2 tests and the 54 `dot-runner` tests that passed in a later run, as described below. | | Failed | 7 | The same 7 tests also fail on the commit before the fixes. | | Skipped | 9 | These tests have their own skip conditions: `sentry` 5, and 1 each in `bundled-plugin-registration`, `customer-success-cloud`, `onepassword-connection`, and `workspace-runtime-https-live-exercise`. | The 7 tests that fail with and without this change, on my machine: - 3 tests in `chat-channels` and `email-channels` stop with `ENOENT` on a `SKILL.md` file. The path is in the parent checkout, above the git worktree that I used, so I expect that the worktree is the cause. I did not prove this. - `workspace-instance-cleanup` compares `/private/var/...` with `/var/...`. This is a macOS symlink difference. - 2 tests in `git-reproducible-managed-checkouts` get `unknown_base_git_files` and expect `unpushed_managed_checkout`. I did not find the cause. - 1 more test in `chat-channels`, `materializes direct external-chat finals once`, gets `undefined`. I did not find the cause. CI is the check for these 7 tests. `ci / verify` passed on `eab4dcc1f4`. Failures during this work: - My fix to `agent-runtime-connector-access` changed an invalid permission key in the setup, but not in the assertion. The runtime check found this. I corrected the assertion, and the file passes. - `heartbeat-plugin-environment` failed one wait in the full run. It passed three times alone, so I think that load caused the failure. It can fail again under load. - `dot-runner` skipped all 54 tests in the full run because `cargo` was not on my `PATH`. It then passed 54 of 54 with the toolchain on the `PATH`. - My first `lib` value removed the default DOM types, and the error count went up by 375. I added the DOM libraries. - My Daytona stub with bare `any` left 48 errors, because generic helpers then inferred `unknown`. I added shapes that return promises. - One fix to `agent-secrets-routes` used `class_3_static_lease`, and the test failed. That class allows only specific paths. I reverted to `unclassified`. - `ci / General tests (server (8/12))` failed on `259737b700` because of the flaky port in `idle-completed-history`. paperclipai/paperclip#18853 fixed this test on master, so this pull request keeps that fix. **Audit of the diff.** I counted these patterns in added and removed lines, to find fixes that only hide an error: | Pattern | Added | Removed | | --- | --- | --- | | `@ts-ignore` | 0 | 0 | | `@ts-expect-error` | 0 | 1 | | `.skip` | 0 | 0 | | `as any` | 9 | 9 | | `: any` | 11 | 2 | | `as never` | 8 | 2 | | `as unknown as` | 35 | 4 | | `Promise<unknown>` | 16 | 5 | | `expect(` | 128 | 113 | No hunk removes more `expect(` calls than it adds. Three removals of test content were intentional, and I checked each one: - `muse-control` removed `signalToken`. Line 1012 already asserts that it is absent. - `merged-pr-confirmation-sweep` removed legacy aliases. Other tests cover them. - `heartbeat-workspace-branch-containment` removed a `finalize` branch that no test row can reach. Some partial fixtures stay unchecked through `as unknown as` and `Promise<unknown>`, mostly in `issue-agent-mutation-ownership-routes` and `issue-comment-reopen-routes`. **Accuracy of the check.** One program over the tree and the three shards reported the same errors at the same positions. Memory, as the peak of the largest process on one thread, was 3.93 GiB for one program and 2.43 GiB for the largest shard. I measured these two values before the fixes. Not run or not proven: - I did not run the full `pnpm test:run`, or a root `pnpm build`. I expect CI to run the full suites. - I did not run the 4 GiB container check, or the full `scripts/typecheck-low-memory.mjs`. The peaks above come from macOS, so they are not a Linux cgroup measurement. - I did not run the UI typecheck after the merges. - `workspace-runtime-https-live-exercise` is skipped, so its `cleanup` fix did not run. ## Risks - **A large diff in test files.** 328 files change. A pull request that edits the same tests can get merge conflicts. New tests on master can also add type errors before this merges. Any such error fails this check and needs a fix here. - **A higher cost for changes to source types.** A change to a type can add errors in test files that the author did not touch. The author must then fix them. This is the purpose of the check, but it is new work. - **Changed test behavior.** Some tests now run paths that they did not run before, such as the 9 chat wizard fixtures and the 3 span tests. They pass here, but they can be more sensitive to timing or to the environment. - **The Daytona stub can go stale.** The stub is written by hand. If the plugin uses a new SDK member, the typecheck reports an error until someone updates the stub. - **The release path.** `release-verify.yml` and `scripts/release.sh` run `pnpm -r typecheck`, so they now run this check. - **The tsconfig change.** The test transform reads `server/src/__tests__/tsconfig.json`. The options that I changed are for type checks, and the runs above passed. - **A gap that remains.** Tsconfigs in `packages/paperclip-runner` and in most sandbox provider plugins exclude `*.test.ts`. This check does not read most of those test files, so they can still drift. That work is separate from this pull request. > For core feature work, check [`ROADMAP.md`](ROADMAP.md) first and discuss it in `#dev` before opening the PR. Feature PRs that overlap with planned core work may need to be redirected — check the roadmap first. See `CONTRIBUTING.md`. ## Model Used Anthropic Claude Opus 5.5 (`claude-opus-5-5`) through Claude Code, with reasoning, tool use, code execution, and subagents. The runtime does not show the context window size. ## Checklist - [x] I have included a thinking path that traces from project context to this change - [x] I have specified the model used (with version and capability details) - [x] I have checked ROADMAP.md and confirmed this PR does not duplicate planned core work - [x] I have searched GitHub for duplicate or related PRs and linked them above - [x] I have either (a) linked existing issues with `Fixes: #` / `Closes #` / `Refs #` OR (b) described the issue in-PR following the relevant issue template - [x] I have not referenced internal/instance-local Paperclip issues or links (only public GitHub `#NNN` / `github.com/paperclipai/paperclip` URLs) - [x] My branch name describes the change (e.g. `docs/...`, `fix/...`) and contains no internal Paperclip ticket id or instance-derived details - [ ] I have run tests locally and they pass (7 tests fail locally, with and without this change. See Verification.) - [x] I have added or updated tests where applicable - [x] I have updated relevant documentation to reflect my changes - [x] I have considered and documented any risks above - [ ] All Paperclip CI gates are green - [ ] Greptile is 5/5 with no open P2s, recommendations, or follow-ups - [x] I will address all Greptile and reviewer comments before requesting merge 🤖 Generated with [Claude Code](https://claude.com/claude-code) --------- Co-authored-by: Claude Opus 5.5 <noreply@anthropic.com>
727 lines
55 KiB
TypeScript
727 lines
55 KiB
TypeScript
import { notifyChatEndpointConfiguration } from "../services/chat-channels/configuration-work-notifications.js";
|
|
import { deleteCompany } from "../services/company-deletion.js";
|
|
import { toolAccessService } from "../services/tool-access.js";
|
|
import { getEmbeddedPostgresTestSupport, startEmbeddedPostgresTestDatabase } from "./helpers/embedded-postgres.js";
|
|
import { randomUUID, generateKeyPairSync, verify, createHash, diffieHellman, hkdfSync, randomBytes, createCipheriv } from "node:crypto";
|
|
import { createServer, type Server } from "node:http";
|
|
import { once } from "node:events";
|
|
import { afterAll, afterEach, beforeAll, describe, expect, it, vi } from "vitest";
|
|
import { eq, sql } from "drizzle-orm";
|
|
import { applyPendingMigrations, companies, activityLog, companySecrets, connectionEventDeliveries, connectionGrants, createDb, EMBEDDED_POSTGRES_TEST_TIMEOUT_MS, toolApplications, toolConnections } from "@paperclipai/db";
|
|
import { resolveExternalTestDatabaseUrl } from "./helpers/external-test-database.js";
|
|
import { createGitHubGrantContinuity, githubGrantContinuityDeadline, githubContinuityRevision } from "../services/tool-access/github-grant-continuity.js";
|
|
import { createPaperclipCloudConnector } from "../services/paperclip-cloud-connector.js";
|
|
import { githubConnectionEventService } from "../services/github-connection-events.js";
|
|
import { createDeliveryWorkCoordinator } from "../services/delivery-work-coordinator.js";
|
|
import { registerGitHubConnectionContinuityWork, registerGitHubConnectionEventWork } from "../services/github-connection-work.js";
|
|
import { notifyGitHubCredentialChange } from "../services/github-connection-notifications.js";
|
|
import { createOAuthCredentialStore } from "../services/tool-access/oauth/credential-store.js";
|
|
import { createOAuthRefreshLeases } from "../services/tool-access/oauth/refresh-leases.js";
|
|
import { createOAuthGrantRefresh } from "../services/tool-access/oauth/grant-refresh.js";
|
|
import { secretService } from "../services/secrets.js";
|
|
import { writeConnectionCredential } from "../services/connection-credentials.js";
|
|
import { GITHUB_CONNECTOR_PROFILES } from "@paperclipai/shared";
|
|
import { notifyToolConnectionHealth } from "../services/tool-connection-health-notifications.js";
|
|
import { idleWorkSnapshot } from "../services/task-admission.js";
|
|
|
|
const databaseUrl = resolveExternalTestDatabaseUrl("PAPERCLIP_GITHUB_WORK_TEST_DATABASE_URL");
|
|
const embeddedSupport = databaseUrl ? { supported: true } : await getEmbeddedPostgresTestSupport();
|
|
const describeDatabase = embeddedSupport.supported ? describe : describe.skip;
|
|
|
|
describeDatabase("GitHub coordinator with PostgreSQL and controlled Cloud HTTP (simulation)", () => {
|
|
let db: ReturnType<typeof createDb>;
|
|
let temporaryDatabase: Awaited<ReturnType<typeof startEmbeddedPostgresTestDatabase>> | undefined;
|
|
let server: Server, origin: string;
|
|
let leaseTimes: number[] = [];
|
|
let bindings: Array<Record<string, unknown>>, leases: number, failBindings: boolean, githubRequests: number;
|
|
const signing = generateKeyPairSync("ed25519"), sealing = generateKeyPairSync("x25519");
|
|
let releaseRefresh: (() => void) | undefined, refreshArrived: (() => void) | undefined;
|
|
let refreshGate: Promise<void> | undefined;
|
|
let discoveryFailureGate: Promise<void> | undefined, discoveryArrived: (() => void) | undefined;
|
|
let heldBindingCompany: string | undefined, bindingGate: Promise<void> | undefined;
|
|
let bindingArrived: (() => void) | undefined, bindingClosed: (() => void) | undefined;
|
|
const realFetch = globalThis.fetch;
|
|
let companyIds: string[] = [];
|
|
beforeAll(async () => {
|
|
if (databaseUrl) await applyPendingMigrations(databaseUrl);
|
|
else temporaryDatabase = await startEmbeddedPostgresTestDatabase("paperclip-github-work-");
|
|
// Only scheduler time is virtual here. Driver housekeeping must not expire
|
|
// a real socket connection while that fake clock advances across idle windows.
|
|
db = createDb(databaseUrl ?? temporaryDatabase!.connectionString, {
|
|
maxConnections: 4, idleTimeoutSeconds: 0, connectTimeoutSeconds: 0, maxLifetimeSeconds: 0,
|
|
});
|
|
// This explicitly assigned external test database is exclusive to this suite.
|
|
await db.transaction(async tx => { await tx.execute(sql`SET LOCAL client_min_messages = warning`); await tx.execute(sql`TRUNCATE companies CASCADE`); });
|
|
server = createServer(async (req, res) => {
|
|
const chunks: Buffer[] = []; for await (const chunk of req) chunks.push(Buffer.from(chunk));
|
|
const raw = Buffer.concat(chunks).toString();
|
|
if (req.url!.startsWith("/v1/connector/")) {
|
|
const body = JSON.parse(raw), [header, payload, signature] = body.request.split(".");
|
|
expect(verify(null, Buffer.from(`${header}.${payload}`), signing.publicKey, Buffer.from(signature, "base64url"))).toBe(true);
|
|
const claims = JSON.parse(Buffer.from(payload, "base64url").toString());
|
|
expect(claims.iss).toBe("instance"); expect(claims.env).toBe("test");
|
|
expect(claims.aud).toBe(`${origin}${req.url}`);
|
|
if (req.url!.endsWith("/refresh")) {
|
|
refreshArrived?.(); await refreshGate;
|
|
const scopes = [...GITHUB_CONNECTOR_PROFILES["github.code"].scopes];
|
|
const algorithm = "X25519-HKDF-SHA256-A256GCM";
|
|
const purpose = "access";
|
|
const aad = Buffer.from([1, algorithm, purpose, "instance", "test", "github", "github.code", scopes.sort().join(" ")].join("\n"));
|
|
const ephemeral = generateKeyPairSync("x25519");
|
|
const epk = Buffer.from(ephemeral.publicKey.export({ format: "jwk" }).x!, "base64url");
|
|
const recipient = Buffer.from(sealing.publicKey.export({ format: "jwk" }).x!, "base64url");
|
|
const key = Buffer.from(hkdfSync("sha256", diffieHellman({ privateKey: ephemeral.privateKey, publicKey: sealing.publicKey }), Buffer.concat([epk, recipient]), aad, 32));
|
|
const iv = randomBytes(12), cipher = createCipheriv("aes-256-gcm", key, iv); cipher.setAAD(aad);
|
|
const credentials = { v: 1, accessToken: "rotated-access", refreshToken: "rotated-refresh", tokenType: "Bearer", scopes,
|
|
accessTokenExpiresAt: new Date(Date.now() + 8 * 3600_000).toISOString(), refreshTokenExpiresAt: null,
|
|
subject: claims.sub, companyId: claims.cid, instanceId: "instance", environment: "test", provider: "github", profile: "github.code" };
|
|
const ct = Buffer.concat([cipher.update(JSON.stringify(credentials)), cipher.final(), cipher.getAuthTag()]);
|
|
res.setHeader("content-type", "application/json"); res.end(JSON.stringify({ sealed: { v: 1, alg: algorithm, purpose, provider: "github", profile: "github.code", epk: epk.toString("base64url"), iv: iv.toString("base64url"), ct: ct.toString("base64url") } })); return;
|
|
}
|
|
if (req.url!.endsWith("/events/lease")) {
|
|
leases++; leaseTimes.push(Date.now()); res.setHeader("content-type", "application/json");
|
|
res.end(JSON.stringify({ events: [], ...(claims.ewv === 1 ? { eventWakeVersion: 1 } : {}) })); return;
|
|
}
|
|
expect(claims.sh).toBe(createHash("sha256").update(body.binding).digest("base64url"));
|
|
bindings.push(JSON.parse(body.binding));
|
|
if (claims.cid === heldBindingCompany) {
|
|
res.writeHead(200, { "content-type": "application/json" }); res.write("{");
|
|
res.once("close", () => bindingClosed?.()); bindingArrived?.(); await bindingGate;
|
|
res.end("}"); return;
|
|
}
|
|
res.writeHead(failBindings ? 503 : 200, { "content-type": "application/json" }).end("{}"); return;
|
|
}
|
|
if (req.url === "/oauth/token") {
|
|
expect(new URLSearchParams(raw).get("grant_type")).toBe("refresh_token");
|
|
expect(new URLSearchParams(raw).get("refresh_token")).toBe("saved-refresh");
|
|
res.writeHead(200, { "content-type": "application/json" }).end(JSON.stringify({ access_token: "renewed-access", refresh_token: "renewed-refresh", token_type: "Bearer", expires_in: 3600 })); return;
|
|
}
|
|
githubRequests++;
|
|
if (req.url === "/user" && discoveryFailureGate) {
|
|
discoveryArrived?.(); await discoveryFailureGate;
|
|
res.writeHead(503, { "content-type": "application/json", "retry-after": "86400" }).end("{}"); return;
|
|
}
|
|
const body = req.url === "/user" ? { id: 42, login: "octocat" }
|
|
: req.url!.startsWith("/user/installations?") ? { installations: [{ id: 101, repository_selection: "all", account: { login: "octocat" } }] }
|
|
: { total_count: 1, repositories: [{ id: 999, full_name: "octocat/repo" }] };
|
|
res.writeHead(200, { "content-type": "application/json" }).end(JSON.stringify(body));
|
|
});
|
|
server.listen(0, "127.0.0.1"); await once(server, "listening");
|
|
const address = server.address(); if (!address || typeof address === "string") throw new Error("HTTP fixture did not start");
|
|
origin = `http://127.0.0.1:${address.port}`;
|
|
}, EMBEDDED_POSTGRES_TEST_TIMEOUT_MS);
|
|
afterEach(async () => {
|
|
vi.restoreAllMocks(); vi.useRealTimers();
|
|
for (const id of companyIds) { await db.delete(activityLog).where(eq(activityLog.companyId, id)); await db.delete(companySecrets).where(eq(companySecrets.companyId, id)); await db.delete(companies).where(eq(companies.id, id)); } companyIds = [];
|
|
});
|
|
afterAll(async () => { server.close(); await once(server, "close"); await db.$client.end({ timeout: 1 }); await temporaryDatabase?.cleanup(); });
|
|
|
|
async function fixture(ids: string[] = []) {
|
|
bindings = []; leases = 0; leaseTimes = []; failBindings = false; githubRequests = 0;
|
|
const companyId = randomUUID(); companyIds.push(companyId);
|
|
const [company] = await db.insert(companies).values({ id: companyId, name: "GitHub work test", issuePrefix: `GW${companyId.replaceAll("-", "").slice(0, 6).toUpperCase()}` }).returning();
|
|
const [app] = await db.insert(toolApplications).values({ companyId, applicationKey: randomUUID(), name: "GitHub", type: "mcp_http", status: "active" }).returning();
|
|
const [connection] = await db.insert(toolConnections).values({ companyId, applicationId: app!.id, uid: randomUUID(), name: "GitHub", status: "active", enabled: true,
|
|
transport: "mcp_remote", authKind: "oauth", credentialPolicy: "per_user", config: { sourceTemplateKey: "github", oauth: { connectorProfile: "github.code" } }, transportConfig: {} }).returning();
|
|
const [grant] = await db.insert(connectionGrants).values({ companyId, connectionId: connection!.id, kind: "user", subjectUserId: "owner", status: "active",
|
|
credentialSecretRefs: [{ configPath: "oauth.access_token", secretId: randomUUID(), label: "Access" }],
|
|
providerTenant: { oauth: { accessTokenExpiresAt: new Date(Date.now() + 8 * 3600_000).toISOString(), refreshedAt: new Date().toISOString() }, github: {
|
|
userId: "42", login: "octocat", installationCount: ids.length, repositoryCount: 1, repositorySelection: "all", installationIds: ids,
|
|
installationOwnerLogins: ["octocat"], accessRevision: randomUUID(), lastAccessRefreshAt: new Date().toISOString(), webhookHealth: "healthy",
|
|
} } }).returning();
|
|
const connector = createPaperclipCloudConnector({ config: { baseUrl: origin, instanceId: "instance", environment: "test",
|
|
signPrivateKey: signing.privateKey.export({ format: "jwk" }).d!, sealPrivateKey: sealing.privateKey.export({ format: "jwk" }).d! } });
|
|
type Dependencies = Parameters<typeof createGitHubGrantContinuity>[1];
|
|
const dependencies: Dependencies = {
|
|
now: () => new Date(), currentCloudConnector: () => connector,
|
|
refreshOAuthGrantCredentials: async input => (await db.select().from(connectionGrants).where(eq(connectionGrants.id, input.grantId)))[0]!,
|
|
resolveOAuthGrantSecret: (async () => ({ value: "controlled-token", latestVersion: 1 })) as Dependencies["resolveOAuthGrantSecret"],
|
|
};
|
|
const service = createGitHubGrantContinuity(db, dependencies);
|
|
return { company: company!, connection: connection!, grant: grant!, connector, service, dependencies };
|
|
}
|
|
const fakeClock = () => vi.useFakeTimers({ toFake: ["Date", "setTimeout", "clearTimeout"], shouldAdvanceTime: true });
|
|
async function settled(coordinator: ReturnType<typeof createDeliveryWorkCoordinator>) {
|
|
await vi.waitFor(() => expect(idleWorkSnapshot().active).toBe(0));
|
|
expect(coordinator.nextWakeAt()).not.toBeNull();
|
|
}
|
|
|
|
it("ordinary grant refresh activates events from quiet, commits durable result, then performs no owner SQL/HTTP across old cadence windows", async () => {
|
|
const f = await fixture(); fakeClock();
|
|
vi.spyOn(globalThis, "fetch").mockImplementation((url, init) => String(url).startsWith("https://api.github.com")
|
|
? realFetch(`${origin}${new URL(String(url)).pathname}${new URL(String(url)).search}`, init) : realFetch(url, init));
|
|
const errors: unknown[] = [];
|
|
const coordinator = createDeliveryWorkCoordinator({ owner: db, canRun: () => true, onError: error => errors.push(error) });
|
|
const events = githubConnectionEventService(db, { connector: f.connector });
|
|
const ew = registerGitHubConnectionEventWork(db, coordinator, events, () => true);
|
|
const cw = registerGitHubConnectionContinuityWork(db, coordinator, f.service, () => true);
|
|
try {
|
|
await Promise.all([ew.ready, cw.ready]); expect(leases).toBe(1);
|
|
const startupLeases = leases;
|
|
await f.service.refreshManagedGitHubGrantAccess(f.connection, f.grant, { actorType: "system", actorId: "test" });
|
|
await vi.advanceTimersByTimeAsync(6_000);
|
|
await vi.waitFor(() => expect(leases).toBeGreaterThan(startupLeases));
|
|
await vi.advanceTimersByTimeAsync(31_000); await settled(coordinator);
|
|
const [saved] = await db.select().from(connectionGrants).where(eq(connectionGrants.id, f.grant.id));
|
|
expect(saved!.providerTenant!.github!.installationIds).toEqual(["101"]);
|
|
expect(saved!.providerTenant!.github!.bindingSync).toBeUndefined();
|
|
expect(bindings).toContainEqual(expect.objectContaining({ active: true, installationId: "101", id: `${f.grant.id}_101` }));
|
|
const reads = vi.spyOn(db, "select"), writes = vi.spyOn(db, "transaction"), execute = vi.spyOn(db, "execute");
|
|
const before = { leases, githubRequests, bindings: bindings.length };
|
|
await vi.advanceTimersByTimeAsync(10 * 60_000);
|
|
expect(reads).not.toHaveBeenCalled(); expect(writes).not.toHaveBeenCalled(); expect(execute).not.toHaveBeenCalled();
|
|
expect({ leases, githubRequests, bindings: bindings.length }).toEqual(before); expect(errors).toEqual([]);
|
|
} finally { await ew.stop(); await cw.stop(); await coordinator.stop(); }
|
|
});
|
|
|
|
it("failed binding removal persists retry/debt through restart, uses current lease deadline and settles without refetching GitHub metadata", async () => {
|
|
const f = await fixture(["101"]); fakeClock();
|
|
await db.update(connectionGrants).set({ providerTenant: { ...f.grant.providerTenant, github: { ...f.grant.providerTenant!.github!,
|
|
bindingSync: { id: randomUUID(), removeInstallationIds: ["202"] } } } }).where(eq(connectionGrants.id, f.grant.id));
|
|
failBindings = true;
|
|
expect(await f.service.sweepGitHubConnectionContinuity()).toMatchObject({ due: 1, failed: 1 });
|
|
const [failed] = await db.select().from(connectionGrants).where(eq(connectionGrants.id, f.grant.id));
|
|
expect(failed!.providerTenant!.github!.bindingSync!.removeInstallationIds).toEqual(["202"]);
|
|
const retryAt = Date.parse(failed!.providerTenant!.github!.continuityRetry!.at);
|
|
const attemptedRemovals = bindings.filter(binding => binding.installationId === "202").length;
|
|
const restarted = createGitHubGrantContinuity(db, f.dependencies);
|
|
expect(await restarted.nextGitHubConnectionContinuityAt()).toBe(retryAt);
|
|
await vi.advanceTimersByTimeAsync(29_000);
|
|
expect(await restarted.sweepGitHubConnectionContinuity()).toMatchObject({ due: 0 });
|
|
failBindings = false; await vi.advanceTimersByTimeAsync(2_000);
|
|
await Promise.all([restarted.sweepGitHubConnectionContinuity(), restarted.sweepGitHubConnectionContinuity()]);
|
|
const [saved] = await db.select().from(connectionGrants).where(eq(connectionGrants.id, f.grant.id));
|
|
expect(saved!.providerTenant!.github!.bindingSync).toBeUndefined(); expect(saved!.providerTenant!.github!.continuityRetry).toBeUndefined();
|
|
expect(bindings.filter(binding => binding.installationId === "202")).toHaveLength(attemptedRemovals + 1);
|
|
expect(githubRequests).toBe(0);
|
|
expect(await restarted.nextGitHubConnectionContinuityAt()).toBeGreaterThan(Date.now());
|
|
await db.update(toolConnections).set({ enabled: false }).where(eq(toolConnections.id, f.connection.id));
|
|
expect(await restarted.nextGitHubConnectionContinuityAt()).toBeNull();
|
|
await db.update(toolConnections).set({ enabled: true }).where(eq(toolConnections.id, f.connection.id));
|
|
await db.update(companies).set({ status: "archived" }).where(eq(companies.id, f.company.id));
|
|
expect(await restarted.nextGitHubConnectionContinuityAt()).toBeNull();
|
|
});
|
|
|
|
it.each(["credential", "installation"])("does not attach a withheld provider failure to newer %s work", async change => {
|
|
const f = await fixture(["101"]);
|
|
vi.spyOn(globalThis, "fetch").mockImplementation((url, init) => String(url).startsWith("https://api.github.com")
|
|
? realFetch(`${origin}${new URL(String(url)).pathname}${new URL(String(url)).search}`, init) : realFetch(url, init));
|
|
await db.update(connectionGrants).set({ providerTenant: { ...f.grant.providerTenant,
|
|
oauth: { ...f.grant.providerTenant!.oauth, accessTokenExpiresAt: new Date(Date.now() + 30 * 60_000).toISOString() } } }).where(eq(connectionGrants.id, f.grant.id));
|
|
let release!: () => void;
|
|
const arrived = new Promise<void>(resolve => { discoveryArrived = resolve; });
|
|
discoveryFailureGate = new Promise<void>(resolve => { release = resolve; });
|
|
const pending = f.service.sweepGitHubConnectionContinuity(f.company.id);
|
|
const finished = Promise.allSettled([pending]);
|
|
try {
|
|
await Promise.race([arrived, pending.then(() => { throw new Error("Expected held provider request"); })]);
|
|
const revision = randomUUID();
|
|
const expiry = new Date(Date.now() + 10 * 60_000).toISOString();
|
|
await db.transaction(async tx => {
|
|
await notifyGitHubCredentialChange(tx, f.connection);
|
|
await tx.update(connectionGrants).set({ providerTenant: change === "credential"
|
|
? sql`jsonb_set(${connectionGrants.providerTenant}, '{oauth,accessTokenExpiresAt}', ${JSON.stringify(expiry)}::jsonb)`
|
|
: sql`jsonb_set(${connectionGrants.providerTenant}, '{github,bindingSync}', ${JSON.stringify({ id: revision, removeInstallationIds: ["202"] })}::jsonb)`,
|
|
updatedAt: new Date() }).where(eq(connectionGrants.id, f.grant.id));
|
|
});
|
|
release();
|
|
expect(await pending).toMatchObject({ due: 1, failed: 1 });
|
|
const [saved] = await db.select().from(connectionGrants).where(eq(connectionGrants.id, f.grant.id));
|
|
expect(saved!.providerTenant!.github!.continuityRetry).toBeUndefined();
|
|
if (change === "credential") expect(saved!.providerTenant!.oauth!.accessTokenExpiresAt).toBe(expiry);
|
|
else expect(saved!.providerTenant!.github!.bindingSync!.id).toBe(revision);
|
|
expect(githubGrantContinuityDeadline(f.connection, saved!)).toBeLessThanOrEqual(Date.now());
|
|
discoveryFailureGate = undefined;
|
|
const restarted = createGitHubGrantContinuity(db, f.dependencies);
|
|
expect(await restarted.sweepGitHubConnectionContinuity(f.company.id)).toMatchObject({ due: 1, refreshed: 1, failed: 0 });
|
|
expect(bindings).toContainEqual(expect.objectContaining({ installationId: "101", active: true }));
|
|
if (change === "installation") expect(bindings).toContainEqual(expect.objectContaining({ installationId: "202", active: false }));
|
|
const [settled] = await db.select().from(connectionGrants).where(eq(connectionGrants.id, f.grant.id));
|
|
expect(settled!.providerTenant!.github!.bindingSync).toBeUndefined();
|
|
const next = await restarted.nextGitHubConnectionContinuityAt(f.company.id);
|
|
expect(next).toBeLessThan(Date.now() + 24 * 60 * 60_000);
|
|
} finally { release(); await finished; discoveryFailureGate = undefined; discoveryArrived = undefined; }
|
|
});
|
|
|
|
it("retains provider backoff for the actual attempted revision after its own credential rotation", async () => {
|
|
const f = await fixture(["101"]);
|
|
await db.update(connectionGrants).set({ providerTenant: { ...f.grant.providerTenant,
|
|
oauth: { ...f.grant.providerTenant!.oauth, accessTokenExpiresAt: new Date(Date.now() + 30 * 60_000).toISOString() } } }).where(eq(connectionGrants.id, f.grant.id));
|
|
vi.spyOn(globalThis, "fetch").mockImplementation((url, init) => String(url).startsWith("https://api.github.com")
|
|
? realFetch(`${origin}${new URL(String(url)).pathname}${new URL(String(url)).search}`, init) : realFetch(url, init));
|
|
const owner = createGitHubGrantContinuity(db, { ...f.dependencies, refreshOAuthGrantCredentials: async () => {
|
|
const [rotated] = await db.update(connectionGrants).set({ providerTenant: sql`jsonb_set(${connectionGrants.providerTenant}, '{oauth}', ${JSON.stringify({
|
|
accessTokenExpiresAt: new Date(Date.now() + 10 * 60_000).toISOString(), refreshedAt: new Date().toISOString(),
|
|
})}::jsonb)` }).where(eq(connectionGrants.id, f.grant.id)).returning();
|
|
return rotated!;
|
|
} });
|
|
discoveryFailureGate = Promise.resolve();
|
|
try {
|
|
expect(await owner.sweepGitHubConnectionContinuity(f.company.id)).toMatchObject({ due: 1, failed: 1 });
|
|
const [saved] = await db.select().from(connectionGrants).where(eq(connectionGrants.id, f.grant.id));
|
|
expect(saved!.providerTenant!.github!.continuityRetry!.revision).toBe(githubContinuityRevision(saved!));
|
|
expect(await owner.nextGitHubConnectionContinuityAt(f.company.id)).toBeGreaterThan(Date.now() + 23 * 60 * 60_000);
|
|
const calls = githubRequests;
|
|
expect(await createGitHubGrantContinuity(db, f.dependencies).sweepGitHubConnectionContinuity(f.company.id)).toMatchObject({ due: 0 });
|
|
expect(githubRequests).toBe(calls);
|
|
} finally { discoveryFailureGate = undefined; }
|
|
});
|
|
|
|
it.each([["suspend", "unsuspend"], ["unsuspend", "suspend"]])("reconciles same-batch installation order %s then %s and duplicate replay", async (first, last) => {
|
|
const f = await fixture(["101"]);
|
|
await db.update(connectionGrants).set({ providerTenant: { ...f.grant.providerTenant,
|
|
github: { ...f.grant.providerTenant!.github!, bindingSync: { id: randomUUID(), removeInstallationIds: ["202"] } } } }).where(eq(connectionGrants.id, f.grant.id));
|
|
const events = [first, last].map(action => ({ id: randomUUID(), provider: "github" as const, event: "installation", action,
|
|
installationId: "101", repositoryId: null, createdAt: new Date(Date.now() + 1000).toISOString(), bindingIds: [`${f.grant.id}_101`], payload: { installationId: "101" } }));
|
|
const acknowledge = vi.fn(async () => events.length);
|
|
const ingress = githubConnectionEventService(db, { connector: { ...f.connector, leaseEvents: async () => ({ leaseId: "installation-batch", events }), acknowledgeEvents: acknowledge } });
|
|
expect(await ingress.pollOnce({ forDelivery: true })).toMatchObject({ processed: 2, failed: 0 });
|
|
const [saved] = await db.select().from(connectionGrants).where(eq(connectionGrants.id, f.grant.id));
|
|
const active = last === "unsuspend";
|
|
expect(saved!.providerTenant!.github!.installationIds).toEqual(active ? ["101"] : []);
|
|
expect(saved!.providerTenant!.github!.bindingSync!.removeInstallationIds.sort()).toEqual(active ? ["202"] : ["101", "202"]);
|
|
expect(githubGrantContinuityDeadline(f.connection, saved!)).toBe(0);
|
|
await f.service.syncGitHubGrantBindings(f.connection, saved!);
|
|
expect(bindings.filter(binding => binding.installationId === "101")).toEqual([expect.objectContaining({ active })]);
|
|
expect(bindings.filter(binding => binding.installationId === "202")).toEqual([expect.objectContaining({ active: false })]);
|
|
const [settled] = await db.select().from(connectionGrants).where(eq(connectionGrants.id, f.grant.id));
|
|
expect(settled!.providerTenant!.github!.bindingSync).toBeUndefined();
|
|
// Batch duplicates must retain terminal receipts and emit no registration changes.
|
|
expect(await ingress.pollOnce({ forDelivery: true })).toMatchObject({ processed: 0, duplicate: active ? 2 : 0, ignored: active ? 0 : 2, failed: 0 });
|
|
await f.service.syncGitHubGrantBindings(f.connection, saved!);
|
|
expect(bindings).toHaveLength(2);
|
|
expect((await db.select().from(connectionGrants).where(eq(connectionGrants.id, f.grant.id)))[0]!.providerTenant).toEqual(settled!.providerTenant);
|
|
expect((await db.select().from(connectionEventDeliveries).where(eq(connectionEventDeliveries.companyId, f.company.id))).map(receipt => receipt.status)).toEqual(["processed", "processed"]);
|
|
expect(acknowledge).toHaveBeenCalledTimes(2);
|
|
});
|
|
|
|
it("does not retire active installations from a legacy conflicting binding intent", async () => {
|
|
const f = await fixture(["101"]);
|
|
const [saved] = await db.update(connectionGrants).set({ providerTenant: { ...f.grant.providerTenant,
|
|
github: { ...f.grant.providerTenant!.github!, bindingSync: { id: randomUUID(), removeInstallationIds: ["101", "202"] } } } }).where(eq(connectionGrants.id, f.grant.id)).returning();
|
|
await f.service.syncGitHubGrantBindings(f.connection, saved!);
|
|
expect(bindings.filter(binding => binding.installationId === "101")).toEqual([expect.objectContaining({ active: true })]);
|
|
expect(bindings.filter(binding => binding.installationId === "202")).toEqual([expect.objectContaining({ active: false })]);
|
|
});
|
|
|
|
it("preserves webhook removal debt across an actual signed, held Cloud token rotation and lease cleanup", async () => {
|
|
const f = await fixture(["101"]);
|
|
vi.stubEnv("PAPERCLIP_SECRETS_MASTER_KEY", "01234567890123456789012345678901");
|
|
const refs = [];
|
|
for (const path of ["oauth.access_token", "oauth.refresh_token"]) {
|
|
const saved = await writeConnectionCredential(db, { companyId: f.company.id, connectionName: "GitHub", configPath: path, label: path, value: path, ownerUserId: "owner" });
|
|
refs.push({ configPath: path, secretId: saved.secret.id, versionSelector: "latest" as const });
|
|
}
|
|
const [connection] = await db.update(toolConnections).set({ credentialSource: "paperclip_vault", config: { ...f.connection.config,
|
|
oauth: { strategy: "paperclip_cloud_connector", connectorProfile: "github.code" } } }).where(eq(toolConnections.id, f.connection.id)).returning();
|
|
const [grant] = await db.update(connectionGrants).set({ credentialSecretRefs: refs }).where(eq(connectionGrants.id, f.grant.id)).returning();
|
|
const getConnection = async () => (await db.select().from(toolConnections).where(eq(toolConnections.id, connection!.id)))[0]!;
|
|
const store = createOAuthCredentialStore(db, secretService(db));
|
|
await store.syncCredentialBindings(connection!, refs);
|
|
const leases = createOAuthRefreshLeases(db, () => new Date(), getConnection, store.getOAuthGrantRow);
|
|
const rotation = createOAuthGrantRefresh(db, { now: () => new Date(), store, leases, getConnectionRow: getConnection,
|
|
currentCloudConnector: () => f.connector, exchangeOAuthToken: vi.fn() });
|
|
const arrived = new Promise<void>(resolve => { refreshArrived = resolve; });
|
|
refreshGate = new Promise<void>(resolve => { releaseRefresh = resolve; });
|
|
const pending = rotation({ companyId: f.company.id, connectionId: connection!.id, grantId: grant!.id, forceRefresh: true });
|
|
const result = Promise.allSettled([pending]);
|
|
try {
|
|
await Promise.race([arrived, pending.then(() => { throw new Error("Rotation returned before the Cloud HTTP boundary"); })]);
|
|
const connector = { ...f.connector, acknowledgeEvents: async () => 1, leaseEvents: async () => ({ leaseId: "removal", events: [{ id: randomUUID(), provider: "github" as const,
|
|
event: "installation", action: "deleted", installationId: "101", repositoryId: null, createdAt: new Date(Date.now() + 1000).toISOString(), bindingIds: [`${grant!.id}_101`], payload: { installationId: "101" } }] }) };
|
|
expect(await githubConnectionEventService(db, { connector }).pollOnce()).toMatchObject({ processed: 1 });
|
|
const [before] = await db.select().from(connectionGrants).where(eq(connectionGrants.id, grant!.id));
|
|
expect(before!.providerTenant!.github!.bindingSync!.removeInstallationIds).toEqual(["101"]);
|
|
releaseRefresh!();
|
|
const [completion] = await result; if (completion.status === "rejected") throw completion.reason;
|
|
const [saved] = await db.select().from(connectionGrants).where(eq(connectionGrants.id, grant!.id));
|
|
expect(saved!.providerTenant!.github).toEqual(before!.providerTenant!.github);
|
|
expect(saved!.providerTenant!.oauth).toMatchObject({ tokenType: "Bearer" });
|
|
expect(saved!.providerTenant!.oauth!.refreshLease).toBeUndefined();
|
|
await f.service.syncGitHubGrantBindings(connection!, saved!);
|
|
expect(bindings).toContainEqual(expect.objectContaining({ installationId: "101", active: false }));
|
|
expect((await db.select().from(connectionGrants).where(eq(connectionGrants.id, grant!.id)))[0]!.providerTenant!.github!.bindingSync).toBeUndefined();
|
|
} finally { releaseRefresh?.(); await result; refreshGate = undefined; refreshArrived = undefined; vi.unstubAllEnvs(); }
|
|
});
|
|
|
|
it("bounds a held Cloud body without releasing revocation admission, while another company progresses and restart retains removal debt", async () => {
|
|
const f = await fixture([]), other = await fixture([]);
|
|
for (const row of [f, other]) await db.update(connectionGrants).set({ providerTenant: {
|
|
...row.grant.providerTenant, github: { ...row.grant.providerTenant!.github!, bindingSync: { id: randomUUID(), removeInstallationIds: ["101"] } },
|
|
} }).where(eq(connectionGrants.id, row.grant.id));
|
|
const latest = async (id: string) => (await db.select().from(connectionGrants).where(eq(connectionGrants.id, id)))[0]!;
|
|
heldBindingCompany = f.company.id;
|
|
let release!: () => void;
|
|
bindingGate = new Promise<void>(resolve => { release = resolve; });
|
|
const arrived = new Promise<void>(resolve => { bindingArrived = resolve; });
|
|
const closed = new Promise<void>(resolve => { bindingClosed = resolve; });
|
|
const timeout = AbortSignal.timeout.bind(AbortSignal);
|
|
const deadlines = vi.spyOn(AbortSignal, "timeout").mockImplementation(ms => timeout(ms === 15_000 ? 3_000 : ms));
|
|
let completed = false;
|
|
const attempt = f.service.syncGitHubGrantBindings(f.connection, await latest(f.grant.id));
|
|
const outcome = attempt.then(value => ({ value, error: null }), error => ({ value: null, error })).finally(() => { completed = true; });
|
|
try {
|
|
await arrived;
|
|
const otherResult = await other.service.syncGitHubGrantBindings(other.connection, await latest(other.grant.id));
|
|
expect(otherResult.providerTenant!.github!.bindingSync).toBeUndefined();
|
|
await db.update(companies).set({ name: "Unrelated company changed while body held" }).where(eq(companies.id, other.company.id));
|
|
expect(completed).toBe(false);
|
|
expect((await outcome).error).toMatchObject({ code: "CONNECTOR_BAD_RESPONSE" });
|
|
await closed;
|
|
expect(deadlines).toHaveBeenCalledWith(15_000);
|
|
expect((await latest(f.grant.id)).providerTenant!.github!.bindingSync!.removeInstallationIds).toEqual(["101"]);
|
|
heldBindingCompany = undefined; release();
|
|
const restarted = createGitHubGrantContinuity(db, f.dependencies);
|
|
expect(await restarted.sweepGitHubConnectionContinuity(f.company.id)).toMatchObject({ due: 1, refreshed: 1, failed: 0 });
|
|
expect((await latest(f.grant.id)).providerTenant!.github!.bindingSync).toBeUndefined();
|
|
} finally { heldBindingCompany = undefined; release(); await outcome; bindingGate = undefined; bindingArrived = undefined; bindingClosed = undefined; }
|
|
});
|
|
|
|
it("serializes binding handoff behind canonical company deletion without a child lock cycle", async () => {
|
|
const f = await fixture([]);
|
|
const [pending] = await db.update(connectionGrants).set({ providerTenant: {
|
|
...f.grant.providerTenant, github: { ...f.grant.providerTenant!.github!,
|
|
bindingSync: { id: randomUUID(), removeInstallationIds: ["101"] } },
|
|
} }).where(eq(connectionGrants.id, f.grant.id)).returning();
|
|
let release!: () => void, entered!: (pid: number) => void;
|
|
const gate = new Promise<void>(resolve => { release = resolve; });
|
|
const companyLocked = new Promise<number>(resolve => { entered = resolve; });
|
|
const transaction = db.transaction.bind(db);
|
|
// Pause after the canonical deletion's actual company lock, before its child writes.
|
|
const spy = vi.spyOn(db, "transaction").mockImplementation(((callback: (tx: unknown) => Promise<unknown>) =>
|
|
transaction(async tx => {
|
|
const select = tx.select.bind(tx);
|
|
tx.select = ((...args: Parameters<typeof select>) => {
|
|
const query = select(...args), from = query.from.bind(query);
|
|
query.from = ((table: unknown) => {
|
|
const builder = from(table as typeof companies);
|
|
if (table === companies) {
|
|
const lock = builder.for.bind(builder);
|
|
builder.for = ((strength: Parameters<typeof lock>[0]) => {
|
|
const locked = lock(strength);
|
|
if (strength === "no key update") {
|
|
const then = locked.then.bind(locked);
|
|
locked.then = ((resolve: (rows: unknown) => unknown, reject: (error: unknown) => unknown) => then(async rows => {
|
|
const [owner] = await tx.execute(sql`select pg_backend_pid() as pid`);
|
|
entered(Number(owner!.pid)); await gate; return resolve(rows);
|
|
}, reject)) as typeof locked.then;
|
|
}
|
|
return locked;
|
|
}) as typeof builder.for;
|
|
}
|
|
return builder;
|
|
}) as typeof query.from;
|
|
return query;
|
|
}) as typeof tx.select;
|
|
return callback(tx);
|
|
})) as typeof db.transaction);
|
|
const deletion = deleteCompany(db, f.company.id);
|
|
const deleted = Promise.allSettled([deletion]);
|
|
let handoff: Promise<unknown> | undefined;
|
|
try {
|
|
const deletionPid = await companyLocked;
|
|
handoff = f.service.syncGitHubGrantBindings(f.connection, pending!);
|
|
const completed = Promise.allSettled([deletion, handoff]);
|
|
await vi.waitFor(async () => {
|
|
const waiting = await db.execute(sql`select pid from pg_stat_activity
|
|
where datname = current_database() and query like '%"companies"%' and query like '%for share%'
|
|
and ${deletionPid} = any(pg_blocking_pids(pid))`);
|
|
expect(waiting).toHaveLength(1);
|
|
}, { timeout: 5_000, interval: 20 });
|
|
release();
|
|
const results = await completed;
|
|
expect(results.map(row => row.status), JSON.stringify(results)).toEqual(["fulfilled", "fulfilled"]);
|
|
expect(await db.select().from(companies).where(eq(companies.id, f.company.id))).toHaveLength(0);
|
|
expect(await db.select().from(connectionGrants).where(eq(connectionGrants.id, f.grant.id))).toHaveLength(0);
|
|
expect(bindings).toHaveLength(0);
|
|
} finally { release(); await deleted; await handoff?.catch(() => {}); spy.mockRestore(); }
|
|
});
|
|
|
|
it("lets an admitted binding finish before canonical company deletion", async () => {
|
|
const f = await fixture([]);
|
|
const [pending] = await db.update(connectionGrants).set({ providerTenant: {
|
|
...f.grant.providerTenant, github: { ...f.grant.providerTenant!.github!,
|
|
bindingSync: { id: randomUUID(), removeInstallationIds: ["101"] } },
|
|
} }).where(eq(connectionGrants.id, f.grant.id)).returning();
|
|
heldBindingCompany = f.company.id;
|
|
let release!: () => void;
|
|
bindingGate = new Promise<void>(resolve => { release = resolve; });
|
|
const arrived = new Promise<void>(resolve => { bindingArrived = resolve; });
|
|
const handoff = f.service.syncGitHubGrantBindings(f.connection, pending!);
|
|
const handedOff = Promise.allSettled([handoff]);
|
|
let deletion: Promise<unknown> | undefined;
|
|
try {
|
|
await arrived;
|
|
deletion = deleteCompany(db, f.company.id);
|
|
const deleted = Promise.allSettled([deletion]);
|
|
await vi.waitFor(async () => {
|
|
const waiting = await db.execute(sql`select pid from pg_stat_activity
|
|
where datname = current_database() and query like '%"companies"%' and query like '%for no key update%'
|
|
and cardinality(pg_blocking_pids(pid)) > 0`);
|
|
expect(waiting).toHaveLength(1);
|
|
}, { timeout: 5_000, interval: 20 });
|
|
release();
|
|
expect((await handedOff)[0]!.status).toBe("fulfilled");
|
|
expect((await deleted)[0]!.status).toBe("fulfilled");
|
|
expect(bindings).toEqual([expect.objectContaining({ grantId: f.grant.id, connectionId: f.connection.id, installationId: "101", active: false })]);
|
|
expect(await db.select().from(companies).where(eq(companies.id, f.company.id))).toHaveLength(0);
|
|
expect(await db.select().from(connectionGrants).where(eq(connectionGrants.id, f.grant.id))).toHaveLength(0);
|
|
} finally { release(); await handedOff; await deletion?.catch(() => {}); heldBindingCompany = undefined;
|
|
bindingGate = undefined; bindingArrived = undefined; }
|
|
});
|
|
|
|
it.each(["revocation", "revision", "rollback"] as const)("revalidates a late %s after waiting for company admission", async change => {
|
|
const f = await fixture([]), revision = randomUUID();
|
|
const [pending] = await db.update(connectionGrants).set({ providerTenant: {
|
|
...f.grant.providerTenant, github: { ...f.grant.providerTenant!.github!,
|
|
bindingSync: { id: randomUUID(), removeInstallationIds: ["101"] } },
|
|
} }).where(eq(connectionGrants.id, f.grant.id)).returning();
|
|
let entered!: (pid: number) => void, release!: () => void;
|
|
const locked = new Promise<number>(resolve => { entered = resolve; });
|
|
const gate = new Promise<void>(resolve => { release = resolve; });
|
|
const mutation = db.transaction(async tx => {
|
|
await tx.select().from(companies).where(eq(companies.id, f.company.id)).for("no key update");
|
|
const [owner] = await tx.execute(sql`select pg_backend_pid() as pid`);
|
|
entered(Number(owner!.pid)); await gate;
|
|
if (change === "revision") await tx.update(connectionGrants).set({ providerTenant: {
|
|
...pending!.providerTenant, github: { ...pending!.providerTenant!.github!,
|
|
bindingSync: { id: revision, removeInstallationIds: ["202"] } },
|
|
} }).where(eq(connectionGrants.id, f.grant.id));
|
|
else await tx.update(connectionGrants).set({ status: "revoked" }).where(eq(connectionGrants.id, f.grant.id));
|
|
if (change === "rollback") throw new Error("rollback late revocation");
|
|
});
|
|
const mutated = Promise.allSettled([mutation]);
|
|
let handoff: Promise<unknown> | undefined;
|
|
try {
|
|
const ownerPid = await locked;
|
|
handoff = f.service.syncGitHubGrantBindings(f.connection, pending!);
|
|
const handedOff = Promise.allSettled([handoff]);
|
|
await vi.waitFor(async () => {
|
|
const waiting = await db.execute(sql`select pid from pg_stat_activity
|
|
where datname = current_database() and query like '%"companies"%' and query like '%for share%'
|
|
and ${ownerPid} = any(pg_blocking_pids(pid))`);
|
|
expect(waiting).toHaveLength(1);
|
|
}, { timeout: 5_000, interval: 20 });
|
|
release();
|
|
expect((await mutated)[0]!.status).toBe(change === "rollback" ? "rejected" : "fulfilled");
|
|
expect((await handedOff)[0]!.status).toBe("fulfilled");
|
|
const [saved] = await db.select().from(connectionGrants).where(eq(connectionGrants.id, f.grant.id));
|
|
if (change === "rollback") {
|
|
expect(bindings).toEqual([expect.objectContaining({ grantId: f.grant.id, installationId: "101", active: false })]);
|
|
expect(saved!.status).toBe("active"); expect(saved!.providerTenant!.github!.bindingSync).toBeUndefined();
|
|
} else {
|
|
expect(bindings).toHaveLength(0);
|
|
if (change === "revocation") expect(saved!.status).toBe("revoked");
|
|
else expect(saved!.providerTenant!.github!.bindingSync).toEqual({ id: revision, removeInstallationIds: ["202"] });
|
|
}
|
|
} finally { release(); await mutated; await handoff?.catch(() => {}); }
|
|
});
|
|
|
|
it("renews an installation created without metadata using the saved connection expiry", async () => {
|
|
const f = await fixture();
|
|
vi.stubEnv("PAPERCLIP_SECRETS_MASTER_KEY", "01234567890123456789012345678901");
|
|
try {
|
|
const refs = [];
|
|
for (const path of ["oauth.access_token", "oauth.refresh_token"]) {
|
|
const saved = await writeConnectionCredential(db, { companyId: f.company.id, connectionName: "OAuth", configPath: path, label: path,
|
|
value: path === "oauth.refresh_token" ? "saved-refresh" : "expired-access" });
|
|
refs.push({ configPath: path, secretId: saved.secret.id, versionSelector: "latest" as const });
|
|
}
|
|
await db.update(toolConnections).set({ credentialSource: "paperclip_vault", credentialPolicy: "shared", config: { oauth: {
|
|
provider: "fixture", tokenUrl: `${origin}/oauth/token`, clientId: "fixture-client", clientRegistrationSource: "dcr", clientTokenEndpointAuthMethod: "none",
|
|
expiresAt: new Date(Date.now() - 60_000).toISOString(),
|
|
} } }).where(eq(toolConnections.id, f.connection.id));
|
|
const api = toolAccessService(db);
|
|
const installation = await api.addConnectionInstallation(f.connection.id, { credentialSecretRefs: refs });
|
|
const [before] = await db.select().from(connectionGrants).where(eq(connectionGrants.id, installation.id));
|
|
expect(before!.providerTenant).toBeNull();
|
|
expect((await db.execute(sql`select provider_tenant is null as absent from connection_grants where id = ${installation.id}`))[0]!.absent).toBe(true);
|
|
const renewed = await api.refreshOAuthGrantCredentials({ companyId: f.company.id, connectionId: f.connection.id, grantId: installation.id });
|
|
expect(renewed.providerTenant!.oauth).toMatchObject({ tokenType: "Bearer", accessTokenExpiresAt: expect.any(String) });
|
|
expect(Date.parse(renewed.providerTenant!.oauth!.accessTokenExpiresAt!)).toBeGreaterThan(Date.now() + 3500_000);
|
|
expect(renewed.providerTenant!.oauth!.refreshLease).toBeUndefined();
|
|
expect((await db.select().from(connectionGrants).where(eq(connectionGrants.id, installation.id)))[0]!.providerTenant).toEqual(renewed.providerTenant);
|
|
} finally { vi.unstubAllEnvs(); }
|
|
}, 40_000);
|
|
|
|
it.each(["sql-null", "json-null", "object"] as const)("claims current %s metadata, fences stale snapshots and rolls back lease writes", async mode => {
|
|
const f = await fixture();
|
|
const api = toolAccessService(db);
|
|
const installation = await api.addConnectionInstallation(f.connection.id, {});
|
|
const metadata = mode === "sql-null" ? sql`null` : mode === "json-null" ? sql`'null'::jsonb` : sql`'{"name":"current"}'::jsonb`;
|
|
await db.update(connectionGrants).set({ providerTenant: metadata }).where(eq(connectionGrants.id, installation.id));
|
|
const getGrant = async () => (await db.select().from(connectionGrants).where(eq(connectionGrants.id, installation.id)))[0]!;
|
|
const initial = await getGrant();
|
|
const leases = createOAuthRefreshLeases(db, () => new Date(), async () => f.connection, getGrant);
|
|
const claimed = await leases.acquireOAuthGrantRefreshLease(f.connection, initial, true);
|
|
expect(claimed.leaseId).toBeTruthy();
|
|
expect(claimed.grant.providerTenant!.oauth!.refreshLease!.id).toBe(claimed.leaseId);
|
|
if (mode === "object") expect(claimed.grant.providerTenant!.name).toBe("current");
|
|
await leases.clearOAuthGrantRefreshLease(claimed.grant, claimed.leaseId!);
|
|
await db.update(connectionGrants).set({ providerTenant: metadata }).where(eq(connectionGrants.id, installation.id));
|
|
const newer = { name: "concurrent", github: { ...f.grant.providerTenant!.github!, bindingSync: { id: randomUUID(), removeInstallationIds: ["202"] } } };
|
|
let raced = false;
|
|
const staleRead = async () => {
|
|
const snapshot = await getGrant();
|
|
if (!raced) { raced = true; await db.update(connectionGrants).set({ providerTenant: newer }).where(eq(connectionGrants.id, installation.id)); }
|
|
return snapshot;
|
|
};
|
|
const racingLeases = createOAuthRefreshLeases(db, () => new Date(), async () => f.connection, staleRead);
|
|
const retained = await racingLeases.acquireOAuthGrantRefreshLease(f.connection, initial, true);
|
|
expect(retained.grant.providerTenant).toMatchObject(newer);
|
|
expect(retained.leaseId).toBeTruthy();
|
|
await racingLeases.clearOAuthGrantRefreshLease(retained.grant, retained.leaseId!);
|
|
await db.update(connectionGrants).set({ providerTenant: metadata }).where(eq(connectionGrants.id, installation.id));
|
|
await expect(db.transaction(async tx => {
|
|
const read = async () => (await tx.select().from(connectionGrants).where(eq(connectionGrants.id, installation.id)))[0]!;
|
|
const transactional = createOAuthRefreshLeases(tx as unknown as typeof db, () => new Date(), async () => f.connection, read);
|
|
const lease = await transactional.acquireOAuthGrantRefreshLease(f.connection, await read(), true);
|
|
expect(lease.leaseId).toBeTruthy();
|
|
throw new Error("rollback lease");
|
|
})).rejects.toThrow("rollback lease");
|
|
expect((await getGrant()).providerTenant).toEqual(initial.providerTenant);
|
|
const retried = await leases.acquireOAuthGrantRefreshLease(f.connection, initial, true);
|
|
expect(retried.leaseId).toBeTruthy();
|
|
await leases.clearOAuthGrantRefreshLease(retried.grant, retried.leaseId!);
|
|
}, 40_000);
|
|
|
|
it("lease acquire and cleanup cannot replace a concurrent webhook metadata revision", async () => {
|
|
const f = await fixture(["101"]);
|
|
let interleave = true;
|
|
const removal = { id: randomUUID(), removeInstallationIds: ["202"] };
|
|
const getGrant = async () => {
|
|
const [snapshot] = await db.select().from(connectionGrants).where(eq(connectionGrants.id, f.grant.id));
|
|
if (interleave) {
|
|
interleave = false;
|
|
await db.update(connectionGrants).set({ providerTenant: sql`jsonb_set(${connectionGrants.providerTenant}, '{github,bindingSync}', ${JSON.stringify(removal)}::jsonb)` }).where(eq(connectionGrants.id, f.grant.id));
|
|
}
|
|
return snapshot!;
|
|
};
|
|
const leaseOwner = createOAuthRefreshLeases(db, () => new Date(), async () => f.connection, getGrant);
|
|
const lease = await leaseOwner.acquireOAuthGrantRefreshLease(f.connection, f.grant, true);
|
|
expect(lease.leaseId).toBeTruthy();
|
|
expect(lease.grant.providerTenant!.github!.bindingSync).toEqual(removal);
|
|
// Cleanup deliberately reads a stale snapshot immediately before a new delta.
|
|
const second = { ...removal, id: randomUUID(), removeInstallationIds: ["202", "303"] };
|
|
const cleanup = createOAuthRefreshLeases(db, () => new Date(), async () => f.connection, async () => {
|
|
const snapshot = await getGrant();
|
|
await db.update(connectionGrants).set({ providerTenant: sql`jsonb_set(${connectionGrants.providerTenant}, '{github,bindingSync}', ${JSON.stringify(second)}::jsonb)` }).where(eq(connectionGrants.id, f.grant.id));
|
|
return snapshot;
|
|
});
|
|
const cleared = await cleanup.clearOAuthGrantRefreshLease(lease.grant, lease.leaseId!);
|
|
expect(cleared.providerTenant!.github!.bindingSync).toEqual(second);
|
|
expect(cleared.providerTenant!.oauth!.refreshLease).toBeUndefined();
|
|
});
|
|
|
|
it("ignores unrelated health write bursts and scopes credential reconstruction to its owning company", async () => {
|
|
const f = await fixture(["101"]), other = await fixture(["202"]); fakeClock();
|
|
// A large unrelated company set must not enter a targeted credential reload.
|
|
for (let index = 0; index < 50; index++) {
|
|
const id = randomUUID(); companyIds.push(id);
|
|
await db.insert(companies).values({ id, name: "Unrelated credential owner", issuePrefix: `U${index}${id.slice(0, 6)}` });
|
|
}
|
|
const [unrelatedConnection] = await db.update(toolConnections).set({ config: {}, transportConfig: {}, transport: "rest_api", authKind: "none", credentialRefs: [], credentialSecretRefs: [] }).where(eq(toolConnections.id, other.connection.id)).returning();
|
|
await db.update(connectionGrants).set({ credentialSecretRefs: [] }).where(eq(connectionGrants.id, other.grant.id));
|
|
const api = toolAccessService(db);
|
|
const loadTimes: number[] = [];
|
|
const schedule = f.service.githubConnectionContinuitySchedule.bind(f.service);
|
|
const load = vi.spyOn(f.service, "githubConnectionContinuitySchedule").mockImplementation(companyId => { loadTimes.push(Date.now()); return schedule(companyId); });
|
|
const sweep = vi.spyOn(f.service, "sweepGitHubConnectionContinuity");
|
|
const errors: unknown[] = [];
|
|
const coordinator = createDeliveryWorkCoordinator({ owner: db, canRun: () => true, onError: error => errors.push(error) });
|
|
const continuity = registerGitHubConnectionContinuityWork(db, coordinator, f.service, () => true);
|
|
const events = registerGitHubConnectionEventWork(db, coordinator, githubConnectionEventService(db, { connector: f.connector }), () => true);
|
|
try {
|
|
await Promise.all([continuity.ready, events.ready]);
|
|
await vi.waitFor(() => expect(leases).toBe(1));
|
|
await vi.waitFor(() => expect(load).toHaveBeenCalledTimes(1));
|
|
load.mockClear(); loadTimes.length = 0; sweep.mockClear();
|
|
for (let i = 0; i < 30; i++) {
|
|
await db.transaction(async tx => { await notifyToolConnectionHealth(tx, other.connection); await notifyChatEndpointConfiguration(tx, { id: randomUUID(), provider: "discord" }); });
|
|
await api.updateConnection(unrelatedConnection!.id, { name: `Unrelated patch ${i}` });
|
|
}
|
|
await vi.advanceTimersByTimeAsync(10_000);
|
|
expect(load).not.toHaveBeenCalled(); expect(sweep).not.toHaveBeenCalled(); expect(leases).toBe(1);
|
|
expect(errors).toEqual([]);
|
|
for (let index = 0; index < 30; index++) await db.transaction(async tx => { await notifyGitHubCredentialChange(tx, f.connection); });
|
|
expect(leases).toBeLessThanOrEqual(2);
|
|
await vi.advanceTimersByTimeAsync(6_000);
|
|
await vi.waitFor(() => expect(load).toHaveBeenCalledWith(f.company.id));
|
|
expect(load.mock.calls.every(([companyId]) => companyId === f.company.id)).toBe(true);
|
|
expect(sweep).not.toHaveBeenCalled();
|
|
await vi.waitFor(() => expect(leases).toBeGreaterThanOrEqual(2));
|
|
// The first hint can run immediately; changes during it get one spaced pass.
|
|
expect(leases).toBeLessThanOrEqual(3); expect(load.mock.calls.length).toBeLessThanOrEqual(2);
|
|
if (loadTimes.length === 2) expect(loadTimes[1]! - loadTimes[0]!).toBeGreaterThanOrEqual(5_000);
|
|
if (leaseTimes.length === 3) expect(leaseTimes[2]! - leaseTimes[1]!).toBeGreaterThanOrEqual(5_000);
|
|
// A hint arriving during the first SQL pass retains one spaced generation.
|
|
// Start the empty-window assertion after that concrete work has settled.
|
|
await vi.advanceTimersByTimeAsync(31_000); await settled(coordinator);
|
|
expect(leases).toBeLessThanOrEqual(3); expect(load.mock.calls.length).toBeLessThanOrEqual(2);
|
|
if (loadTimes.length === 2) expect(loadTimes[1]! - loadTimes[0]!).toBeGreaterThanOrEqual(5_000);
|
|
const quiet = { leases, loads: load.mock.calls.length };
|
|
await vi.advanceTimersByTimeAsync(60_000);
|
|
expect({ leases, loads: load.mock.calls.length }).toEqual(quiet);
|
|
expect(errors).toEqual([]);
|
|
} finally { await events.stop(); await continuity.stop(); await coordinator.stop(); }
|
|
});
|
|
|
|
it("settles only the company and grant that own saved binding debt", async () => {
|
|
const f = await fixture(["101"]), other = await fixture(["202"]);
|
|
await db.update(connectionGrants).set({ providerTenant: { ...f.grant.providerTenant, github: { ...f.grant.providerTenant!.github!,
|
|
bindingSync: { id: randomUUID(), removeInstallationIds: [] } } } }).where(eq(connectionGrants.id, f.grant.id));
|
|
await f.service.sweepGitHubConnectionContinuity();
|
|
expect(bindings).toHaveLength(1); expect(bindings[0]).toMatchObject({ grantId: f.grant.id, installationId: "101" });
|
|
const [untouched] = await db.select().from(connectionGrants).where(eq(connectionGrants.id, other.grant.id));
|
|
expect(untouched!.providerTenant).toEqual(other.grant.providerTenant);
|
|
});
|
|
|
|
it("retains a real deadline when successful renewal leaves a short-lived credential unchanged", async () => {
|
|
const f = await fixture(["101"]); fakeClock();
|
|
vi.spyOn(globalThis, "fetch").mockImplementation((url, init) => String(url).startsWith("https://api.github.com")
|
|
? realFetch(`${origin}${new URL(String(url)).pathname}${new URL(String(url)).search}`, init) : realFetch(url, init));
|
|
await db.update(connectionGrants).set({ providerTenant: { ...f.grant.providerTenant,
|
|
oauth: { ...f.grant.providerTenant!.oauth, accessTokenExpiresAt: new Date(Date.now() + 30 * 60_000).toISOString() } } }).where(eq(connectionGrants.id, f.grant.id));
|
|
expect(await f.service.sweepGitHubConnectionContinuity()).toMatchObject({ due: 1, refreshed: 1 });
|
|
const [saved] = await db.select().from(connectionGrants).where(eq(connectionGrants.id, f.grant.id));
|
|
const retryAt = Date.parse(saved!.providerTenant!.github!.continuityRetry!.at);
|
|
expect(retryAt).toBeGreaterThanOrEqual(Date.now() + 59 * 60_000);
|
|
const restarted = createGitHubGrantContinuity(db, f.dependencies);
|
|
expect(await restarted.nextGitHubConnectionContinuityAt()).toBe(retryAt);
|
|
const requests = githubRequests; await vi.advanceTimersByTimeAsync(10 * 60_000);
|
|
expect(await restarted.sweepGitHubConnectionContinuity()).toMatchObject({ due: 0 }); expect(githubRequests).toBe(requests);
|
|
});
|
|
|
|
it("outer commit, rollback and first producer during drain retain the deadline and resume obligation without cross-company work", async () => {
|
|
const f = await fixture(["101"]); fakeClock();
|
|
let enabled = false;
|
|
const errors: unknown[] = [];
|
|
const coordinator = createDeliveryWorkCoordinator({ owner: db, canRun: () => enabled, canReconcile: () => true, onError: error => errors.push(error) });
|
|
const worker = registerGitHubConnectionContinuityWork(db, coordinator, f.service, () => enabled);
|
|
try {
|
|
await worker.ready;
|
|
const dueAt = Date.now() + 90 * 60_000;
|
|
let release!: () => void;
|
|
const held = new Promise<void>(resolve => { release = resolve; });
|
|
const transaction = db.transaction(async tx => {
|
|
await tx.transaction(async nested => {
|
|
await notifyGitHubCredentialChange(nested, f.connection);
|
|
await nested.update(connectionGrants).set({ providerTenant: { ...f.grant.providerTenant,
|
|
oauth: { ...f.grant.providerTenant!.oauth, accessTokenExpiresAt: new Date(dueAt + 3600_000).toISOString() } } }).where(eq(connectionGrants.id, f.grant.id));
|
|
});
|
|
await held;
|
|
});
|
|
await new Promise(resolve => setImmediate(resolve));
|
|
expect(bindings).toHaveLength(0); release(); await transaction;
|
|
await vi.advanceTimersByTimeAsync(31_000);
|
|
expect(idleWorkSnapshot().active).toBe(0); expect(bindings).toHaveLength(0);
|
|
enabled = true; await vi.advanceTimersByTimeAsync(31_000);
|
|
await vi.waitFor(() => expect(coordinator.nextWakeAt()).toBe(dueAt));
|
|
await expect(db.transaction(async tx => { await notifyGitHubCredentialChange(tx, f.connection);
|
|
await tx.update(connectionGrants).set({ status: "revoked" }).where(eq(connectionGrants.id, f.grant.id)); throw new Error("rollback"); })).rejects.toThrow("rollback");
|
|
await vi.advanceTimersByTimeAsync(31_000);
|
|
await vi.waitFor(() => expect(coordinator.nextWakeAt()).toBe(dueAt)); expect(errors).toEqual([]);
|
|
} finally { await worker.stop(); await coordinator.stop(); }
|
|
});
|
|
});
|