def test_register_code_right(self): username = util.get_random_str() email = username + "@qq.com" pwd = "123456" confirm_pwd = "123456" captcha = "" expected = "注册成功,点击确认进行登录。" self.driver.find_element_by_name("username").clear() self.driver.find_element_by_name("username").send_keys(username) self.driver.find_element_by_name("email").clear() self.driver.find_element_by_name("email").send_keys(email) self.driver.find_element_by_name("pwd").clear() self.driver.find_element_by_name("pwd").send_keys(pwd) self.driver.find_element_by_name("confirmPwd").clear() self.driver.find_element_by_name("confirmPwd").send_keys(confirm_pwd) captcha = util.get_code(self.driver, "captcha") self.driver.find_element_by_name("captcha").clear() self.driver.find_element_by_name("captcha").send_keys(captcha) self.driver.find_element_by_class_name("btn").click() WebDriverWait(self.driver, 5).until(EC.alert_is_present()) alert = self.driver.switch_to.alert self.assertEqual(alert.text, expected) alert.accept() sleep(2)
class TestUserRegister(object): def setup_class(self): self.driver = webdriver.Chrome() self.driver.get("http://*****:*****@qq.com', '123456', '123456', '666', '验证码不正确'), (util.get_random_str(), util.get_random_str() + '@qq.com', '123456', '123456', '', '验证码不正确'), ] @pytest.mark.parametrize( 'username, email, pwd, confirm_pwd, captcha, expected', login_data) def test_register_code_error(self, username, email, pwd, confirm_pwd, captcha, expected): self.driver.find_element_by_name("username").clear() self.driver.find_element_by_name("username").send_keys(username) self.driver.find_element_by_name("email").clear() self.driver.find_element_by_name("email").send_keys(email) self.driver.find_element_by_name("pwd").clear() self.driver.find_element_by_name("pwd").send_keys(pwd) self.driver.find_element_by_name("confirmPwd").clear() self.driver.find_element_by_name("confirmPwd").send_keys(confirm_pwd) if captcha != '666': captcha = util.get_code(self.driver, "captcha") self.driver.find_element_by_name("captcha").clear() self.driver.find_element_by_name("captcha").send_keys(captcha) self.driver.find_element_by_class_name("btn").click() WebDriverWait(self.driver, 5).until(EC.alert_is_present()) alert = self.driver.switch_to.alert assert alert.text == expected alert.accept() sleep(2)