GoogleCloudPlatform · msampathkumar · Oct 4, 2024 · Oct 1, 2024 · Oct 1, 2024
@@ -22,7 +22,13 @@


 def create_assessment(
-    project_id: str, recaptcha_site_key: str, token: str, recaptcha_action: str, user_ip_address: str, user_agent: str, ja3: str
+    project_id: str,
+    recaptcha_site_key: str,
+    token: str,
+    recaptcha_action: str,
+    user_ip_address: str,
+    user_agent: str,
+    ja3: str,
 ) -> Assessment:
    """Create an assessment to analyze the risk of a UI action.
    Args:

@@ -3,4 +3,4 @@ Flask==2.2.2
 Werkzeug==2.3.7
 pytest==8.2.0
 pytest-flask==1.2.0
-webdriver-manager==4.0.1
+webdriver-manager==4.0.2
@@ -93,7 +93,14 @@ def test_assessment(
    # Get token.
    token, action = get_token(recaptcha_site_key, browser)
    # Create assessment.
-    assessment_response = assess_token(recaptcha_site_key, token=token, action=action, user_ip_address="", user_agent="", ja3="")
+    assessment_response = assess_token(
+        recaptcha_site_key,
+        token=token,
+        action=action,
+        user_ip_address="",
+        user_agent="",
+        ja3="",
+    )
    score = str(assessment_response.risk_analysis.score)
    client = recaptchaenterprise_v1.RecaptchaEnterpriseServiceClient()
    # Parse the assessment_response.name which is of the format:
@@ -152,15 +159,22 @@ def get_token(recaptcha_site_key: str, browser: WebDriver) -> tuple:
    return token, action


-def assess_token(recaptcha_site_key: str, token: str, action: str, user_ip_address: str, user_agent: str, ja3: str) -> Assessment:
+def assess_token(
+    recaptcha_site_key: str,
+    token: str,
+    action: str,
+    user_ip_address: str,
+    user_agent: str,
+    ja3: str,
+) -> Assessment:
    return create_assessment(
        project_id=GOOGLE_CLOUD_PROJECT,
        recaptcha_site_key=recaptcha_site_key,
        token=token,
        recaptcha_action=action,
        user_ip_address=user_ip_address,
        user_agent=user_agent,
-        ja3=ja3
+        ja3=ja3,
    )