24
24
import pytest
25
25
from selenium import webdriver
26
26
from selenium .webdriver .chrome .webdriver import WebDriver
27
+ from selenium .webdriver .common .by import By
27
28
28
29
from annotate_assessment import annotate_assessment
29
30
from create_assessment import create_assessment
@@ -102,14 +103,14 @@ def get_token(recaptcha_site_key: str, browser: WebDriver) -> typing.Tuple:
102
103
browser .get (url_for ("assess" , site_key = recaptcha_site_key , _external = True ))
103
104
time .sleep (5 )
104
105
105
- browser .find_element_by_id ( "username" ).send_keys ("username" )
106
- browser .find_element_by_id ( "password" ).send_keys ("password" )
107
- browser .find_element_by_id ( "recaptchabutton" ).click ()
106
+ browser .find_element ( By . ID , "username" ).send_keys ("username" )
107
+ browser .find_element ( By . ID , "password" ).send_keys ("password" )
108
+ browser .find_element ( By . ID , "recaptchabutton" ).click ()
108
109
109
110
# Timeout of 5 seconds
110
111
time .sleep (5 )
111
112
112
- element = browser .find_element_by_css_selector ( "#assessment" )
113
+ element = browser .find_element ( By . CSS_SELECTOR , "#assessment" )
113
114
token = element .get_attribute ("data-token" )
114
115
action = element .get_attribute ("data-action" )
115
116
return token , action
@@ -125,4 +126,4 @@ def assess_token(recaptcha_site_key: str, token: str, action: str) -> Assessment
125
126
126
127
127
128
def set_score (browser : WebDriver , score : str ) -> None :
128
- browser .find_element_by_css_selector ( "#assessment" ).send_keys (score )
129
+ browser .find_element ( By . CSS_SELECTOR , "#assessment" ).send_keys (score )
0 commit comments