diff --git a/recaptcha_enterprise/snippets/create_assessment.py b/recaptcha_enterprise/snippets/create_assessment.py index b04e07cd6ac..9c0aea12469 100644 --- a/recaptcha_enterprise/snippets/create_assessment.py +++ b/recaptcha_enterprise/snippets/create_assessment.py @@ -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: diff --git a/recaptcha_enterprise/snippets/requirements-test.txt b/recaptcha_enterprise/snippets/requirements-test.txt index 56fd69c7da7..9db643e25dc 100644 --- a/recaptcha_enterprise/snippets/requirements-test.txt +++ b/recaptcha_enterprise/snippets/requirements-test.txt @@ -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 \ No newline at end of file +webdriver-manager==4.0.2 \ No newline at end of file diff --git a/recaptcha_enterprise/snippets/test_create_assessment.py b/recaptcha_enterprise/snippets/test_create_assessment.py index b7257534eac..195e6d8672c 100644 --- a/recaptcha_enterprise/snippets/test_create_assessment.py +++ b/recaptcha_enterprise/snippets/test_create_assessment.py @@ -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,7 +159,14 @@ 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, @@ -160,7 +174,7 @@ def assess_token(recaptcha_site_key: str, token: str, action: str, user_ip_addre recaptcha_action=action, user_ip_address=user_ip_address, user_agent=user_agent, - ja3=ja3 + ja3=ja3, )