Set the assessment.event.expected_action when sending a CreateAssessment request. (GoogleCloudPlatform#11170)

Cory-Kramer · web-flow · commit 7d1c21b10881 · 2024-02-02T15:42:29.000-05:00
diff --git a/recaptcha_enterprise/demosite/app/backend/create_recaptcha_assessment.py b/recaptcha_enterprise/demosite/app/backend/create_recaptcha_assessment.py
@@ -17,13 +17,14 @@
 
 
 def create_assessment(
-    project_id: str, recaptcha_site_key: str, token: str
+    project_id: str, recaptcha_site_key: str, token: str, expected_action: str
 ) -> Assessment:
     """Create an assessment to analyze the risk of a UI action.
     Args:
         project_id: Google Cloud Project ID
         recaptcha_site_key: Site key obtained by registering a domain/app to use recaptcha services.
         token: The token obtained from the client on passing the recaptchaSiteKey.
+        expected_action: The expected action for this type of event.
     Returns: Assessment response.
     """
 
@@ -34,6 +35,7 @@ def create_assessment(
     event = recaptchaenterprise_v1.Event()
     event.site_key = recaptcha_site_key
     event.token = token
+    event.expected_action = expected_action
 
     assessment = recaptchaenterprise_v1.Assessment()
     assessment.event = event
diff --git a/recaptcha_enterprise/demosite/app/urls.py b/recaptcha_enterprise/demosite/app/urls.py
@@ -68,6 +68,7 @@ def on_homepage_load() -> Response:
             context.get("project_id"),
             context.get("site_key"),
             json_data["token"],
+            recaptcha_action
         )
 
         # Check if the token is valid, score is above threshold score and the action equals expected.
@@ -114,6 +115,7 @@ def on_signup() -> Response:
             context.get("project_id"),
             context.get("site_key"),
             json_data["token"],
+            recaptcha_action
         )
 
         # Check if the token is valid, score is above threshold score and the action equals expected.
@@ -162,6 +164,7 @@ def on_login() -> Response:
             context.get("project_id"),
             context.get("site_key"),
             json_data["token"],
+            recaptcha_action
         )
 
         # Check if the token is valid, score is above threshold score and the action equals expected.
@@ -210,6 +213,7 @@ def on_store_checkout() -> Response:
             context.get("project_id"),
             context.get("site_key"),
             json_data["token"],
+            recaptcha_action
         )
 
         # Check if the token is valid, score is above threshold score and the action equals expected.
@@ -257,6 +261,7 @@ def on_comment_submit() -> Response:
             context.get("project_id"),
             context.get("site_key"),
             json_data["token"],
+            recaptcha_action
         )
 
         # Check if the token is valid, score is above threshold score and the action equals expected.
