在上一篇文章中,我們建立了 Pytest 夾具,它將在測試方法之前/之後建立/刪除 Postgres 資料庫。在這一部分中,我想在Pytest 工廠固定裝置的幫助下改進固定裝置,使其更加靈活和可配置。
例如,如果您有多個資料庫要在測試中模擬
您必須創建幾乎兩個相同的燈具:
「靜態」裝置在這裡有點限制。當需要幾乎相同而僅有細微差別時,您需要複製程式碼。希望 Pytest 有工廠作為固定裝置的概念。
工廠固定裝置是一個返回另一個固定裝置的固定裝置。因為,像每個工廠一樣,它是一個函數,它可以接受參數來自訂返回的固定裝置。按照慣例,您可以在它們前面加上 make_* 前綴,例如 make_test_db。
我們的裝置工廠 make_test_db 的唯一參數將是要建立/刪除的測試資料庫名稱。
那麼,讓我們基於 make_test_db 工廠裝置來建立兩個「專用」裝置。
用法如下:
你注意到產量了嗎? Yield 和 Yield 之間的一個關鍵區別在於它們如何處理生成器內的資料流和控制。
在Python中,yield和yield from都在生成器函數中使用來產生一系列值,但是
也就是說,我們不想從專門的夾具“屈服”,而是從夾具工廠“屈服”。因此這裡需要yield from。
除了將程式碼包裝到內部函數之外,對我們原始夾具創建/刪除資料庫所需的更改實際上幾乎不需要任何更改。
在上一部分中,我還有一個固定裝置,將 Yoyo 遷移應用於剛剛建立的空資料庫。它也不是很靈活。讓我們做同樣的事情並將實際程式碼包裝到內部函數中。
在這種情況下,因為程式碼不需要在從測試方法返回後進行清理(其中沒有yield),所以
需要兩個資料庫並對它們應用遷移的測試方法:
建立自己的夾具工廠,為 Pytest 方法建立和刪除資料庫實際上是練習 Python 產生器和運算子的產量/產量的一個很好的練習。
我希望這篇文章對您自己的資料庫測試套件有所幫助。請隨時在評論中留下您的問題,祝您編碼愉快!
以上是Pytest 和 PostgreSQL:每次測試的新資料庫(第二部分)的詳細內容。更多資訊請關注PHP中文網其他相關文章!