Comment simuler le téléchargement de fichiers dans les tests Go Http.Request.FormFile ?

Test Go Http.Request.FormFile

Le test des points de terminaison qui attendent des données de formulaire en plusieurs parties nécessite la simulation d'un téléchargement de fichier. Ceci peut être réalisé en générant une demande de publication avec un fichier de formulaire à l'aide de la bibliothèque httptest.

Au lieu de nous moquer de la structure FormFile complète, nous pouvons utiliser le package mime/multipart pour créer un fichier de formulaire. La fonction CreateFormFile génère un en-tête de champ en plusieurs parties avec le nom de champ et le nom de fichier spécifiés.

En passant le io.Writer créé par CreateFormFile à httptest.NewRequest, nous pouvons simuler une demande de publication avec un fichier de formulaire.

Voici un exemple de la façon dont cela peut être implémenté :

<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>

Cette fonction crée une image dynamiquement à l'aide du package image et l'écrit directement dans l'écrivain multipart à l'aide de png.Encode. Cela simule le téléchargement d'un fichier image, sans avoir besoin d'E/S de fichier réelles.

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!