SurfSense/surfsense_web/tests/fixtures/connectors/native-gmail.fixture.ts

import {
	type ConnectorRow,
	deleteConnector,
	runNativeGoogleGmailOAuth,
} from "../../helpers/api/connectors";
import { workspaceFixtures } from "../workspace.fixture";

export type NativeGmailFixtures = {
	/**
	 * A pre-connected native Google Gmail connector inside the fixture's
	 * `workspace`. OAuth uses E2E native Google fakes and is cleaned up
	 * automatically after the test.
	 */
	nativeGmailConnector: ConnectorRow;
};

export const nativeGmailFixtures = workspaceFixtures.extend<NativeGmailFixtures>({
	nativeGmailConnector: async ({ request, apiToken, workspace }, use) => {
		const { connector } = await runNativeGoogleGmailOAuth(request, apiToken, workspace.id);
		if (!connector) {
			throw new Error(
				"nativeGmailConnector fixture: OAuth completed but no GOOGLE_GMAIL_CONNECTOR was created. " +
					"Check the backend log — the native Google fake likely rejected an unmodelled call."
			);
		}
		try {
			await use(connector);
		} finally {
			await deleteConnector(request, apiToken, connector.id);
		}
	},
});