如何确定我的Selenium脚本是否能够登录？

Question

如何知道这个selenium脚本是否能够登录？目前我正在使用is_displayed()查看登录表单是否仍在页面上。还有更好的方法吗？

driver.find_element_by_id("uname").send_keys("admin")
driver.find_element_by_id("pwd").send_keys("admin")
driver.find_element_by_xpath('//*[@id="login_form"]/div/button').click()

# below xpath is code for invalid user id or password message
if driver.find_element_by_xpath('//*[@id="login_form"]/div[1]').is_displayed():
    output=open("OutputAdmin","w+")
    output.write("Login unsuccessful")
    output.close()
else:
    output = open("OutputAdmin", "w+")
    output.write("Login successful")
    output.close()
driver.quit()

现在如果用户ID或密码不正确，它工作正常，但如果它们是正确的，它不会执行其他代码。

Answer 1

try:
    LoginFailMessage = driver.find_element_by_xpath('//*[@id="login_form"]/div[1]')
    if LoginFailMessage.is_displayed():
        output=open("OutputAdmin","w+")
        output.write("Login unsuccessful")
        output.close()
except:
    output = open("OutputAdmin", "w+")
    output.write("Login successful")
    output.close()

我已经解决了这个问题。谢谢那些试图解决它的人。