scheduler & tasks tests

Author: Ehesp

integration_test/functions/package-lock.json

@@ -15,6 +15,8 @@
   "main": "lib/index.js",
   "dependencies": {
     "@google-cloud/pubsub": "^5.2.0",
+    "@google-cloud/scheduler": "^5.3.1",
+    "@google-cloud/tasks": "^6.2.1",
     "firebase": "^12.6.0",
     "firebase-admin": "^12.6.0",
     "firebase-functions": "file:firebase-functions-local.tgz",

integration_test/functions/package.json

@@ -8,9 +8,14 @@ export function expectCloudEvent(data: any) {
   expect(data.source).toBeDefined();
   assertType<string>(data.source);
   expect(data.source.length).toBeGreaterThan(0);
-  expect(data.subject).toBeDefined();
-  assertType<string>(data.subject);
-  expect(data.subject.length).toBeGreaterThan(0);
+
+  // Subject is optional (e.g. pubsub)
+  if ("subject" in data) {
+    expect(data.subject).toBeDefined();
+    assertType<string>(data.subject);
+    expect(data.subject.length).toBeGreaterThan(0);
+  }
+
   expect(data.type).toBeDefined();
   assertType<string>(data.type);
   expect(data.type.length).toBeGreaterThan(0);

…on_test/functions/src/assertions/auth.ts …est/functions/src/assertions/identity.tsintegration_test/functions/src/assertions/auth.ts renamed to integration_test/functions/src/assertions/identity.ts integration_test/functions/src/assertions/auth.ts renamed to integration_test/functions/src/assertions/identity.ts

@@ -1,16 +1,9 @@
-import { initializeApp } from "firebase/app";
-import { getAuth } from "firebase/auth";
-import { getFunctions } from "firebase/functions";
-
-export const app = initializeApp({
+export const config = {
   apiKey: "AIzaSyBBt77mpu6TV0IA2tcNSyf4OltsVu_Z1Zw",
   authDomain: "cf3-integration-tests-v2-qa.firebaseapp.com",
   databaseURL: "https://cf3-integration-tests-v2-qa-default-rtdb.firebaseio.com",
   projectId: "cf3-integration-tests-v2-qa",
   storageBucket: "cf3-integration-tests-v2-qa.firebasestorage.app",
   messagingSenderId: "576826020291",
-  appId: "1:576826020291:web:488d568c5d4109df12ed76",
-});
-
-export const auth = getAuth(app);
-export const functions = getFunctions(app);
+  appId: "1:576826020291:web:488d568c5d4109df12ed76"
+};

integration_test/functions/src/assertions/index.ts

@@ -0,0 +1,8 @@
+import { initializeApp } from "firebase/app";
+import { getAuth } from "firebase/auth";
+import { getFunctions } from "firebase/functions";
+import { config } from "./config";
+
+export const app = initializeApp(config);
+export const auth = getAuth(app);
+export const functions = getFunctions(app);

integration_test/functions/src/client.ts integration_test/functions/src/config.tsintegration_test/functions/src/client.ts renamed to integration_test/functions/src/config.ts

@@ -0,0 +1,40 @@
+import admin from "firebase-admin";
+import { GoogleAuth } from "google-auth-library";
+import { applicationDefault } from "firebase-admin/app";
+import { getFunctions } from "firebase-admin/functions";
+import { getDatabase } from "firebase-admin/database";
+import { getAuth } from "firebase-admin/auth";
+import { getRemoteConfig } from "firebase-admin/remote-config";
+import { getFirestore } from "firebase-admin/firestore";
+import { config } from "./config";
+
+export const app = admin.initializeApp({
+  credential: applicationDefault(),
+  projectId: config.projectId,
+  databaseURL: config.databaseURL,
+});
+
+export const firestore = getFirestore(app);
+firestore.settings({ ignoreUndefinedProperties: true });
+export const database = getDatabase(app);
+export const auth = getAuth(app);
+export const remoteConfig = getRemoteConfig(app);
+export const functions = getFunctions(app);
+
+// See https://github.com/firebase/functions-samples/blob/a6ae4cbd3cf2fff3e2b97538081140ad9befd5d8/Node/taskqueues-backup-images/functions/index.js#L111-L128
+export async function getFunctionUrl(name: string) {
+  const auth = new GoogleAuth({
+    projectId: config.projectId,
+  });
+
+  const url = `https://cloudfunctions.googleapis.com/v2beta/projects/${config.projectId}/locations/us-central1/functions/${name}`;
+  const client = await auth.getClient();
+  const res: any = await client.request({ url });
+  const uri = res.data?.serviceConfig?.uri;
+
+  if (!uri) {
+    throw new Error(`Function ${name} not found`);
+  }
+
+  return uri;
+}

integration_test/functions/src/firebase.client.ts

@@ -1,5 +1,9 @@
-export * from "./firestore.v2";
-export * from "./database.v2";
-export * from "./eventarc.v2";
-export * from "./auth.v2";
-export * from "./https.v2";
+export * from "./v2/database.v2";
+export * from "./v2/eventarc.v2";
+export * from "./v2/firestore.v2";
+export * from "./v2/https.v2";
+export * from "./v2/identity.v2";
+export * from "./v2/pubsub.v2";
+export * from "./v2/remoteConfig.v2";
+export * from "./v2/scheduler.v2";
+export * from "./v2/tasks.v2";

integration_test/functions/src/firebase.server.ts

@@ -1,16 +1,4 @@
-import admin from "firebase-admin";
-import { GeoPoint } from "firebase-admin/firestore";
-import { logger } from "firebase-functions";
-
-const adminApp = admin.initializeApp({
-  projectId: "cf3-integration-tests-v2-qa",
-  databaseURL: "https://cf3-integration-tests-v2-qa-default-rtdb.firebaseio.com",
-  
-});
-export const firestore = adminApp.firestore();
-firestore.settings({ ignoreUndefinedProperties: true });
-export const database = adminApp.database();
-export const auth = adminApp.auth();
+import { firestore } from "./firebase.server";
 
 export const RUN_ID = String(process.env.RUN_ID);
 
@@ -60,98 +48,3 @@ export function waitForEvent<T = unknown>(
       .catch(reject);
   });
 }
