Rumah >pembangunan bahagian belakang >Tutorial Python >Cara Mengejek Permintaan Respons Modul dalam Python

Cara Mengejek Permintaan Respons Modul dalam Python

Susan Sarandon
Susan Sarandonasal
2024-10-20 19:47:30861semak imbas

How to Mock Requests Module Responses in Python

Permintaan Mengejek untuk Mengawal Respons

Dalam Python, pakej olok-olok menyediakan cara yang berkuasa untuk mengejek modul atau kelas luaran, membolehkan anda memanipulasi tingkah laku dan mengesahkan interaksi. Dalam konteks permintaan HTTP, mengejek modul Permintaan boleh sangat berguna untuk menguji kod yang bergantung pada perkhidmatan luaran.

Langkah 1: Mock the Requests Module

Untuk mengejek modul Permintaan, anda perlu menampal fungsi get() dengan fungsi tersuai yang mengembalikan respons yang diingini. Anda boleh mentakrifkan kaedah olok-olok seperti ini:

<code class="python">def mocked_requests_get(url, **kwargs):
  if url == "aurl":
    return MockResponse("a response")
  elif url == "burl":
    return MockResponse("b response")
  else:
    raise Exception("URL not mocked")</code>

Perhatikan bahawa kaedah ini menjangkakan URL yang sah dan mengembalikan objek MockResponse, yang mewakili respons berpura-pura dengan kandungan yang dipratentukan.

Langkah 2: Tampal Modul Permintaan Asal

Setelah anda menentukan kaedah olok-olok, anda boleh menampal requests.get() asal dengannya menggunakan penghias @mock.patch. Ini akan menggantikan semua panggilan ke requests.get() dalam kod yang anda uji dengan tingkah laku ejek anda.

<code class="python">@mock.patch("requests.get", side_effect=mocked_requests_get)
def test_myview(self, mock_get):
  # Your test goes here</code>

Langkah 3: Panggil Lihat dan Sahkan Respons

Kini anda boleh menghubungi fungsi anda seperti biasa dan mengesahkan bahawa jawapan yang dijangkakan telah diperolehi. Objek olok-olok boleh diperiksa untuk menegaskan bahawa fungsi get() dipanggil dengan argumen tertentu dan mengembalikan nilai yang dikehendaki.

Kod Contoh:

<code class="python">import requests
from unittest import mock

class MyViewTest(unittest.TestCase):
    # ...
    def test_myview(self, mock_get):
        self.assertEqual(res1.text, "a response")
        self.assertEqual(res2.text, "b response")
        self.assertEqual(res3.text, "c response")

        # Verify mock calls
        mock_get.assert_called_with('aurl')
        mock_get.assert_called_with('burl')
        mock_get.assert_called_with('curl')</code>

Ingat untuk mengesahkan kedua-dua kandungan teks bagi respons dan kiraan panggilan serta hujah yang dihantar kepada kaedah olok-olok anda. Ini membolehkan anda memastikan bahawa interaksi yang diharapkan berlaku dan tingkah laku yang diingini tercapai.

Atas ialah kandungan terperinci Cara Mengejek Permintaan Respons Modul dalam Python. Untuk maklumat lanjut, sila ikut artikel berkaitan lain di laman web China PHP!

Kenyataan:
Kandungan artikel ini disumbangkan secara sukarela oleh netizen, dan hak cipta adalah milik pengarang asal. Laman web ini tidak memikul tanggungjawab undang-undang yang sepadan. Jika anda menemui sebarang kandungan yang disyaki plagiarisme atau pelanggaran, sila hubungi admin@php.cn