Bagaimana untuk mengejek `http.Request.FormFile` dalam Go Tests?

Barbara Streisand
Lepaskan: 2024-10-31 12:31:02
asal
839 orang telah melayarinya

How to Mock `http.Request.FormFile` in Go Tests?

Permintaan Pengujian.FormFile dalam Go

Dalam menguji titik akhir yang melibatkan muat naik fail menggunakan http.Request.FormFile, ia menjadi perlu untuk menjana permintaan dengan nilai yang boleh diperoleh melalui kaedah ini.

Perpustakaan httptest tidak menyediakan cara yang mudah untuk mengejek struct FormFile yang lengkap. Walau bagaimanapun, menggunakan pakej mime/multipart, anda boleh mencipta FormFile menggunakan fungsi CreateFormFile:

<code class="go">func (w *Writer) CreateFormFile(fieldname, filename string) (io.Writer, error)</code>
Salin selepas log masuk

Fungsi ini mengambil nama medan dan nama fail sebagai parameter dan mengembalikan io.Writer yang boleh anda gunakan untuk menulis data fail sebenar.

Untuk menggunakan CreateFormFile dalam ujian, anda boleh sama ada menulis data ke penimbal io.ReaderWriter atau menggunakan io.Pipe. Contoh berikut menggunakan io.Pipe:

<code class="go">func TestUploadImage(t *testing.T) {
    // Set up a pipe to avoid buffering
    pr, pw := io.Pipe()
    // Create a multipart form data writer using the pipe as the destination
    writer := multipart.NewWriter(pw)

    go func() {
        defer writer.Close()
        // Create the 'fileupload' form data field
        part, err := writer.CreateFormFile("fileupload", "someimg.png")
        if err != nil {
            t.Error(err)
        }
        // Write an image to the form data field using an io.Writer interface
        // (e.g., png.Encode)
    }()

    // Read from the pipe, which contains the multipart form data generated by the multipart writer
    request := httptest.NewRequest("POST", "/", pr)
    request.Header.Add("Content-Type", writer.FormDataContentType())

    response := httptest.NewRecorder()
    handler := UploadFileHandler()
    handler.ServeHTTP(response, request)

    // Assert HTTP status code and other test verifications
}</code>
Salin selepas log masuk

Contoh ini menjana fail imej secara dinamik menggunakan pakej imej dan menulisnya kepada penulis berbilang bahagian melalui antara muka io.Writer. Anda juga boleh menggunakan pendekatan yang serupa untuk mencipta data dalam format lain (cth., CSV) dengan segera.

Atas ialah kandungan terperinci Bagaimana untuk mengejek `http.Request.FormFile` dalam Go Tests?. Untuk maklumat lanjut, sila ikut artikel berkaitan lain di laman web China PHP!

sumber:php.cn
Kenyataan Laman Web ini
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
Artikel terbaru oleh pengarang
Tutorial Popular
Lagi>
Muat turun terkini
Lagi>
kesan web
Kod sumber laman web
Bahan laman web
Templat hujung hadapan
Tentang kita Penafian Sitemap
Laman web PHP Cina:Latihan PHP dalam talian kebajikan awam,Bantu pelajar PHP berkembang dengan cepat!