Pytest 和 PostgreSQL:每次測試的新資料庫(第二部分)

WBOY
發布: 2024-09-03 16:09:19
原創
670 人瀏覽過

Pytest and PostgreSQL: Fresh database for every test (part II)

在上一篇文章中,我們建立了 Pytest 夾具,它將在測試方法之前/之後建立/刪除 Postgres 資料庫。在這一部分中,我想在Pytest 工廠固定裝置的幫助下改進固定裝置,使其更加靈活和可配置

靜態夾具的限制

例如,如果您有多個資料庫要在測試中模擬

雷雷

您必須創建幾乎兩個相同的燈具:

雷雷

Pytest 夾具工廠

「靜態」裝置在這裡有點限制。當需要幾乎相同而僅有細微差別時,您需要複製程式碼。希望 Pytest 有工廠作為固定裝置的概念。

工廠固定裝置是一個返回另一個固定裝置的固定裝置。因為,像每個工廠一樣,它是一個函數,它可以接受參數來自訂返回的固定裝置。按照慣例,您可以在它們前面加上 make_* 前綴,例如 make_test_db。

專用夾具

我們的裝置工廠 make_test_db 的唯一參數將是要建立/刪除的測試資料庫名稱。

那麼,讓我們基於 make_test_db 工廠裝置來建立兩個「專用」裝置。

用法如下:

雷雷

附註:產量來自

你注意到產量了嗎? Yield 和 Yield 之間的一個關鍵區別在於它們如何處理生成器內的資料流和控制。

在Python中,yield和yield from都在生成器函數中使用來產生一系列值,但是

  • Yield 用於暫停生成器函數的執行並向呼叫者傳回單一值。
  • 而yield from用於將值的產生委託給另一個產生器。它本質上「展平」了嵌套生成器,將其生成的值直接傳遞給外部生成器的呼叫者。

也就是說,我們不想從專門的夾具“屈服”,而是從夾具工廠“屈服”。因此這裡需要yield from。

用於建立/刪除資料庫的夾具工廠

除了將程式碼包裝到內部函數之外,對我們原始夾具創建/刪除資料庫所需的更改實際上幾乎不需要任何更改。

雷雷

獎勵:將遷移固定裝置重寫為工廠固定裝置

在上一部分中,我還有一個固定裝置,將 Yoyo 遷移應用於剛剛建立的空資料庫。它也不是很靈活。讓我們做同樣的事情並將實際程式碼包裝到內部函數中。

在這種情況下,因為程式碼不需要在從測試方法返回後進行清理(其中沒有yield),所以

  • 工廠裝置回傳(不是yield)內部函數
  • 專門的夾具呼叫(不是從工廠夾具產生)
雷雷

需要兩個資料庫並對它們應用遷移的測試方法:

雷雷

結論

建立自己的夾具工廠,為 Pytest 方法建立和刪除資料庫實際上是練習 Python 產生器和運算子的產量/產量的一個很好的練習。

我希望這篇文章對您自己的資料庫測試套件有所幫助。請隨時在評論中留下您的問題,祝您編碼愉快!

以上是Pytest 和 PostgreSQL:每次測試的新資料庫(第二部分)的詳細內容。更多資訊請關注PHP中文網其他相關文章!

來源:dev.to
本網站聲明
本文內容由網友自願投稿,版權歸原作者所有。本站不承擔相應的法律責任。如發現涉嫌抄襲或侵權的內容,請聯絡admin@php.cn
最新下載
更多>
網站特效
網站源碼
網站素材
前端模板
關於我們 免責聲明 Sitemap
PHP中文網:公益線上PHP培訓,幫助PHP學習者快速成長!