Bagaimana untuk mengejek Fungsi Terbuka dengan Pernyataan dalam Ujian Unit Python?

Mengejek Terbuka dengan Penyata Dengan dalam Python

Apabila menguji kod yang menggunakan fungsi open() dengan pernyataan with, ia menjadi perlu untuk mengejek panggilan terbuka untuk menegaskan tingkah laku yang diharapkan. Begini cara untuk melakukannya menggunakan rangka kerja Mock dalam Python:

Python 3

1. Patch Builtins.open: Tampal binaan. buka fungsi menggunakan mock_open daripada rangka kerja olok-olok.
2. Gunakan Patch sebagai Pengurus Konteks: Gunakan tampung sebagai pengurus konteks, yang mengembalikan objek ejek yang menggantikan yang asal.
3. Panggilan Terbuka: Buka fail menggunakan laluan fail.
4. Tegaskan Kandungan: Tegaskan bahawa kandungan yang dibaca daripada fail adalah seperti yang dijangkakan.
5. Tegaskan Panggilan Diejek: Tegaskan bahawa objek yang diejek telah dipanggil dengan argumen laluan fail yang betul.

<code class="python">from unittest.mock import patch, mock_open

with patch("builtins.open", mock_open(read_data="data")):
    mock_file = open("path/to/open")
    assert mock_file.read() == "data"
mock_file.assert_called_with("path/to/open")</code>

Sebagai alternatif, anda boleh menggunakan tampung sebagai penghias dengan argumen new_callable ditetapkan kepada mock_open:

<code class="python">@patch("builtins.open", new_callable=mock_open, read_data="data")
def test_patch(mock_file):
    open("path/to/open")
    assert mock_file.read() == "data"
    mock_file.assert_called_with("path/to/open")</code>

Python 2

1. Patch __builtin__.open: Patch __builtin__.open bukannya builtin.open in 2.
2. Import Mock: Pasang mock menggunakan pip install mock.
3. Gunakan Patch sebagai Pengurus Konteks: Ikuti langkah yang sama seperti dalam Python 3.

Atas ialah kandungan terperinci Bagaimana untuk mengejek Fungsi Terbuka dengan Pernyataan dalam Ujian Unit Python?. Untuk maklumat lanjut, sila ikut artikel berkaitan lain di laman web China PHP!