我們都知道go語言的testing包提供了豐富的測試功能，方便我們在開發(fā)時進行單元測試，但是之前一直沒有看到過如何進行文件上傳單元測試相關的文章，直到看到了B站的這個視頻「教程」Go語言基礎 (O'Reilly)，不得不說這個go語言學習視頻比國內(nèi)的不知要高到哪里去了，講解清晰，涵蓋范圍廣，學完感覺水平瞬間上了一個等級。

文件上傳服務端代碼

func upload(w http.ResponseWriter, r *http.Request) {
    file, head, err := r.FormFile("my_file")
    if err != nil {
        fmt.Sprintln(err)
        fmt.Fprintln(w, err)

        return
    }

    localFileDir := "/tmp/upload/"
    err = os.MkdirAll(localFileDir, 0777)
    if err != nil {
        fmt.Sprintln(err)
        fmt.Fprintln(w, err)

        return
    }

    localFilePath := localFileDir + head.Filename

    localFile, err := os.Create(localFilePath)
    if err != nil {
        fmt.Sprintln(err)
        fmt.Fprintln(w, err)

        return
    }
    defer localFile.Close()

    io.Copy(localFile, file)
    fmt.Fprintln(w, localFilePath)

}

測試代碼

func TestUpload(t *testing.T) {
    path := "/home/ubuntu/test.go"http://要上傳文件所在路徑
    file, err := os.Open(path)
    if err != nil {
        t.Error(err)
    }

    defer file.Close()
    body := &bytes.Buffer{}
    writer := multipart.NewWriter(body)
    part, err := writer.CreateFormFile("my_file", filepath.Base(path))
    if err != nil {
                writer.Close()
        t.Error(err)
    }
    io.Copy(part, file)
    writer.Close()

    req := httptest.NewRequest("POST", "/upload", body)
    req.Header.Set("Content-Type", writer.FormDataContentType())
    res := httptest.NewRecorder()

    upload(res, req)

    if res.Code != http.StatusOK {
        t.Error("not 200")
    }

    t.Log(res.Body.String())
    // t.Log(io.read)

}

測試代碼中關鍵的部分在于使用了"mime/multipart"包

1. 首先創(chuàng)建一個writer

    body := &bytes.Buffer{}
    writer := multipart.NewWriter(body)

2. 然后往multipart中寫入域"my_file"和文件名filepath.Base(path)

"my_file"和服務端中
file, head, err := r.FormFile("my_file")
對應。

3.最后上傳文件

    io.Copy(part, file)
    writer.Close()