Go 테스트에서 파일 업로드를 시뮬레이션하는 방법은 무엇입니까?

Go에서 파일 업로드 테스트

파일 업로드를 처리하는 엔드포인트를 테스트할 때 Request.FormFile 필드를 설정해야 합니다. 불행하게도 단순히 전체 FormFile 구조체를 조롱하는 것은 지나치게 복잡한 접근 방식입니다. 대신 mime/multipart 패키지를 활용하여 필요한 FormFile 인스턴스를 생성할 수 있습니다.

CreateFormFile 사용

Writer 유형의 멤버인 CreateFormFile 함수는 특정 필드 이름과 파일 이름을 사용하여 양식 데이터 헤더를 생성하는 편리한 방법을 제공합니다. 그런 다음 결과 io.Writer를 httptest.NewRequest 함수에 전달할 수 있습니다.

파이프 사용 예

한 가지 접근 방식은 FormFile을 io.ReaderWriter 버퍼에 쓰는 것입니다. 또는 io.Pipe를 사용하세요. 다음 예에서는 후자의 방법을 보여줍니다.

<code class="go">// Create a pipe to prevent buffering.
pr, pw := io.Pipe()

// Transform data to multipart form data and write it to the pipe.
writer := multipart.NewWriter(pw)
defer writer.Close()

go func() {
    // Create the "fileupload" form data field.
    part, err := writer.CreateFormFile("fileupload", "someimg.png")
    if err != nil {
        t.Error(err)
    }

    // Generate the image bytes.
    img := createImage()

    // Encode the image to the form data field writer.
    err = png.Encode(part, img)
    if err != nil {
        t.Error(err)
    }
}

// Read from the pipe into a new httptest.Request.
request := httptest.NewRequest("POST", "/", pr)
request.Header.Add("Content-Type", writer.FormDataContentType())</code>

요청 처리

요청의 FormFile 데이터를 사용하면 평소와 같이 테스트된 엔드포인트에서 이를 처리할 수 있습니다. . 예제 함수는 uploads 디렉터리에 파일을 생성하는 방법을 보여줍니다.

추가 참고 사항

이 접근 방식을 사용하면 양식 데이터를 즉시 생성하여 임시 파일로 작업할 필요 없이 프레임워크를 테스트합니다. 마찬가지로 인코딩/csv를 사용하면 파일 시스템에서 읽을 필요 없이 CSV 파일을 생성할 수 있습니다.

위 내용은 Go 테스트에서 파일 업로드를 시뮬레이션하는 방법은 무엇입니까?의 상세 내용입니다. 자세한 내용은 PHP 중국어 웹사이트의 기타 관련 기사를 참조하세요!