如何為僅使用記錄器語句的except塊編寫python單元測試用例

Question

芹菜任務代碼：

import logging

from celery_once import QueueOnce
from celery import shared_task, current_task
from test.set_data import set_data_in_db

logger = logging.getLogger("celery")


@shared_task(base=QueueOnce, once={"graceful": True}, ignore_result=False)
def set_data_task():
    try:
        logger.info("Set data in db initiated")
        set_data_in_db(value=None)

    except Exception:
        logger.error("data could not be set", exc_info=True)

我的單元測試用例涵蓋了 try 塊中的所有內容。 如何強制我的單元測試也覆蓋塊除外？

Answer 1

當您調用方法set_data_task ，可能只有兩種情況中的一種，方法正常執行（try 塊）或拋出一些異常（塊除外）。

如果您需要測試您的except塊，您需要將您的方法配置為拋出異常，以便它可以在您的測試中被捕獲和斷言

您將需要配置set_data_in_db方法來拋出異常被捕獲

對於測試異常，您可以閱讀pytest.raises