9.3 使用pytest进行测试

`pytest`是一个流行的Python测试框架，提供了丰富的插件和功能，支持参数化测试、测试报告生成等。

##### 1. **安装`pytest`**

bash复制

```bash
pip install pytest
```

##### 2. **编写测试用例**

测试用例通常放在`tests/`目录下，文件名以`test_`开头。

**示例：登录测试用例**

Python复制

```python
# tests/test_login.py
import pytest
from pages.login_page import LoginPage

@pytest.fixture
def browser():
    driver = webdriver.Chrome()
    yield driver
    driver.quit()

def test_login_success(browser):
    login_page = LoginPage(browser)
    login_page.open()
    login_page.login("valid_user", "valid_password")
    assert "Dashboard" in browser.title

def test_login_failure(browser):
    login_page = LoginPage(browser)
    login_page.open()
    login_page.login("invalid_user", "invalid_password")
    assert "Login failed" in browser.page_source
```

##### 3. **运行测试**

在项目根目录下运行以下命令：

bash复制

```bash
pytest
```

##### 4. **生成测试报告**

可以使用`pytest-html`插件生成HTML格式的测试报告。

bash复制

```bash
pip install pytest-html
pytest --html=reports/report.html
```

------

### 综合示例

以下是一个完整的测试框架示例，展示如何使用页面对象模式和`pytest`进行测试。

**目录结构：**

复制

```
test_project/
│
├── tests/
│   └── test_login.py
│
├── pages/
│   └── login_page.py
│
├── utils/
│   └── browser.py
│
├── config/
│   └── settings.ini
│
├── reports/
│   └── report.html
│
├── logs/
│   └── test.log
│
├── requirements.txt
└── run_tests.py
```

**页面对象：**

Python复制

```python
# pages/login_page.py
from selenium.webdriver.common.by import By

class LoginPage:
    def __init__(self, driver):
        self.driver = driver
        self.url = "https://example.com/login"
        self.username_input = (By.ID, "username")
        self.password_input = (By.ID, "password")
        self.login_button = (By.ID, "login_button")

def open(self):
        self.driver.get(self.url)

def login(self, username, password):
        self.driver.find_element(*self.username_input).send_keys(username)
        self.driver.find_element(*self.password_input).send_keys(password)
        self.driver.find_element(*self.login_button).click()
```

**测试用例：**

Python复制

```python
# tests/test_login.py
import pytest
from pages.login_page import LoginPage

@pytest.fixture
def browser():
    driver = webdriver.Chrome()
    yield driver
    driver.quit()

def test_login_success(browser):
    login_page = LoginPage(browser)
    login_page.open()
    login_page.login("valid_user", "valid_password")
    assert "Dashboard" in browser.title

def test_login_failure(browser):
    login_page = LoginPage(browser)
    login_page.open()
    login_page.login("invalid_user", "invalid_password")
    assert "Login failed" in browser.page_source
```

**运行测试：**

bash复制

```bash
pytest --html=reports/report.html
```

------

### 总结

本章详细介绍了测试框架的基本结构、页面对象模式（POM）以及如何使用`pytest`进行测试。通过这些内容，你可以构建一个高效、可维护的自动化测试框架。页面对象模式可以提高代码的可重用性和可维护性，而`pytest`提供了强大的测试功能和灵活的插件支持。