Golang如何测试文件上传与下载_Golang 文件上传下载测试方法

func handleUpload(w http.ResponseWriter, r *http.Request) {
    err := r.ParseMultipartForm(32 << 20) // 32MB limit
    if err != nil {
        http.Error(w, "Unable to parse form", http.StatusBadRequest)
        return
    }

    file, handler, err := r.FormFile("uploadfile")
    if err != nil {
        http.Error(w, "Error retrieving file", http.StatusBadRequest)
        return
    }
    defer file.Close()

    // 模拟保存文件（实际可写入磁盘或存储系统）
    fmt.Fprintf(w, "Uploaded file: %s, Size: %d", handler.Filename, handler.Size)
}

func TestUploadFile(t *testing.T) {
    body := new(bytes.Buffer)
    writer := multipart.NewWriter(body)
    part, _ := writer.CreateFormFile("uploadfile", "test.txt")
    part.Write([]byte("hello world"))
    writer.Close()

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

    handleUpload(w, req)

    if w.Code != http.StatusOK {
        t.Errorf("Expected status 200, got %d", w.Code)
    }

    expected := "Uploaded file: test.txt, Size: 11"
    if !strings.Contains(w.Body.String(), expected) {
        t.Errorf("Expected response to contain %q, got %q", expected, w.Body.String())
    }
}

func handleDownload(w http.ResponseWriter, r *http.Request) {
    filename := r.URL.Query().Get("f")
    if filename == "" {
        http.Error(w, "Missing file name", http.StatusBadRequest)
        return
    }

    w.Header().Set("Content-Disposition", fmt.Sprintf("attachment; filename=%s", filename))
    w.Header().Set("Content-Type", "application/octet-stream")

    fmt.Fprint(w, "mock file content")
}

func TestDownloadFile(t *testing.T) {
    req := httptest.NewRequest("GET", "/download?f=test.txt", nil)
    w := httptest.NewRecorder()

    handleDownload(w, req)

    if w.Code != http.StatusOK {
        t.Errorf("Expected 200 OK, got %v", w.Code)
    }

    content := w.Header().Get("Content-Type")
    if content != "application/octet-stream" {
        t.Errorf("Expected Content-Type application/octet-stream, got %s", content)
    }

    disposition := w.Header().Get("Content-Disposition")
    if !strings.Contains(disposition, "attachment; filename=test.txt") {
        t.Errorf("Wrong Content-Disposition: %s", disposition)
    }

    body := w.Body.String()
    if body != "mock file content" {
        t.Errorf("Expected body 'mock file content', got %q", body)
    }
}

func TestIntegration(t *testing.T) {
    mux := http.NewServeMux()
    mux.HandleFunc("/upload", handleUpload)
    mux.HandleFunc("/download", handleDownload)

    server := httptest.NewServer(mux)
    defer server.Close()

    // 测试上传
    resp, err := http.Post(server.URL+"/upload", "text/plain", strings.NewReader("test data"))
    if err != nil {
        t.Fatal(err)
    }
    resp.Body.Close()

    if resp.StatusCode != http.StatusOK {
        t.Errorf("Upload failed with status: %d", resp.StatusCode)
    }
}