【pytest官方文件】解讀fixtures - 2. fixtures的呼叫方式
阿新 • • 發佈:2021-03-03
既然fixtures是給執行測試做準備工作的,那麼pytest如何知道哪些測試函式 或者 fixtures要用到哪一個fixtures呢?
說白了,就是fixtures的呼叫。
### 一、測試函式宣告傳參請求fixture
測試函式通過將fixture宣告為引數來請求fixture。
```
def test_my_fruit_in_basket(my_fruit, fruit_basket):
# 這是一個測試函式
assert my_fruit in fruit_basket
```
參考上一章出現的示例,測試函式`test_my_fruit_in_basket`通過傳入`my_fruit, fruit_basket`來呼叫這2個fixture。
當pytest執行測試函式時,它會檢視該測試函式中的引數,然後搜尋與這些引數具有相同名稱的fixture。
一旦pytest找到這些物件,它就會執行這些fixture。
### 二、fixture中的返回值傳遞給測試函式
此外,如果fixture中還有返回的內容,pytest可以拿到,並將這些物件作為引數傳遞給測試函式。
舉個例子:
```
class Fruit:
def __init__(self, name):
self.name = name
self.cubed = False
def cube(self):
self.cubed = True
class FruitSalad:
def __init__(self, *fruit_bowl):
self.fruit = fruit_bowl
self._cube_fruit()
def _cube_fruit(self):
for fruit in self.fruit:
fruit.cube()
# Arrange
@pytest.fixture
def fruit_bowl():
return [Fruit("apple"), Fruit("banana")]
def test_fruit_salad(fruit_bowl):
# Act
# 這裡接收到fixture函式fruit_bowl的返回值,
# 也就是[Fruit("apple"), Fruit("banana")],並使用
fruit_salad = FruitSalad(*fruit_bowl)
# Assert
# python內建函式all(),用於判斷給定的可迭代引數 iterable 中的所有元素是否都為 TRUE,
# 如果是返回 True,否則返回 False
assert all(fruit.cubed for fruit in fruit_salad.fruit)
```
ps:其實這裡可以寫幾行非常簡單的程式碼說明意思,不過突然覺得看點稍微繞的程式碼也沒啥壞處。
可能python不太熟悉的朋友會覺得官方示例比較晦澀,其實我們重點不是關注這個,而且弄明白這裡面的傳遞關係:
1. 首先,測試函式`test_fruit_salad `請求`fruit_bowl`(也就是 `def test_fruit_salad(fruit_bowl):`)
2. 此時,pytest將會執行這個fixture函式`fruit_bowl`,並將返回的物件作為`fruit_bowl`引數傳遞給測試函式`test_fruit_salad`。
這就是當一個fixture被請求呼叫的時候,發生的事情。
如果上面的fixture函式做的事情換做我們自己手動來執行,應該是這樣的:
```
# 上面的2個類不變
...
def fruit_bowl():
return [Fruit("apple"), Fruit("banana")]
def test_fruit_salad(fruit_bowl):
# Act
fruit_salad = FruitSalad(*fruit_bowl)
# Assert
assert all(fruit.cubed for fruit in fruit_salad.fruit)
# Arrange
bowl = fruit_bowl()
test_fruit_salad(fruit_bowl=bowl)
```
相信看到這裡,大家應該對fixture的呼叫過程已經瞭解。
如果覺得官方程式碼示例有些晦澀,那麼這裡再附上一個簡易版的:
```
import pytest
# Arrange
@pytest.fixture
def fruit_bowl():
return ["蘋果", "香蕉"]
def test_fruit_salad(fruit_bowl):
# Act
fruit_salad = fruit_bowl[0] + fruit_bowl[1]
# Assert
assert fruit_salad == "蘋果香蕉"
```
接下來,繼續跟著官方文件解讀fixture的特點:fixture呼叫別的fixture、fixture的復