用python写5个自动化测试的代码
时间: 2024-04-02 14:34:13 浏览: 10
好的,以下是5个用Python编写的自动化测试代码示例:
1. 测试网站是否正常运行
```python
import requests
def test_website_status():
response = requests.get('http://www.example.com')
assert response.status_code == 200
```
2. 测试登录功能
```python
from selenium import webdriver
def test_login():
driver = webdriver.Chrome()
driver.get('http://www.example.com/login')
username = driver.find_element_by_name('username')
password = driver.find_element_by_name('password')
submit_button = driver.find_element_by_id('submit')
username.send_keys('myusername')
password.send_keys('mypassword')
submit_button.click()
assert driver.current_url == 'http://www.example.com/home'
driver.quit()
```
3. 测试API功能
```python
import requests
def test_api():
response = requests.get('http://api.example.com/users')
assert len(response.json()) > 0
```
4. 测试数据库连接
```python
import psycopg2
def test_database_connection():
conn = psycopg2.connect(database='mydatabase', user='myuser', password='mypassword', host='localhost', port='5432')
cur = conn.cursor()
cur.execute('SELECT COUNT(*) FROM users')
result = cur.fetchone()
assert result[0] > 0
cur.close()
conn.close()
```
5. 测试邮件发送
```python
import smtplib
from email.mime.text import MIMEText
def test_email():
sender = 'myemail@example.com'
receiver = 'youremail@example.com'
message = MIMEText('This is a test email.')
message['Subject'] = 'Test Email'
message['From'] = sender
message['To'] = receiver
smtp_server = smtplib.SMTP('smtp.example.com', 587)
smtp_server.starttls()
smtp_server.login(sender, 'mypassword')
smtp_server.sendmail(sender, receiver, message.as_string())
smtp_server.quit()
```
这些示例代码可以帮助您理解如何使用Python编写自动化测试用例。当然,具体的测试用例还要根据不同的需求来编写。