feat(integration_test): add remaining templates

Author: cabljac

integration_test/cloudbuild.yaml

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

integration_test/config/suites.yaml

@@ -129,7 +129,9 @@ suites:
     functions:
       - name: schedulerOnSchedule
         trigger: onSchedule
+        decorator: on_schedule
         schedule: "every 1 minutes"
+        collection: schedulerOnScheduleV2Tests
 
   # V2 Task Queue functions
   - name: v2_tasks
@@ -139,7 +141,9 @@ suites:
     functions:
       - name: taskQueueOnEnqueue
         trigger: onTaskDispatched
+        decorator: on_task_dispatched
         queueName: "v2-test-queue"
+        collection: tasksOnTaskDispatchedTests
 
   # V2 Identity triggers
   - name: v2_identity
@@ -149,8 +153,12 @@ suites:
     functions:
       - name: identityBeforeUserCreated
         trigger: beforeUserCreated
+        decorator: before_user_created
+        collection: identityBeforeUserCreatedTests
       - name: identityBeforeUserSignedIn
         trigger: beforeUserSignedIn
+        decorator: before_user_signed_in
+        collection: identityBeforeUserSignedInTests
 
   # V2 Eventarc triggers
   - name: v2_eventarc
@@ -182,6 +190,8 @@ suites:
     functions:
       - name: testLabOnTestMatrixCompleted
         trigger: onTestMatrixCompleted
+        decorator: on_test_matrix_completed
+        collection: testLabOnTestMatrixCompletedTests
 
   # V2 Remote Config triggers
   - name: v2_remoteconfig
@@ -191,3 +201,5 @@ suites:
     functions:
       - name: remoteConfigOnConfigUpdated
         trigger: onConfigUpdated
+        decorator: on_config_updated
+        collection: remoteConfigOnConfigUpdatedTests

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

@@ -0,0 +1,46 @@
+"""
+Identity trigger tests for v2
+Test Run ID: {{testRunId}}
+"""
+
+import os
+from firebase_admin import firestore
+from firebase_functions import identity_fn
+
+REGION = "{{region}}"
+
+{{#each functions}}
+@identity_fn.{{decorator}}(
+    region=REGION,
+    timeout_sec={{timeout}}
+)
+def {{name}}_{{../testRunId}}(event: identity_fn.AuthBlockingEvent) -> None:
+    """
+    Test function: {{name}}
+    Trigger: {{trigger}}
+    """
+    # Extract uid from event data
+    uid = event.data.uid
+
+    # Get project ID from environment
+    project_id = os.environ.get('PROJECT_ID') or os.environ.get('GCLOUD_PROJECT') or "functions-integration-tests-v2"
+
+    # Prepare context data for storage
+    context_data = {
+        "eventId": event.event_id,
+        "eventType": event.event_type,
+        "timestamp": event.timestamp.isoformat() if hasattr(event.timestamp, 'isoformat') else str(event.timestamp),
+        "resource": {
+            "name": f"projects/{project_id}"
+        }
+    }
+
+    # Store context in Firestore for verification
+    db = firestore.client()
+    collection_name = "{{#if collection}}{{collection}}{{else}}{{name}}{{/if}}"
+    db.collection(collection_name).document(uid).set(context_data)
+
+    # Return None (no modifications to user)
+    return None
+
+{{/each}}

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

@@ -0,0 +1,41 @@
+"""
+Remote Config trigger tests for v2
+Test Run ID: {{testRunId}}
+"""
+
+from firebase_admin import firestore
+from firebase_functions import remote_config_fn
+
+REGION = "{{region}}"
+
+{{#each functions}}
+@remote_config_fn.{{decorator}}(
+    region=REGION,
+    timeout_sec={{timeout}}
+)
+def {{name}}_{{../testRunId}}(event: remote_config_fn.CloudEvent[remote_config_fn.ConfigUpdateData]) -> None:
+    """
+    Test function: {{name}}
+    Trigger: {{trigger}}
+    """
+    # Extract test_id from event data description
+    test_id = event.data.description if event.data else None
+
+    if not test_id:
+        print(f"Warning: No testId found in event data description")
+        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,
+    }
+
+    # 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/scheduler_tests.py.hbs

@@ -0,0 +1,34 @@
+"""
+Scheduler trigger tests for v2
+Test Run ID: {{testRunId}}
+"""
+
+from firebase_admin import firestore
+from firebase_functions import scheduler_fn
+
+REGION = "{{region}}"
+
+{{#each functions}}
+@scheduler_fn.{{decorator}}(
+    schedule="{{schedule}}",
+    region=REGION,
+    timeout_sec={{timeout}}
+)
+def {{name}}_{{../testRunId}}(event: scheduler_fn.ScheduledEvent) -> None:
+    """
+    Test function: {{name}}
+    Trigger: {{trigger}}
+    Schedule: {{schedule}}
+    """
+    # Extract test_id from job_name
+    test_id = event.job_name
+    if not test_id:
+        print(f"Warning: No job_name found for scheduled function execution")
+        return
+
+    # Store success flag in Firestore for verification
+    db = firestore.client()
+    collection_name = "{{#if collection}}{{collection}}{{else}}{{name}}{{/if}}"
+    db.collection(collection_name).document(test_id).set({"success": True})
+
+{{/each}}

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

@@ -0,0 +1,44 @@
+"""
+Task Queue trigger tests for v2
+Test Run ID: {{testRunId}}
+"""
+
+import json
+import uuid
+from firebase_admin import firestore
+from firebase_functions import tasks_fn
+from firebase_functions.https_fn import CallableRequest
+
+REGION = "{{region}}"
+
+{{#each functions}}
+@tasks_fn.{{decorator}}(
+    region=REGION,
+    timeout_sec={{timeout}}
+)
+def {{name}}_{{../testRunId}}(request: CallableRequest) -> None:
+    """
+    Test function: {{name}}
+    Trigger: {{trigger}}
+    """
+    # Extract test_id from request data
+    data = request.data
+    if not data or not isinstance(data, dict) or 'testId' not in data:
+        print(f"Warning: Invalid data structure for tasks onTaskDispatched")
+        return
+
+    test_id = data.get('testId')
+
+    # Prepare context data for storage
+    # Generate a unique ID since CallableRequest doesn't have one
+    context_data = {
+        "id": str(uuid.uuid4()),
+        "data": json.dumps(data) if 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/testlab_tests.py.hbs

@@ -0,0 +1,42 @@
+"""
+Test Lab trigger tests for v2
+Test Run ID: {{testRunId}}
+"""
+
+from firebase_admin import firestore
+from firebase_functions import test_lab_fn
+
+REGION = "{{region}}"
+
+{{#each functions}}
+@test_lab_fn.{{decorator}}(
+    region=REGION,
+    timeout_sec={{timeout}}
+)
+def {{name}}_{{../testRunId}}(event: test_lab_fn.CloudEvent[test_lab_fn.TestMatrixCompletedData]) -> None:
+    """
+    Test function: {{name}}
+    Trigger: {{trigger}}
+    """
+    # Extract test_id from client_info details
+    test_id = event.data.client_info.details.get('testId') if event.data.client_info else None
+
+    if not test_id:
+        print(f"Warning: No testId found in client_info details")
+        return
+
+    # Prepare context data for storage
+    context_data = {
+        "testId": test_id,
+        "id": event.id,
+        "time": event.time.isoformat() if hasattr(event.time, 'isoformat') else str(event.time),
+        "type": event.type,
+        "state": str(event.data.state),
+    }
+
+    # 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}}