SurfSense/surfsense_web/tests/fixtures/connectors/composio-drive.fixture.ts

import { type ConnectorRow, deleteConnector, runComposioOAuth } from "../../helpers/api/connectors";
import { searchSpaceFixtures } from "../search-space.fixture";

export type ComposioDriveFixtures = {
	/**
	 * A pre-connected Composio Google Drive connector inside the
	 * fixture's `searchSpace`. OAuth happens against the strict fake
	 * (no real network). Cleaned up automatically after the test.
	 */
	composioDriveConnector: ConnectorRow;
};

export const composioDriveFixtures = searchSpaceFixtures.extend<ComposioDriveFixtures>({
	composioDriveConnector: async ({ request, apiToken, searchSpace }, use) => {
		const { connector } = await runComposioOAuth(request, apiToken, searchSpace.id, "googledrive");
		if (!connector) {
			throw new Error(
				"composioDriveConnector fixture: OAuth completed but no connector was created. " +
					"Check the backend log — the strict Composio fake likely rejected an unmodelled call."
			);
		}
		try {
			await use(connector);
		} finally {
			await deleteConnector(request, apiToken, connector.id);
		}
	},
});