Rumah >pembangunan bahagian belakang >Golang >Bagaimana untuk Mensimulasikan Muat Naik Fail dalam Ujian Go Http.Request.FormFile?

Bagaimana untuk Mensimulasikan Muat Naik Fail dalam Ujian Go Http.Request.FormFile?

Patricia Arquette
Patricia Arquetteasal
2024-11-03 08:21:021084semak imbas

How to Simulate File Upload in Go Http.Request.FormFile Tests?

Menguji Go Http.Request.FormFile

Menguji titik akhir yang menjangkakan data borang berbilang bahagian memerlukan simulasi muat naik fail. Ini boleh dicapai dengan menjana permintaan pos dengan fail borang menggunakan perpustakaan httptest.

Daripada mengejek struct FormFile yang lengkap, kami boleh menggunakan pakej mime/multipart untuk mencipta fail borang. Fungsi CreateFormFile menjana pengepala medan berbilang bahagian dengan nama medan dan nama fail yang ditentukan.

Dengan menghantar io.Writer yang dibuat oleh CreateFormFile ke httptest.NewRequest, kami boleh mensimulasikan permintaan siaran dengan fail borang.

Berikut ialah contoh bagaimana ini boleh dilaksanakan:

<code class="go">func TestUploadImage(t *testing.T) {
    // Set up an io.Pipe to avoid buffering
    pr, pw := io.Pipe()
    writer := multipart.NewWriter(pw)

    go func() {
        defer writer.Close()
        part, err := writer.CreateFormFile("fileupload", "someimg.png")
        if err != nil {
            t.Error(err)
        }
        img := createImage()
        err = png.Encode(part, img)
        if err != nil {
            t.Error(err)
        }
    }()

    request := httptest.NewRequest("POST", "/", pr)
    request.Header.Add("Content-Type", writer.FormDataContentType())

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

    if response.Code != 200 {
        t.Errorf("Expected %s, received %d", 200, response.Code)
    }
    if _, err := os.Stat("./uploads/someimg.png"); os.IsNotExist(err) {
        t.Error("Expected file ./uploads/someimg.png' to exist")
    }
}</code>

Fungsi ini mencipta imej secara dinamik menggunakan pakej imej dan menulisnya terus kepada Penulis berbilang bahagian menggunakan png.Enkod. Ini mensimulasikan muat naik fail imej, tanpa memerlukan I/O fail sebenar.

Atas ialah kandungan terperinci Bagaimana untuk Mensimulasikan Muat Naik Fail dalam Ujian Go Http.Request.FormFile?. 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