-
-export function serializeData(data: any): any {
-  if (data === null || data === undefined) {
-    return null;
-  }
-
-  if (typeof data === "function") {
-    logger.debug("serializeData function", data);
-    const result = serializeData(data());
-    logger.debug("serializeData function result", result);
-    return result;
-  }
-
-  // Handle Date objects
-  if (data instanceof Date) {
-    return data.toISOString();
-  }
-
-  if (data instanceof GeoPoint) {
-    return {
-      _type: "geopoint",
-      latitude: data.latitude,
-      longitude: data.longitude,
-    };
-  }
-
-  // Handle Firestore DocumentReference (check for path, id, and firestore properties)
-  if (
-    data &&
-    typeof data === "object" &&
-    "path" in data &&
-    "id" in data &&
-    "firestore" in data &&
-    typeof (data as any).path === "string"
-  ) {
-    const ref = data as { path: string; id: string };
-    // Only serialize if path is non-empty to avoid the error
-    if (ref.path) {
-      return {
-        _type: "reference",
-        path: ref.path,
-        id: ref.id,
-      };
-    } else {
-      // If path is empty, return a safe representation
-      logger.warn("DocumentReference with empty path detected", { id: ref.id });
-      return {
-        _type: "reference",
-        path: "",
-        id: ref.id || "",
-      };
-    }
-  }
-
-  // Handle Firestore Timestamp (check for toDate method and seconds property)
-  if (
-    data &&
-    typeof data === "object" &&
-    "toDate" in data &&
-    typeof (data as any).toDate === "function"
-  ) {
-    const timestamp = data as { toDate: () => Date; seconds?: number; nanoseconds?: number };
-    return {
-      _type: "timestamp",
-      seconds: timestamp.seconds,
-      nanoseconds: timestamp.nanoseconds,
-      iso: timestamp.toDate().toISOString(),
-    };
-  }
-
-  // Handle Firestore GeoPoint (check for latitude and longitude properties)
-  if (data && typeof data === "object" && "latitude" in data && "longitude" in data) {
-    const geoPoint = data as { latitude: number; longitude: number };
-    return {
-      _type: "geopoint",
-      latitude: geoPoint.latitude,
-      longitude: geoPoint.longitude,
-    };
-  }
-
-  // Handle arrays (must check before plain objects since arrays are objects)
-  if (Array.isArray(data)) {
-    return data.map(serializeData);
-  }
-
-  if (typeof data === "object") {
-    const result: Record<string, unknown> = {};
-    for (const [key, value] of Object.entries(data)) {
-      result[key] = serializeData(value);
-    }
-    return result;
-  }
-
-  return data;
-}