python+selenium+unittest 搭建web自动化测试框架(二)

基于po模型的business层文件 register_business.py

from handle.register_handle import RegisterHandle

class RegisterBusiness:
    def __init__(self,driver):
        self.register_h = RegisterHandle(driver)

    def user_base(self,email,username,password,code):
        self.register_h.send_user_email(email)
        self.register_h.send_user_name(username)
        self.register_h.send_user_password(password)
        self.register_h.send_user_code(code)
        self.register_h.click_button()

    def register_function(self,email,name,password,code,assertText):
        self.user_base(email,name,password,code)
        if self.register_h.get_text_info(assertText) == None:
            return True
        else:
            return False