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

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!