那如果在上面的代码中,添加记录登录不上就记录错误的代码?
时间: 2024-03-20 07:45:17 浏览: 15
你可以在 `test_excel01` 方法中添加一个 try-except 块来捕获登录失败的异常,并在捕获到异常时记录错误信息。例如:
```python
@ddt
class ExcelText(unittest.TestCase):
@data(*read_excel()) #读取xlsx
@unpack
def test_excel01(self, flag, username, password):
driver = self.driver
sleep(2)
driver.find_element(By.XPATH,'//*[@id="username"]').send_keys(username)
driver.find_element(By.XPATH,'//*[@id="password"]').send_keys(password)
try:
# 尝试登录
login_button = driver.find_element(By.XPATH,'//*[@id="loginForm"]/table/tbody/tr[3]/td/a/div/div/div')
login_button.click()
# 如果登录失败,则记录错误信息
if 'login' in driver.current_url:
self.fail(f'Login failed for user {username}')
except Exception as e:
# 捕获登录失败的异常,并记录错误信息
self.fail(f'Login failed for user {username}: {e}')
```
在这个例子中,我们使用了 `try-except` 块来尝试登录,如果登录失败,则使用 `self.fail()` 函数记录错误信息。如果登录成功,则测试用例继续执行,不会触发异常。这样就可以在测试过程中记录登录失败的情况,帮助我们更好地排查问题。