class RegisterHandle:
def __init__(self, driver):
self.driver = driver
self.register_page = RegisterPage(driver)
# 发送username的值
def send_username_element(self, value):
self.register_page.get_username_element().send_keys(value)
# 发送password的值
def send_password_element(self, value):
self.register_page.get_password_element().send_keys(value)
# 发送password2的值
def send_password2_element(self, value):
self.register_page.get_password2_element().send_keys(value)
# 发送email的值
def send_email_element(self, value):
self.register_page.get_email_element().send_keys(value)
# 发送answer的值
def send_answer_element(self, value):
self.register_page.get_answer_element().send_keys(value)
# 裁剪后的验证码图片元素
def crop_vcode_img_element(self, file_name):
element = self.register_page.get_vcode_img_element()
self.driver.save_screenshot(file_name)
left = element.location['x']
top = element.location['y']
right = left + element.size['width']
bottom = top + element.size['height']
im = Image.open(file_name)
code_img = im.crop((left, top, right, bottom))
code_img.save(file_name)
# 发送code值
def send_vcode_element(self, value):
self.register_page.get_vcode_element().send_keys(value)
# 获取登录按钮元素
def click_login_btn_element(self):
self.register_page.get_login_btn_element().click()
# 对比报错js弹框内容，并关闭弹框
def compare_alert_text(self, text):
result = False
try:
alert_element = self.register_page.get_alert_element()
result = alert_element.text == text
alert_element.accept()
except:
pass
finally:
return result
class RegisterHandle(object):
def __init__(self, driver):
self.driver = driver
self.register_p = RegisterPage(self.driver)
#输入邮箱
def send_user_email(self, user_email):
self.register_p.get_email_element().send_keys(
user_email) # 传的值取得是谁？配置文件里的值吗
#输入用户名
def send_username(self, user_name):
self.register_p.get_user_name_element().send_keys(user_name)
#输入密码
def send_password1(self, password1):
self.register_p.get_password1_element().send_keys(password1)
#确认密码
def send_password2(self, password2):
self.register_p.get_password2_element().send_keys(password2)
#获取邮箱错误信息 的显示文字
def get_error_info(self, info, user_info): #添加容错处理
try:
if info == 'email_error_info':
text = self.register_p.get_email_error_info_element(
).get_attribute('value') # 没有value属性的可以用text方法
elif info == 'password_error_info':
text = self.register_p.get_password_error_info_element(
).get_attribute('value')
elif info == 'repeat_error_info':
text = self.register_p.get_repaet_error_info_element(
).get_attribute('value')
except:
text = None
return text
#点击确认框 checkbox
def click_check_box(self):
self.register_p.get_checkbox_element().click()
self.register_p.get_agreebutton_element().click()
#点击注册按钮
def click_register_button(self):
self.register_p.get_register_element().click()
#获取注册按钮的文字
def get_register_text(self):
return self.register_p.get_register_element().text
