feat(integration_test): add eventarc and pubsub suites

Author: cabljac

integration_test/cloudbuild.yaml

@@ -130,7 +130,7 @@ steps:
 
         node scripts/run-tests.js \
           --sequential \
-          v2_firestore v2_database v2_storage \
+          v2_firestore v2_database v2_storage v2_pubsub v2_eventarc \
           --use-published-sdk=file:$$WHEEL_FILE
 
 # Artifacts to store

integration_test/config/suites.yaml

@@ -85,7 +85,10 @@ suites:
     functions:
       - name: pubsubOnMessagePublished
         trigger: onMessagePublished
-        topic: "firebase-functions-pubsub-v2"
+        decorator: on_message_published
+        eventType: messagePublished
+        topic: custom_message_tests
+        collection: pubsubOnMessagePublishedTests
 
   # V2 Storage triggers
   - name: v2_storage
@@ -157,7 +160,9 @@ suites:
     functions:
       - name: eventarcOnCustomEventPublished
         trigger: onCustomEventPublished
-        eventType: "test.event.v1"
+        decorator: on_custom_event_published
+        eventType: achieved-leaderboard
+        collection: eventarcOnCustomEventPublishedTests
 
   # V2 Alerts triggers
   - name: v2_alerts

integration_test/templates/functions/src/v2/eventarc_tests.py.hbs

@@ -0,0 +1,46 @@
+"""
+Eventarc trigger tests for v2
+Test Run ID: {{testRunId}}
+"""
+
+import json
+from firebase_admin import firestore
+from firebase_functions import eventarc_fn
+
+REGION = "{{region}}"
+
+{{#each functions}}
+@eventarc_fn.{{decorator}}(
+    event_type="{{eventType}}",
+    region=REGION,
+    timeout_sec={{timeout}}
+)
+def {{name}}_{{../testRunId}}(event: eventarc_fn.CloudEvent) -> None:
+    """
+    Test function: {{name}}
+    Trigger: {{trigger}}
+    Event Type: {{eventType}}
+    """
+    # Extract test_id from event data
+    test_id = event.data.get('testId') if event.data else None
+
+    if not test_id:
+        print(f"Warning: No testId found in event data")
+        return
+
+    # Prepare context data for storage
+    context_data = {
+        "id": event.id,
+        "time": event.time.isoformat() if hasattr(event.time, 'isoformat') else str(event.time),
+        "type": event.type,
+        "source": event.source,
+        # Stringify the data dict for storage (matching JS behavior)
+        "data": json.dumps(event.data) if event.data else "{}"
+    }
+
+    # Store context in Firestore for verification
+    db = firestore.client()
+    collection_name = "{{#if collection}}{{collection}}{{else}}{{name}}{{/if}}"
+    db.collection(collection_name).document(test_id).set(context_data)
+
+{{/each}}

integration_test/templates/functions/src/v2/pubsub_tests.py.hbs

@@ -0,0 +1,54 @@
+"""
+Pub/Sub trigger tests for v2
+Test Run ID: {{testRunId}}
+"""
+
+import json
+from firebase_admin import firestore
+from firebase_functions import pubsub_fn
+
+REGION = "{{region}}"
+
+{{#each functions}}
+@pubsub_fn.{{decorator}}(
+    topic="{{topic}}",
+    region=REGION,
+    timeout_sec={{timeout}}
+)
+def {{name}}_{{../testRunId}}(event: pubsub_fn.CloudEvent[pubsub_fn.MessagePublishedData[dict]]) -> None:
+    """
+    Test function: {{name}}
+    Trigger: {{trigger}}
+    Topic: {{topic}}
+    """
+    # Extract test_id from message JSON data
+    message = event.data.message
+    message_json = message.json
+    test_id = message_json.get('testId') if message_json else None
+
+    if not test_id:
+        print(f"Warning: No testId found in message")
+        return
+
+    # Prepare context data for storage
+    context_data = {
+        "id": event.id,
+        "time": event.time,
+        "type": f"google.cloud.pubsub.topic.v1.{{eventType}}",
+        "source": event.source,
+        # Stringify the message object for storage (matching JS behavior)
+        "message": json.dumps({
+            "data": message.data,
+            "messageId": message.message_id,
+            "publishTime": message.publish_time,
+            "attributes": message.attributes,
+            "orderingKey": message.ordering_key,
+        })
+    }
+
+    # Store context in Firestore for verification
+    db = firestore.client()
+    collection_name = "{{#if collection}}{{collection}}{{else}}{{name}}{{/if}}"
+    db.collection(collection_name).document(test_id).set(context_data)
+
+{{/each}}