请用unittest框架帮我写一个自动化测试的登录脚本
时间: 2024-06-10 07:07:31 浏览: 15
好的,我可以帮您编写一个使用unittest框架的自动化测试登录脚本。以下是示例代码:
```python
import unittest
from selenium import webdriver
class LoginTest(unittest.TestCase):
def setUp(self):
self.driver = webdriver.Chrome()
self.driver.maximize_window()
self.driver.implicitly_wait(10)
self.driver.get("https://example.com/login")
def test_login_with_valid_credentials(self):
username_input = self.driver.find_element_by_id("username")
username_input.send_keys("testuser")
password_input = self.driver.find_element_by_id("password")
password_input.send_keys("testpass")
login_button = self.driver.find_element_by_xpath("//button[contains(text(), 'Login')]")
login_button.click()
welcome_message = self.driver.find_element_by_xpath("//h1[contains(text(), 'Welcome')]")
self.assertTrue(welcome_message.is_displayed())
def test_login_with_invalid_credentials(self):
username_input = self.driver.find_element_by_id("username")
username_input.send_keys("invaliduser")
password_input = self.driver.find_element_by_id("password")
password_input.send_keys("invalidpass")
login_button = self.driver.find_element_by_xpath("//button[contains(text(), 'Login')]")
login_button.click()
error_message = self.driver.find_element_by_xpath("//p[contains(text(), 'Incorrect username or password')]")
self.assertTrue(error_message.is_displayed())
def tearDown(self):
self.driver.quit()
if __name__ == '__main__':
unittest.main()
```
这个脚本包含两个测试方法,一个是使用有效凭据登录,另一个是使用无效凭据登录。在每个测试方法中,我们使用Selenium WebDriver在浏览器中执行登录操作,并验证结果是否符合预期。
请注意,这只是一个示例代码,您需要根据实际情况来更改代码,并提供正确的用户名,密码,网站URL以及查找元素的方式等。