Comment se moquer de « http.Request.FormFile » dans les tests Go ?

Test de Request.FormFile dans Go

Lors du test d'un point de terminaison impliquant des téléchargements de fichiers à l'aide de http.Request.FormFile, il devient nécessaire de générer une requête avec une valeur qui peut être récupéré via cette méthode.

La bibliothèque httptest ne fournit pas de moyen simple de se moquer de la structure FormFile complète. Cependant, en utilisant le package mime/multipart, vous pouvez créer un FormFile à l'aide de la fonction CreateFormFile :

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

Cette fonction prend un nom de champ et un nom de fichier comme paramètres et renvoie un io.Writer que vous pouvez utiliser pour écrire les données réelles du fichier.

Pour utiliser CreateFormFile dans un test, vous pouvez soit écrire les données dans un tampon io.ReaderWriter, soit utiliser un io.Pipe. L'exemple suivant utilise un 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>

Cet exemple génère dynamiquement un fichier image à l'aide du package image et l'écrit dans l'enregistreur multipart via une interface io.Writer. Vous pouvez également utiliser une approche similaire pour créer des données dans d'autres formats (par exemple CSV) à la volée.

Ce qui précède est le contenu détaillé de. pour plus d'informations, suivez d'autres articles connexes sur le site Web de PHP en chinois!